add cuda-python[all] for passing through

leofang · leofang · commit 6bb7c0bca962 · 2025-01-13T18:19:35.000Z
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}"],
+    },
 )

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ classifiers = [`
`32`	`32`	`"Environment :: GPU :: NVIDIA CUDA",`
`33`	`33`	`"Environment :: GPU :: NVIDIA CUDA :: 12",`
`34`	`34`	`]`
`35`		`-dynamic = ["version", "dependencies"]`
	`35`	`+dynamic = ["version", "dependencies", "optional-dependencies"]`
`36`	`36`
`37`	`37`	`[project.urls]`
`38`	`38`	`homepage = "https://nvidia.github.io/cuda-python/"`
Original file line number	Diff line number	Diff line change
`@@ -17,4 +17,7 @@`
`17`	`17`	`install_requires=[`
`18`	`18`	`f"cuda-bindings~={version}",`
`19`	`19`	`],`
	`20`	`+ extras_require={`
	`21`	`+ "all": [f"cuda-bindings[all]~={version}"],`
	`22`	`+ },`
`20`	`23`	`)`