Update variable name

rapidsai · Oct 1, 2024 · c3ba673 · c3ba673
1 parent 8d498c8
commit c3ba673
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/ci/build_wheel_python.sh b/ci/build_wheel_python.sh
@@ -24,7 +24,7 @@ cd "${package_dir}"
 echo "libkvikio-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${CPP_WHEELHOUSE}/libkvikio_*.whl)" > ./constraints.txt
 
 PIP_CONSTRAINT="${PWD}/constraints.txt" \
-SKBUILD_CMAKE_ARGS="-DUSE_CUDA_NVCOMP_WHEEL=ON" \
+SKBUILD_CMAKE_ARGS="-DUSE_NVCOMP_WHEEL=ON" \
     python -m pip wheel . -w dist -vvv --no-deps --disable-pip-version-check
 
 mkdir -p final_dist

diff --git a/python/kvikio/CMakeLists.txt b/python/kvikio/CMakeLists.txt
@@ -26,7 +26,7 @@ project(
   LANGUAGES CXX CUDA
 )
 
-option(USE_CUDA_NVCOMP_WHEEL "Use the CUDA nvcomp wheel instead of the system library" OFF)
+option(USE_NVCOMP_WHEEL "Use the CUDA nvcomp wheel instead of the system library" OFF)
 
 # TODO: Should we symlink FindcuFile.cmake into python/cmake? find cuFile
 include(../../cpp/cmake/Modules/FindcuFile.cmake)

diff --git a/python/kvikio/kvikio/_lib/CMakeLists.txt b/python/kvikio/kvikio/_lib/CMakeLists.txt
@@ -22,7 +22,7 @@ rapids_cython_create_modules(
   SOURCE_FILES "${cython_modules}"
   LINKED_LIBRARIES kvikio::kvikio nvcomp::nvcomp
 )
-if(USE_CUDA_NVCOMP_WHEEL)
+if(USE_NVCOMP_WHEEL)
   set(rpaths "$ORIGIN/../../nvidia/nvcomp")
   foreach(tgt IN LISTS RAPIDS_CYTHON_CREATED_TARGETS)
     set_property(