diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index b60468ee15..88a96a13b7 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -24,6 +24,7 @@ dependencies: - ninja - numpy>=1.21 - nvcc_linux-64=11.8 +- nvcomp==2.6.1 - packaging - pre-commit - pydata-sphinx-theme diff --git a/conda/environments/all_cuda-120_arch-x86_64.yaml b/conda/environments/all_cuda-120_arch-x86_64.yaml index 08982b7666..2a5ee53ec9 100644 --- a/conda/environments/all_cuda-120_arch-x86_64.yaml +++ b/conda/environments/all_cuda-120_arch-x86_64.yaml @@ -23,6 +23,7 @@ dependencies: - libcufile-dev - ninja - numpy>=1.21 +- nvcomp==2.6.1 - packaging - pre-commit - pydata-sphinx-theme diff --git a/conda/recipes/kvikio/conda_build_config.yaml b/conda/recipes/kvikio/conda_build_config.yaml index e28b98da7f..7bce9d8853 100644 --- a/conda/recipes/kvikio/conda_build_config.yaml +++ b/conda/recipes/kvikio/conda_build_config.yaml @@ -15,3 +15,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 index 0e17176bd7..2f32ed7326 100644 --- a/conda/recipes/kvikio/meta.yaml +++ b/conda/recipes/kvikio/meta.yaml @@ -62,6 +62,7 @@ requirements: - cudatoolkit {% endif %} - cuda-version ={{ cuda_version }} + - nvcomp {{ nvcomp_version }} - scikit-build >=0.13.1 - libkvikio ={{ version }} run: diff --git a/dependencies.yaml b/dependencies.yaml index 4353ab5702..26741d6a1f 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -204,6 +204,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]