Switch to using rapids-cython

rapidsai · Jul 28, 2023 · 1851e2e · 1851e2e
1 parent ef7c961
commit 1851e2e
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 66 deletions.
diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
@@ -34,15 +34,9 @@ option(FIND_KVIKIO_CPP
        "Search for existing KVIKIO C++ installations before defaulting to local files" OFF
 )
 
-find_package(PythonExtensions REQUIRED)
-find_package(Cython REQUIRED)
-
 # TODO: Should we symlink FindcuFile.cmake into python/cmake? find cuFile
 include(../cpp/cmake/Modules/FindcuFile.cmake)
 
-# Ignore unused variable warning.
-set(ignored_variable "${SKBUILD}")
-
 if(FIND_KVIKIO_CPP)
   find_package(KvikIO ${kvikio_version})
 else()
@@ -53,7 +47,12 @@ find_package(CUDAToolkit REQUIRED)
 
 if(NOT KVIKIO_FOUND)
   add_subdirectory(../cpp kvikio-cpp)
+  set(cython_lib_dir kvikio)
+  install(TARGETS kvikio DESTINATION ${cython_lib_dir})
 endif()
 
+include(rapids-cython)
+rapids_cython_init()
+
 add_subdirectory(cmake)
 add_subdirectory(kvikio/_lib)
diff --git a/python/cmake/CMakeLists.txt b/python/cmake/CMakeLists.txt
@@ -13,4 +13,3 @@
 # =============================================================================
 
 include(thirdparty/get_nvcomp.cmake)
-include(kvikio_python_helpers.cmake)
diff --git a/python/cmake/kvikio_python_helpers.cmake b/python/cmake/kvikio_python_helpers.cmake
diff --git a/python/kvikio/_lib/CMakeLists.txt b/python/kvikio/_lib/CMakeLists.txt
@@ -13,10 +13,10 @@
 # =============================================================================
 
 # Set the list of Cython files to build
-set(cython_modules arr libnvcomp libnvcomp_ll libkvikio)
+set(cython_modules arr.pyx libnvcomp.pyx libnvcomp_ll.pyx libkvikio.pyx)
 
-# Build all of the Cython targets
-add_cython_modules("${cython_modules}")
-
-target_link_libraries(libnvcomp nvcomp::nvcomp)
-target_link_libraries(libnvcomp_ll nvcomp::nvcomp)
+rapids_cython_create_modules(
+    CXX
+    SOURCE_FILES "${cython_modules}"
+    LINKED_LIBRARIES nvcomp::nvcomp
+)