Skip to content

Commit dd85925

Browse files
authored
Guard CuPy CTK extra for wheel metadata (#1469)
Contributes to rapidsai/build-planning#279 This keeps `cupy-cuda12x[ctk]` and `cupy-cuda13x[ctk]` in wheel metadata when CUDA wheels are used, while generating bare `cupy-cuda12x` and `cupy-cuda13x` metadata when `use_cuda_wheels=false`. Authors: - Bradley Dice (https://github.com/bdice) - Ramakrishna Prabhu (https://github.com/ramakrishnap-nv) Approvers: - James Lamb (https://github.com/jameslamb) URL: #1469
1 parent 97c51d1 commit dd85925

1 file changed

Lines changed: 21 additions & 1 deletion

File tree

dependencies.yaml

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -515,8 +515,28 @@ dependencies:
515515
# other packages with -cu{nn}x suffixes in this file.
516516
# All RAPIDS wheel builds (including in devcontainers) expect cupy to be suffixed.
517517
specific:
518-
- output_types: [requirements, pyproject]
518+
- output_types: requirements
519+
matrices:
520+
- matrix:
521+
cuda: "12.*"
522+
packages:
523+
- &cupy_cu12 cupy-cuda12x>=14.0.1,!=14.1.0
524+
# fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided
525+
- matrix:
526+
packages:
527+
- &cupy_cu13 cupy-cuda13x>=14.0.1,!=14.1.0
528+
- output_types: pyproject
519529
matrices:
530+
- matrix:
531+
cuda: "12.*"
532+
use_cuda_wheels: "false"
533+
packages:
534+
- *cupy_cu12
535+
- matrix:
536+
cuda: "13.*"
537+
use_cuda_wheels: "false"
538+
packages:
539+
- *cupy_cu13
520540
- matrix:
521541
cuda: "12.*"
522542
packages:

0 commit comments

Comments
 (0)