Use nvcomp conda package.

rapidsai · Jun 14, 2023 · fb5426f · fb5426f
1 parent d9eee8b
commit fb5426f
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 1 deletion.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -22,6 +22,7 @@ dependencies:
 - libcufile=1.4.0.31
 - ninja
 - numpy>=1.21
+- nvcomp==2.6.1
 - pre-commit
 - pydata-sphinx-theme
 - pytest

diff --git a/conda/recipes/kvikio/conda_build_config.yaml b/conda/recipes/kvikio/conda_build_config.yaml
@@ -12,3 +12,6 @@ sysroot_version:
 
 cmake_version:
  - ">=3.26.4"
+
+nvcomp_version:
+ - "=2.6.1"
diff --git a/conda/recipes/kvikio/meta.yaml b/conda/recipes/kvikio/meta.yaml
@@ -51,7 +51,8 @@ requirements:
  - setuptools
  - pip
  - cython >=0.29,<0.30
- - cudatoolkit {{ cuda_version }}.*
+ - cudatoolkit ={{ cuda_version }}
+ - nvcomp {{ nvcomp_version }}
  - scikit-build >=0.13.1
  - libkvikio {{ version }}
  run:

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -170,6 +170,19 @@ dependencies:
  # Fallback matrix for aarch64, which doesn't support libcufile.
  - matrix:
  packages:
+ - output_types: conda
+ matrices:
+ - matrix:
+ arch: x86_64
+ packages:
+ # Align nvcomp version with rapids-cmake
+ # TODO: not yet available for aarch64 CUDA 12
+ - &nvcomp nvcomp==2.6.1
+ - matrix:
+ arch: aarch64
+ cuda: "11.8"
+ packages:
+ - *nvcomp
  docs:
  common:
  - output_types: [conda, requirements]