Skip to content

Commit

Permalink
Don't modify cuda-python
Browse files Browse the repository at this point in the history
  • Loading branch information
KyleFromNVIDIA committed Mar 14, 2024
1 parent 72536c3 commit 9e0eee7
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 8 deletions.
4 changes: 0 additions & 4 deletions rapids_build_backend/impls.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,10 +136,6 @@ def _add_cuda_suffix(req, cuda_suffix, cuda_major):
req = Requirement(req)
if req.name == "cupy" and cuda_major is not None:
req.name += f"-cuda{cuda_major}x"
elif req.name == "cuda-python" and cuda_major is not None:
req.specifier &= SpecifierSet(
f">={cuda_major}.0.0,<{int(cuda_major) + 1}.0.0.dev0"
)
elif req.name in _VERSIONED_RAPIDS_WHEELS:
req.name += cuda_suffix

Expand Down
4 changes: 0 additions & 4 deletions tests/test_impls.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,10 +55,6 @@ def check_initial_contents(filename):
("cupy", "", "11", "cupy-cuda11x"),
("cupy", "", "12", "cupy-cuda12x"),
("cupy", "", None, "cupy"),
("cuda-python", "", "11", "cuda-python<12.0.0.dev0,>=11.0.0"),
("cuda-python>=11.7.1", "", "12", "cuda-python<13.0.0.dev0,>=11.7.1,>=12.0.0"),
("cuda-python", "", None, "cuda-python"),
("cuda-python>=11.7.1", "", None, "cuda-python>=11.7.1"),
*((name, "-cu11", "11", f"{name}-cu11") for name in _VERSIONED_RAPIDS_WHEELS),
*((name, "-cu12", "12", f"{name}-cu12") for name in _VERSIONED_RAPIDS_WHEELS),
*((name, "", None, name) for name in _VERSIONED_RAPIDS_WHEELS),
Expand Down

0 comments on commit 9e0eee7

Please sign in to comment.