Merge pull request #391 from leofang/all_11

[Backport] Add `cuda-python[all]` for passing through
NVIDIA · Jan 14, 2025 · 8d4b5a2 · 8d4b5a2
2 parents 11b53ba + ace211b
commit 8d4b5a2
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml
@@ -321,7 +321,11 @@ jobs:
 
       - name: Ensure cuda-python installable
         run: |
-          pip install cuda_python*.whl
+          if [[ "${{ matrix.local-ctk }}" == 1 ]]; then
+            pip install cuda_python*.whl
+          else
+            pip install $(ls cuda_python*.whl)[all]
+          fi
 
   checks:
     name: Check job status

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}"],
+    },
 )