[Feature] Add Nvcomp python bindings to kvikio.

python/CMakeLists.txt

@@ -24,7 +24,8 @@ project(
             # various linking options for the python library is hardcoded to build with C, so until
             # that is fixed we need to keep C.
             C
-            CXX)
+            CXX
+            CUDA)
 
 option(FIND_KVIKIO_CPP "Search for existing KVIKIO C++ installations before defaulting to local files"
        OFF)
@@ -45,6 +46,8 @@ else()
     set(KVIKIO_FOUND OFF)
 endif()
 
+find_package(CUDAToolkit REQUIRED)
+
 if(NOT KVIKIO_FOUND)
   add_subdirectory(../cpp kvikio-cpp)
 endif()

python/cmake/CMakeLists.txt

@@ -12,4 +12,5 @@
 # the License.
 # =============================================================================
 
+include(thirdparty/get_nvcomp.cmake)
 include(kvikio_python_helpers.cmake)

python/cmake/thirdparty/get_nvcomp.cmake

@@ -0,0 +1,40 @@
+# =============================================================================
+# Copyright (c) 2022, NVIDIA CORPORATION.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+# in compliance with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software distributed under the License
+# is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+# or implied. See the License for the specific language governing permissions and limitations under
+# the License.
+# =============================================================================
+
+# This function finds nvcomp and sets any additional necessary environment variables.
+function(find_and_configure_nvcomp VERSION)
+
+  # Find or install nvcomp
+  rapids_cpm_find(
+    nvcomp ${VERSION}
+    GLOBAL_TARGETS nvcomp::nvcomp
+    CPM_ARGS GITHUB_REPOSITORY NVIDIA/nvcomp
+    GIT_TAG c435afaf4ba8a8d12f379d688effcb185886cec1
+    OPTIONS "BUILD_STATIC ON" "BUILD_TESTS OFF" "BUILD_BENCHMARKS OFF" "BUILD_EXAMPLES OFF"
+  )
+
+  if(NOT TARGET nvcomp::nvcomp)
+    add_library(nvcomp::nvcomp ALIAS nvcomp)
+  endif()
+
+  # Per-thread default stream
+  if(TARGET nvcomp AND PER_THREAD_DEFAULT_STREAM)
+    target_compile_definitions(nvcomp PRIVATE CUDA_API_PER_THREAD_DEFAULT_STREAM)
+  endif()
+
+endfunction()
+
+set(CUDF_MIN_VERSION_nvCOMP 2.1.0)
+
+find_and_configure_nvcomp(${CUDF_MIN_VERSION_nvCOMP})

python/kvikio/__init__.py

@@ -1,7 +1,6 @@
 # Copyright (c) 2021-2022, NVIDIA CORPORATION. All rights reserved.
 # See file LICENSE for terms.
 
-
 from . import _version
 from ._lib import libkvikio  # type: ignore
 

python/kvikio/_lib/CMakeLists.txt

@@ -13,7 +13,9 @@
 # =============================================================================
 
 # Set the list of Cython files to build
-set(cython_modules arr libkvikio)
-
+set(cython_modules arr libnvcomp libkvikio)
+  
 # Build all of the Cython targets
 add_cython_modules("${cython_modules}")
+
+target_link_libraries(libnvcomp nvcomp)