Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/test-wheel-linux.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/test-wheel-windows.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
6 changes: 3 additions & 3 deletions cuda_pathfinder/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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'",
]

Expand Down
4 changes: 2 additions & 2 deletions toolshed/collect_site_packages_dll_files.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -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 '..'
Expand Down
4 changes: 2 additions & 2 deletions toolshed/collect_site_packages_so_files.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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 ..
Expand Down
Loading