add cuda-python[all] for passing through

NVIDIA · Jan 13, 2025 · 6bb7c0b · 6bb7c0b
1 parent 188370e
commit 6bb7c0b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/.github/workflows/test-wheel.yml b/.github/workflows/test-wheel.yml
@@ -217,4 +217,8 @@ jobs:
 
       - name: Ensure cuda-python installable
         run: |
-          pip install cuda_python*.whl
+          if [[ "${{ inputs.local-ctk }}" == 1 ]]; then
+            pip install cuda_python*.whl
+          else
+            pip install $(ls cuda_python*.whl)[all]
+          fi
diff --git a/cuda_python/pyproject.toml b/cuda_python/pyproject.toml
@@ -32,7 +32,7 @@ classifiers = [
     "Environment :: GPU :: NVIDIA CUDA",
     "Environment :: GPU :: NVIDIA CUDA :: 12",
 ]
-dynamic = ["version", "dependencies"]
+dynamic = ["version", "dependencies", "optional-dependencies"]
 
 [project.urls]
 homepage = "https://nvidia.github.io/cuda-python/"

diff --git a/cuda_python/setup.py b/cuda_python/setup.py
@@ -17,4 +17,7 @@
     install_requires=[
         f"cuda-bindings~={version}",
     ],
+    extras_require={
+        "all": [f"cuda-bindings[all]~={version}"],
+    },
 )