Skip to content

Commit efbb6f4

Browse files
committed
Fix CuPy constraints for wheel tests
1 parent 8e0a791 commit efbb6f4

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
@@ -407,7 +407,17 @@ dependencies:
407407
- psutil # Used for timeout_with_stack.py
408408
- pytest
409409
specific:
410-
- output_types: [requirements, pyproject]
410+
- output_types: requirements
411+
matrices:
412+
- matrix:
413+
cuda: "12.*"
414+
packages:
415+
- cupy-cuda12x>=14.0.1,!=14.1.0
416+
# fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided
417+
- matrix:
418+
packages:
419+
- cupy-cuda13x>=14.0.1,!=14.1.0
420+
- output_types: pyproject
411421
matrices:
412422
- matrix:
413423
cuda: "12.*"

0 commit comments

Comments
 (0)