Skip to content

Commit 6bb7c0b

Browse files
committed
add cuda-python[all] for passing through
1 parent 188370e commit 6bb7c0b

File tree

3 files changed

+9
-2
lines changed

3 files changed

+9
-2
lines changed

.github/workflows/test-wheel.yml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,4 +217,8 @@ jobs:
217217
218218
- name: Ensure cuda-python installable
219219
run: |
220-
pip install cuda_python*.whl
220+
if [[ "${{ inputs.local-ctk }}" == 1 ]]; then
221+
pip install cuda_python*.whl
222+
else
223+
pip install $(ls cuda_python*.whl)[all]
224+
fi

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)