From 749f2c214821aa881d1c07f6fa8222523f4f2489 Mon Sep 17 00:00:00 2001 From: "H. Vetinari" Date: Wed, 13 Sep 2023 11:42:38 +1100 Subject: [PATCH] use newer CUDAToolkit_ROOT variable for CMake --- recipe/build-arrow.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipe/build-arrow.sh b/recipe/build-arrow.sh index 629a93094..ef0b03881 100644 --- a/recipe/build-arrow.sh +++ b/recipe/build-arrow.sh @@ -30,7 +30,7 @@ fi # Enable CUDA support if [[ ! -z "${cuda_compiler_version+x}" && "${cuda_compiler_version}" != "None" ]] then - EXTRA_CMAKE_ARGS=" ${EXTRA_CMAKE_ARGS} -DARROW_CUDA=ON -DCUDA_TOOLKIT_ROOT_DIR=${CUDA_HOME} -DCMAKE_LIBRARY_PATH=${CONDA_BUILD_SYSROOT}/lib" + EXTRA_CMAKE_ARGS=" ${EXTRA_CMAKE_ARGS} -DARROW_CUDA=ON -DCUDAToolkit_ROOT=${CUDA_HOME} -DCMAKE_LIBRARY_PATH=${CONDA_BUILD_SYSROOT}/lib" else EXTRA_CMAKE_ARGS=" ${EXTRA_CMAKE_ARGS} -DARROW_CUDA=OFF" fi