move cuda_core test requirements to pyproject.toml optional dependencies as well

Signed-off-by: Keith Kraus <keith.j.kraus@gmail.com>

Author: kkraus14

.github/workflows/test-wheel-linux.yml

@@ -318,13 +318,12 @@ jobs:
           pushd "${CUDA_BINDINGS_ARTIFACTS_DIR}"
           if [[ "${{ matrix.LOCAL_CTK }}" == 1 ]]; then
             ls $CUDA_PATH
-            pip install *.whl
+            pip install *.whl[test]
           else
-            pip install $(ls *.whl)[all]
+            pip install $(ls *.whl)[all,test]
           fi
           popd
 
-          pip install $(ls *.whl)[test]
           pushd ./cuda_bindings
           ${SANITIZER_CMD} pytest -rxXs -v tests/
 
@@ -356,11 +355,10 @@ jobs:
           fi
           TEST_CUDA_MAJOR="$(cut -d '.' -f 1 <<< ${{ matrix.CUDA_VER }})"
           pushd "${CUDA_CORE_ARTIFACTS_DIR}"
-          pip install $(ls *.whl)["cu${TEST_CUDA_MAJOR}"]
+          pip install $(ls *.whl)["cu${TEST_CUDA_MAJOR}","test-cu${TEST_CUDA_MAJOR}"]
           popd
 
           pushd ./cuda_core
-          pip install -r "tests/requirements-cu${TEST_CUDA_MAJOR}.txt"
           ${SANITIZER_CMD} pytest -rxXs -v tests/
 
           # It is a bit convoluted to run the Cython tests against CTK wheels,

.github/workflows/test-wheel-windows.yml

@@ -246,13 +246,12 @@ jobs:
           if ('${{ matrix.LOCAL_CTK }}' -eq '1') {
             Get-ChildItem $env:CUDA_PATH
             echo $PATH
-            pip install (Get-ChildItem -Filter *.whl).FullName
+            pip install "$((Get-ChildItem -Filter *.whl).FullName)[test]"
           } else {
-            pip install "$((Get-ChildItem -Filter *.whl).FullName)[all]"
+            pip install "$((Get-ChildItem -Filter *.whl).FullName)[all,test]"
           }
           Pop-Location
 
-          pip install "$((Get-ChildItem -Filter *.whl).FullName)[test]"
           Push-Location ./cuda_bindings
           pytest -rxXs -v tests/
           # skip Cython tests for now (NVIDIA/cuda-python#466)
@@ -273,11 +272,10 @@ jobs:
           }
           $TEST_CUDA_MAJOR = '${{ matrix.CUDA_VER }}' -split '\.' | Select-Object -First 1
           Push-Location $env:CUDA_CORE_ARTIFACTS_DIR
-          pip install "$((Get-ChildItem -Filter *.whl).FullName)[cu${TEST_CUDA_MAJOR}]"
+          pip install "$((Get-ChildItem -Filter *.whl).FullName)[cu${TEST_CUDA_MAJOR},test-cu${TEST_CUDA_MAJOR}]"
           Pop-Location
 
           Push-Location ./cuda_core
-          pip install -r "tests/requirements-cu${TEST_CUDA_MAJOR}.txt"
           pytest -rxXs -v tests/
           Pop-Location
 

cuda_bindings/pyproject.toml

@@ -37,7 +37,7 @@ all = [
     "nvidia-nvjitlink-cu12>=12.3",
 ]
 
-tests = [
+test = [
     "numpy>=1.21.1",
     "pytest>=6.2.4",
     "pytest-benchmark>=3.4.1",

cuda_core/pyproject.toml

@@ -48,6 +48,9 @@ dependencies = [
 [project.optional-dependencies]
 cu11 = ["cuda-bindings==11.8.*"]
 cu12 = ["cuda-bindings==12.*"]
+test = ["pytest>=6.2.4"]
+test-cu11 = ["cuda-core[test]", "cupy-cuda11x", "nvidia-cuda-runtime-cu11"]  # runtime headers needed by CuPy
+test-cu12 = ["cuda-core[test]", "cupy-cuda12x", "nvidia-cuda-runtime-cu12"]  # runtime headers needed by CuPy
 
 [project.urls]
 homepage = "https://nvidia.github.io/cuda-python/"