Skip to content

Commit 8d4b5a2

Browse files
authored
Merge pull request #391 from leofang/all_11
[Backport] Add `cuda-python[all]` for passing through
2 parents 11b53ba + ace211b commit 8d4b5a2

3 files changed

Lines changed: 9 additions & 2 deletions

File tree

.github/workflows/build-and-test.yml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -321,7 +321,11 @@ jobs:
321321
322322
- name: Ensure cuda-python installable
323323
run: |
324-
pip install cuda_python*.whl
324+
if [[ "${{ matrix.local-ctk }}" == 1 ]]; then
325+
pip install cuda_python*.whl
326+
else
327+
pip install $(ls cuda_python*.whl)[all]
328+
fi
325329
326330
checks:
327331
name: Check job status

cuda_python/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ classifiers = [
3232
"Environment :: GPU :: NVIDIA CUDA",
3333
"Environment :: GPU :: NVIDIA CUDA :: 12",
3434
]
35-
dynamic = ["version", "dependencies"]
35+
dynamic = ["version", "dependencies", "optional-dependencies"]
3636

3737
[project.urls]
3838
homepage = "https://nvidia.github.io/cuda-python/"

cuda_python/setup.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,7 @@
1717
install_requires=[
1818
f"cuda-bindings~={version}",
1919
],
20+
extras_require={
21+
"all": [f"cuda-bindings[all]~={version}"],
22+
},
2023
)

0 commit comments

Comments
 (0)