Skip to content

Commit

Permalink
Remove upper bounds on cuda-python to allow 12.6.2 and 11.8.5 (#2517)
Browse files Browse the repository at this point in the history
Now that some upstream bugs have been fixed, we can allow cuda-python 12.6.2 and 11.8.5.

See NVIDIA/cuda-python#226 (comment) for more information.

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

Approvers:
  - James Lamb (https://github.com/jameslamb)
  - Corey J. Nolet (https://github.com/cjnolet)

URL: #2517
  • Loading branch information
bdice authored Dec 5, 2024
1 parent fc7818f commit 3ce5b6a
Show file tree
Hide file tree
Showing 8 changed files with 21 additions and 16 deletions.
3 changes: 2 additions & 1 deletion conda/environments/all_cuda-118_arch-aarch64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ dependencies:
- cmake>=3.26.4,!=3.30.0
- cuda-nvtx=11.8
- cuda-profiler-api=11.8.86
- cuda-python>=11.7.1,<12.0a0,<=11.8.3
- cuda-python>=11.7.1,<12.0a0
- cuda-version=11.8
- cudatoolkit
- cupy>=12.0.0
Expand Down Expand Up @@ -54,6 +54,7 @@ dependencies:
- scikit-build-core>=0.10.0
- scikit-learn
- scipy
- spdlog>=1.14.1,<1.15
- sphinx-copybutton
- sphinx-markdown-tables
- sysroot_linux-aarch64==2.17
Expand Down
3 changes: 2 additions & 1 deletion conda/environments/all_cuda-118_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ dependencies:
- cmake>=3.26.4,!=3.30.0
- cuda-nvtx=11.8
- cuda-profiler-api=11.8.86
- cuda-python>=11.7.1,<12.0a0,<=11.8.3
- cuda-python>=11.7.1,<12.0a0
- cuda-version=11.8
- cudatoolkit
- cupy>=12.0.0
Expand Down Expand Up @@ -54,6 +54,7 @@ dependencies:
- scikit-build-core>=0.10.0
- scikit-learn
- scipy
- spdlog>=1.14.1,<1.15
- sphinx-copybutton
- sphinx-markdown-tables
- sysroot_linux-64==2.17
Expand Down
3 changes: 2 additions & 1 deletion conda/environments/all_cuda-125_arch-aarch64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ dependencies:
- cuda-nvcc
- cuda-nvtx-dev
- cuda-profiler-api
- cuda-python>=12.0,<13.0a0,<=12.6.0
- cuda-python>=12.0,<13.0a0
- cuda-version=12.5
- cupy>=12.0.0
- cxx-compiler
Expand Down Expand Up @@ -50,6 +50,7 @@ dependencies:
- scikit-build-core>=0.10.0
- scikit-learn
- scipy
- spdlog>=1.14.1,<1.15
- sphinx-copybutton
- sphinx-markdown-tables
- sysroot_linux-aarch64==2.17
Expand Down
3 changes: 2 additions & 1 deletion conda/environments/all_cuda-125_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ dependencies:
- cuda-nvcc
- cuda-nvtx-dev
- cuda-profiler-api
- cuda-python>=12.0,<13.0a0,<=12.6.0
- cuda-python>=12.0,<13.0a0
- cuda-version=12.5
- cupy>=12.0.0
- cxx-compiler
Expand Down Expand Up @@ -50,6 +50,7 @@ dependencies:
- scikit-build-core>=0.10.0
- scikit-learn
- scipy
- spdlog>=1.14.1,<1.15
- sphinx-copybutton
- sphinx-markdown-tables
- sysroot_linux-64==2.17
Expand Down
8 changes: 4 additions & 4 deletions conda/recipes/pylibraft/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,10 @@ requirements:
- {{ stdlib("c") }}
host:
{% if cuda_major == "11" %}
- cuda-python >=11.7.1,<12.0a0,<=11.8.3
- cuda-python >=11.7.1,<12.0a0
- cudatoolkit
{% else %}
- cuda-python >=12.0,<13.0a0,<=12.6.0
- cuda-python >=12.0,<13.0a0
- cuda-cudart-dev
{% endif %}
- cuda-version ={{ cuda_version }}
Expand All @@ -61,10 +61,10 @@ requirements:
- {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
{% if cuda_major == "11" %}
- cudatoolkit
- cuda-python >=11.7.1,<12.0a0,<=11.8.3
- cuda-python >=11.7.1,<12.0a0
{% else %}
- cuda-cudart
- cuda-python >=12.0,<13.0a0,<=12.6.0
- cuda-python >=12.0,<13.0a0
{% endif %}
- libraft {{ version }}
- libraft-headers {{ version }}
Expand Down
8 changes: 4 additions & 4 deletions conda/recipes/raft-dask/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,10 @@ requirements:
- {{ stdlib("c") }}
host:
{% if cuda_major == "11" %}
- cuda-python >=11.7.1,<12.0a0,<=11.8.3
- cuda-python >=11.7.1,<12.0a0
- cudatoolkit
{% else %}
- cuda-python >=12.0,<13.0a0,<=12.6.0
- cuda-python >=12.0,<13.0a0
- cuda-cudart-dev
{% endif %}
- cuda-version ={{ cuda_version }}
Expand All @@ -62,10 +62,10 @@ requirements:
run:
{% if cuda_major == "11" %}
- cudatoolkit
- cuda-python >=11.7.1,<12.0a0,<=11.8.3
- cuda-python >=11.7.1,<12.0a0
{% else %}
- cuda-cudart
- cuda-python >=12.0,<13.0a0,<=12.6.0
- cuda-python >=12.0,<13.0a0
{% endif %}
- {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
- dask-cuda ={{ minor_version }}
Expand Down
2 changes: 1 addition & 1 deletion cpp/cmake/thirdparty/get_rmm.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
function(find_and_configure_rmm)
include(${rapids-cmake-dir}/cpm/rmm.cmake)
rapids_cpm_rmm(BUILD_EXPORT_SET raft-exports
INSTALL_EXPORT_SET raft-exports)
INSTALL_EXPORT_SET raft-exports)
endfunction()

find_and_configure_rmm()
7 changes: 4 additions & 3 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -143,8 +143,9 @@ dependencies:
packages:
- c-compiler
- cxx-compiler
- nccl>=2.19
- libucxx==0.42.*,>=0.0.0a0
- nccl>=2.19
- spdlog>=1.14.1,<1.15
specific:
- output_types: conda
matrices:
Expand Down Expand Up @@ -196,11 +197,11 @@ dependencies:
- matrix:
cuda: "12.*"
packages:
- &cuda_python12 cuda-python>=12.0,<13.0a0,<=12.6.0
- &cuda_python12 cuda-python>=12.0,<13.0a0
- matrix:
cuda: "11.*"
packages:
- &cuda_python11 cuda-python>=11.7.1,<12.0a0,<=11.8.3
- &cuda_python11 cuda-python>=11.7.1,<12.0a0
- matrix:
packages:
- &cuda_python cuda-python
Expand Down

0 comments on commit 3ce5b6a

Please sign in to comment.