Make rpath usage configurable

rapidsai · Sep 30, 2024 · 128c2a5 · 128c2a5
1 parent 606a0ac
commit 128c2a5
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 8 deletions.
diff --git a/ci/build_wheel_python.sh b/ci/build_wheel_python.sh
@@ -24,6 +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" \
     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,6 +26,8 @@ project(
   LANGUAGES CXX CUDA
 )
 
+option(USE_CUDA_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,11 +22,13 @@ rapids_cython_create_modules(
   SOURCE_FILES "${cython_modules}"
   LINKED_LIBRARIES kvikio::kvikio nvcomp::nvcomp
 )
-set(rpaths "$ORIGIN/../../nvidia/nvcomp")
-foreach(tgt IN LISTS RAPIDS_CYTHON_CREATED_TARGETS)
-  set_property(
-    TARGET ${tgt}
-    PROPERTY INSTALL_RPATH ${rpaths}
-    APPEND
-  )
-endforeach()
+if(USE_CUDA_NVCOMP_WHEEL)
+  set(rpaths "$ORIGIN/../../nvidia/nvcomp")
+  foreach(tgt IN LISTS RAPIDS_CYTHON_CREATED_TARGETS)
+    set_property(
+      TARGET ${tgt}
+      PROPERTY INSTALL_RPATH ${rpaths}
+      APPEND
+    )
+  endforeach()
+endif()