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 . --group "test_nvidia_wheels_cu${TEST_CUDA_MAJOR}" --group test_nvidia_wheels_host
pip install --only-binary=:all: -v . --group "test-cu${TEST_CUDA_MAJOR}"
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 . --group "test_nvidia_wheels_cu${TEST_CUDA_MAJOR}" --group test_nvidia_wheels_host
pip install --only-binary=:all: -v . --group "test-cu${TEST_CUDA_MAJOR}"
pip list
popd

Expand Down
19 changes: 16 additions & 3 deletions cuda_pathfinder/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,8 @@ dependencies = []
test = [
"pytest>=6.2.4",
]
test_nvidia_wheels_cu12 = [
# Internal organization of test dependencies.
cu12 = [
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl]==12.*",
"cuda-toolkit[cufile]==12.*; sys_platform != 'win32'",
"nvidia-cublasmp-cu12; sys_platform != 'win32'",
Expand All @@ -24,18 +25,30 @@ test_nvidia_wheels_cu12 = [
"nvidia-nccl-cu12; sys_platform != 'win32'",
"nvidia-nvshmem-cu12; sys_platform != 'win32'",
]
test_nvidia_wheels_cu13 = [
cu13 = [
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,nvvm]==13.*",
"cuda-toolkit[cufile]==13.*; sys_platform != 'win32'",
"nvidia-cublasmp-cu13; sys_platform != 'win32'",
"nvidia-cudss-cu13",
"nvidia-nccl-cu13; sys_platform != 'win32'",
"nvidia-nvshmem-cu13; sys_platform != 'win32'",
]
test_nvidia_wheels_host = [
host = [
"nvpl-fft; platform_system == 'Linux' and platform_machine == 'aarch64'",
]

# User-facing dependencies, e.g. pip install --group test-cu13
test-cu12 = [
{ include-group = "test" },
{ include-group = "cu12" },
{ include-group = "host" },
]
test-cu13 = [
{ include-group = "test" },
{ include-group = "cu13" },
{ include-group = "host" },
]

[project.urls]
Repository = "https://github.com/NVIDIA/cuda-python"
Documentation = "https://nvidia.github.io/cuda-python/"
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,test_nvidia_wheels_cu12,test_nvidia_wheels_host]'
pip install --only-binary=:all: -e . --group test-cu12
deactivate

Fresh-Venv -Path '..\TmpCp13Venv'
pip install --only-binary=:all: -e '.[test,test_nvidia_wheels_cu13,test_nvidia_wheels_host]'
pip install --only-binary=:all: -e . --group test-cu13
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 . --group test --group test_nvidia_wheels_cu12 --group test_nvidia_wheels_host
pip install --only-binary=:all: -e . --group test-cu12
set +x
deactivate
fresh_venv ../TmpCp13Venv
set -x
pip install --only-binary=:all: -e . --group test --group test_nvidia_wheels_cu13 --group test_nvidia_wheels_host
pip install --only-binary=:all: -e . --group test-cu13
set +x
deactivate
cd ..
Expand Down
Loading