Skip to content

Commit 12c136d

Browse files
committed
Fix CuPy constraints for wheel tests
1 parent e1e43ee commit 12c136d

1 file changed

Lines changed: 11 additions & 1 deletion

File tree

dependencies.yaml

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -450,7 +450,17 @@ dependencies:
450450
# other packages with -cu{nn}x suffixes in this file.
451451
# All RAPIDS wheel builds (including in devcontainers) expect cupy to be suffixed.
452452
specific:
453-
- output_types: [requirements, pyproject]
453+
- output_types: requirements
454+
matrices:
455+
- matrix:
456+
cuda: "12.*"
457+
packages:
458+
- cupy-cuda12x>=14.0.1,!=14.1.0
459+
# fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided
460+
- matrix:
461+
packages:
462+
- cupy-cuda13x>=14.0.1,!=14.1.0
463+
- output_types: pyproject
454464
matrices:
455465
- matrix:
456466
cuda: "12.*"

0 commit comments

Comments
 (0)