diff --git a/.github/workflows/test-wheel-linux.yml b/.github/workflows/test-wheel-linux.yml index ce8f6d540e..a6d9fc51c6 100644 --- a/.github/workflows/test-wheel-linux.yml +++ b/.github/workflows/test-wheel-linux.yml @@ -283,7 +283,7 @@ jobs: run: | set -euo pipefail pushd cuda_pathfinder - pip install --only-binary=:all: -v ".[nvidia_wheels_cu${TEST_CUDA_MAJOR},nvidia_wheels_host]" + pip install --only-binary=:all: -v ".[test_nvidia_wheels_cu${TEST_CUDA_MAJOR},test_nvidia_wheels_host]" pip list popd diff --git a/.github/workflows/test-wheel-windows.yml b/.github/workflows/test-wheel-windows.yml index 6fe2270c6d..573ec6d9a9 100644 --- a/.github/workflows/test-wheel-windows.yml +++ b/.github/workflows/test-wheel-windows.yml @@ -256,7 +256,7 @@ jobs: shell: bash --noprofile --norc -xeuo pipefail {0} run: | pushd cuda_pathfinder - pip install --only-binary=:all: -v ".[nvidia_wheels_cu${TEST_CUDA_MAJOR},nvidia_wheels_host]" + pip install --only-binary=:all: -v ".[test_nvidia_wheels_cu${TEST_CUDA_MAJOR},test_nvidia_wheels_host]" pip list popd diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index bffb42a828..96ad7fb6aa 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -14,7 +14,7 @@ dependencies = [] test = [ "pytest>=6.2.4", ] -nvidia_wheels_cu12 = [ +test_nvidia_wheels_cu12 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg]==12.*", "cuda-toolkit[cufile]==12.*; sys_platform != 'win32'", "nvidia-cudss-cu12", @@ -23,13 +23,13 @@ nvidia_wheels_cu12 = [ "nvidia-nccl-cu12; sys_platform != 'win32'", "nvidia-nvshmem-cu12; sys_platform != 'win32'", ] -nvidia_wheels_cu13 = [ +test_nvidia_wheels_cu13 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,nvvm]==13.*", "cuda-toolkit[cufile]==13.*; sys_platform != 'win32'", "nvidia-nccl-cu13; sys_platform != 'win32'", "nvidia-nvshmem-cu13; sys_platform != 'win32'", ] -nvidia_wheels_host = [ +test_nvidia_wheels_host = [ "nvpl-fft; platform_system == 'Linux' and platform_machine == 'aarch64'", ] diff --git a/toolshed/collect_site_packages_dll_files.ps1 b/toolshed/collect_site_packages_dll_files.ps1 index 9f1ccce93c..3a9954ba84 100644 --- a/toolshed/collect_site_packages_dll_files.ps1 +++ b/toolshed/collect_site_packages_dll_files.ps1 @@ -23,11 +23,11 @@ function Fresh-Venv { Set-Location -Path 'cuda_pathfinder' Fresh-Venv -Path '..\TmpCp12Venv' -pip install --only-binary=:all: -e '.[test,nvidia_wheels_cu12,nvidia_wheels_host]' +pip install --only-binary=:all: -e '.[test,test_nvidia_wheels_cu12,test_nvidia_wheels_host]' deactivate Fresh-Venv -Path '..\TmpCp13Venv' -pip install --only-binary=:all: -e '.[test,nvidia_wheels_cu13,nvidia_wheels_host]' +pip install --only-binary=:all: -e '.[test,test_nvidia_wheels_cu13,test_nvidia_wheels_host]' deactivate Set-Location -Path '..' diff --git a/toolshed/collect_site_packages_so_files.sh b/toolshed/collect_site_packages_so_files.sh index 000bdb64ce..48a6e7c779 100755 --- a/toolshed/collect_site_packages_so_files.sh +++ b/toolshed/collect_site_packages_so_files.sh @@ -17,12 +17,12 @@ fresh_venv() { cd cuda_pathfinder/ fresh_venv ../TmpCp12Venv set -x -pip install --only-binary=:all: -e .[test,nvidia_wheels_cu12,nvidia_wheels_host] +pip install --only-binary=:all: -e .[test,test_nvidia_wheels_cu12,test_nvidia_wheels_host] set +x deactivate fresh_venv ../TmpCp13Venv set -x -pip install --only-binary=:all: -e .[test,nvidia_wheels_cu13,nvidia_wheels_host] +pip install --only-binary=:all: -e .[test,test_nvidia_wheels_cu13,test_nvidia_wheels_host] set +x deactivate cd ..