From f7d74c98ebaab28005173bd5dd866c7845167802 Mon Sep 17 00:00:00 2001 From: AbdelRauf Date: Tue, 9 Mar 2021 02:51:29 +0100 Subject: [PATCH] cross_build: set jetson compute 5.3. small modifications Signed-off-by: AbdelRauf --- libnd4j/cmake/rpi.cmake | 4 ++-- libnd4j/pi_build.sh | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/libnd4j/cmake/rpi.cmake b/libnd4j/cmake/rpi.cmake index a7f7b0890..6e4438bff 100644 --- a/libnd4j/cmake/rpi.cmake +++ b/libnd4j/cmake/rpi.cmake @@ -20,8 +20,8 @@ SET(CMAKE_SYSTEM_NAME Linux) SET(CMAKE_SYSTEM_VERSION 1) -set(CMAKE_C_COMPILER "$ENV{RPI_BIN}-gcc" CACHE STRING "" FORCE) -set(CMAKE_CXX_COMPILER "$ENV{RPI_BIN}-g++" CACHE STRING "" FORCE) +set(CMAKE_C_COMPILER "$ENV{RPI_BIN}-gcc" ) +set(CMAKE_CXX_COMPILER "$ENV{RPI_BIN}-g++" ) if (SD_CUDA) if(${SD_ARCH} MATCHES "armv8") diff --git a/libnd4j/pi_build.sh b/libnd4j/pi_build.sh index fcd85ddab..178b4810a 100755 --- a/libnd4j/pi_build.sh +++ b/libnd4j/pi_build.sh @@ -345,8 +345,8 @@ else message "jetson cuda build " cuda_cross_setup ${CUDA_VER} XTRA_ARGS="${XTRA_ARGS} -c cuda -h cudnn " - XTRA_MVN_ARGS="${XTRA_MVN_ARGS} -Dcuda.version=${CUDA_VER} -Dlibnd4j.cuda=${CUDA_VER} -Dlibnd4j.chip=cuda " - XTRA_MVN_ARGS="${XTRA_MVN_ARGS} -Dlibnd4j.helper=cudnn " + XTRA_MVN_ARGS="${XTRA_MVN_ARGS} -Djavacpp.version=1.5.3 -Dcuda.version=${CUDA_VER} -Dlibnd4j.cuda=${CUDA_VER} -Dlibnd4j.chip=cuda -Dlibnd4j.compute=5.3 " + XTRA_MVN_ARGS="${XTRA_MVN_ARGS} -Dlibnd4j.helper=cudnn " export SYSROOT=${CROSS_COMPILER_DIR}/${PREFIX}/libc else XTRA_MVN_ARGS="${XTRA_MVN_ARGS} -pl \":libnd4j,:nd4j-native\" "