Skip to content

Commit

Permalink
Use nvcomp conda package. (#245)
Browse files Browse the repository at this point in the history
This PR uses conda-forge packages of `nvcomp` rather than fetching a tarball. This means that the nvcomp binary should not be shipped in the kvikio conda package, but is instead listed as a dependency. This will reduce kvikio's conda package size.

Resolves #242.

Authors:
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - https://github.com/jakirkham
  - Ray Douglass (https://github.com/raydouglass)

URL: #245
  • Loading branch information
bdice committed Jun 26, 2023
1 parent 4a4d289 commit 1ffe8b1
Show file tree
Hide file tree
Showing 5 changed files with 19 additions and 0 deletions.
1 change: 1 addition & 0 deletions conda/environments/all_cuda-118_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ dependencies:
- ninja
- numpy>=1.21
- nvcc_linux-64=11.8
- nvcomp==2.6.1
- packaging
- pre-commit
- pydata-sphinx-theme
Expand Down
1 change: 1 addition & 0 deletions conda/environments/all_cuda-120_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ dependencies:
- libcufile-dev
- ninja
- numpy>=1.21
- nvcomp==2.6.1
- packaging
- pre-commit
- pydata-sphinx-theme
Expand Down
3 changes: 3 additions & 0 deletions conda/recipes/kvikio/conda_build_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,6 @@ sysroot_version:

cmake_version:
- ">=3.26.4"

nvcomp_version:
- "=2.6.1"
1 change: 1 addition & 0 deletions conda/recipes/kvikio/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@ requirements:
- cudatoolkit
{% endif %}
- cuda-version ={{ cuda_version }}
- nvcomp {{ nvcomp_version }}
- scikit-build >=0.13.1
- libkvikio ={{ version }}
run:
Expand Down
13 changes: 13 additions & 0 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit 1ffe8b1

Please sign in to comment.