Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support CUDA 12.0 for pip wheels #1489

Merged
merged 20 commits into from
May 5, 2023
Merged

Support CUDA 12.0 for pip wheels #1489

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
00734e9
Update cupy dependency
vyasr May 3, 2023
8638daa
update pip dependencies to cuda 12
divyegala May 4, 2023
83f0fcb
merging upstream
divyegala May 4, 2023
4f91bfc
replace occurrences of sha wheel target branch
divyegala May 4, 2023
b60b549
updates
divyegala May 5, 2023
e8674b4
Merge remote-tracking branch 'upstream/branch-23.06' into cuda-120-pip
divyegala May 5, 2023
1f13356
remove trailing space in cuda-python pinnings
divyegala May 5, 2023
c7276fe
try evaluating pip cuda version later
divyegala May 5, 2023
87a4eec
main fork branch for shared-action-workflows
divyegala May 5, 2023
18c0b40
try to fix bad substitution
divyegala May 5, 2023
e84c065
trying different things
divyegala May 5, 2023
07647c1
try single curly braces
divyegala May 5, 2023
03e2559
temporarily have raft-dask build wheels only depend on pylibraft buil…
divyegala May 5, 2023
fb3ac46
bump
divyegala May 5, 2023
cce058a
bump
divyegala May 5, 2023
38f4b94
different var name
divyegala May 5, 2023
90b2efe
bump
divyegala May 5, 2023
e279fbc
Revert "temporarily have raft-dask build wheels only depend on pylibr…
divyegala May 5, 2023
200f41c
Merge remote-tracking branch 'upstream/branch-23.06' into cuda-120-pip
divyegala May 5, 2023
e262725
update sha branch back to cuda-120-pip
divyegala May 5, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions ci/release/apply_wheel_modifications.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,3 +18,8 @@ sed -i "s/rmm/rmm${CUDA_SUFFIX}/g" python/pylibraft/pyproject.toml
sed -i "s/^name = \"raft-dask\"/name = \"raft-dask${CUDA_SUFFIX}\"/g" python/raft-dask/pyproject.toml
sed -i "s/pylibraft/pylibraft${CUDA_SUFFIX}/g" python/raft-dask/pyproject.toml
sed -i "s/ucx-py/ucx-py${CUDA_SUFFIX}/g" python/raft-dask/pyproject.toml

if [[ $CUDA_SUFFIX == "-cu12" ]]; then
sed -i "/cuda-python/ s/>=11.7.1,<12.0/>=12.0,<13.0/g" python/pylibraft/pyproject.toml
sed -i "/cupy-cuda11x/ s/cupy-cuda11x/cupy-cuda12x/g" python/pylibraft/pyproject.toml
fi
11 changes: 9 additions & 2 deletions dependencies.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -291,6 +291,13 @@ dependencies:
- output_types: conda
packages:
- cupy>=12.0.0
specific:
- output_types: pyproject
packages:
- cupy-cuda11x>=12.0.0
matrices:
- matrix:
cuda: "12.0"
packages:
- cupy-cuda12x>=12.0.0
- matrix: # All CUDA 11 versions
packages:
- cupy-cuda11x>=12.0.0