From ee6d296ecc604b43a484885d188b8acd1da322ca Mon Sep 17 00:00:00 2001 From: Leo Fang Date: Mon, 13 Jan 2025 18:22:12 +0000 Subject: [PATCH 1/2] add cuda-python[all] for passing through --- .github/workflows/build-and-test.yml | 6 +++++- cuda_python/pyproject.toml | 2 +- cuda_python/setup.py | 3 +++ 3 files changed, 9 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml index a809567fd..deb91f4be 100644 --- 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 [[ "${{ inputs.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 index 93c68abcf..9566bcf92 100644 --- 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 index 3c422f91b..7cddd8fd1 100644 --- 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}"], + }, ) From ace211b65c9eeb506df90e29b81012177fc7248b Mon Sep 17 00:00:00 2001 From: Leo Fang Date: Mon, 13 Jan 2025 15:45:51 -0500 Subject: [PATCH 2/2] fix copy/pasta --- .github/workflows/build-and-test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml index deb91f4be..b8eb7bb23 100644 --- a/.github/workflows/build-and-test.yml +++ b/.github/workflows/build-and-test.yml @@ -321,7 +321,7 @@ jobs: - name: Ensure cuda-python installable run: | - if [[ "${{ inputs.local-ctk }}" == 1 ]]; then + if [[ "${{ matrix.local-ctk }}" == 1 ]]; then pip install cuda_python*.whl else pip install $(ls cuda_python*.whl)[all]