Skip to content

Expand wheel build to arm64 and Windows #2387

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
clin1234 wants to merge 10 commits into
base: main
Choose a base branch
from
Draft

Expand wheel build to arm64 and Windows #2387

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
5b541a5
Expand test matrix for Python 3.15
Jul 18, 2026
79a0eed
Don't exclude 3.15 wheels from downloaded artifacts
Jul 18, 2026
07e037b
Attempt to build 3.15 wheels on all platforms
Jul 18, 2026
6189daf
Fix pytest param
Jul 18, 2026
79a6760
scipy not supported on 3.15 yet
Jul 18, 2026
60ee552
Restrict Python 3.15 configurations to nightly wheel builds
clin1234 Jul 20, 2026
b32546e
Apply suggestion from @leofang
clin1234 Jul 20, 2026
ecddeb6
Add Python 3.15 configurations to test matrix
clin1234 Jul 20, 2026
44af69f
Revert TOML changes
clin1234 Jul 20, 2026
3c7bf6a
lol
clin1234 Jul 20, 2026
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
8 changes: 5 additions & 3 deletions ci/test-matrix.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,6 @@ linux:
- { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '12.9.1', LOCAL_CTK: '1', GPU: 't4', GPU_COUNT: '1', DRIVER: 'latest' }
- { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '13.0.2', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest' }
- { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest' }
- { ARCH: 'amd64', PY_VER: '3.15', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest' }
- { ARCH: 'amd64', PY_VER: '3.15t', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest' }
Comment thread
clin1234 marked this conversation as resolved.
# linux-aarch64
- { ARCH: 'arm64', PY_VER: '3.10', CUDA_VER: '12.9.1', LOCAL_CTK: '1', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest' }
- { ARCH: 'arm64', PY_VER: '3.10', CUDA_VER: '13.0.2', LOCAL_CTK: '0', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest' }
Expand Down Expand Up @@ -106,6 +104,9 @@ linux:
- { ARCH: 'arm64', PY_VER: '3.14', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'gb300', GPU_COUNT: '1', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'arm64', PY_VER: '3.14', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '2', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'arm64', PY_VER: '3.14t', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '2', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'arm64', PY_VER: '3.15', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'gb300', GPU_COUNT: '1', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'arm64', PY_VER: '3.15', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '2', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'arm64', PY_VER: '3.15t', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '2', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
Comment thread
clin1234 marked this conversation as resolved.
Outdated

windows:
pull-request:
Expand All @@ -127,7 +128,6 @@ windows:
- { ARCH: 'amd64', PY_VER: '3.14', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM' }
- { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '12.9.1', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'TCC' }
- { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '13.0.2', LOCAL_CTK: '0', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM' }
- { ARCH: 'amd64', PY_VER: '3.14t', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM' }
Comment thread
clin1234 marked this conversation as resolved.
# special runners
- { ARCH: 'amd64', PY_VER: '3.14', CUDA_VER: '13.3.0', LOCAL_CTK: '1', GPU: 't4', GPU_COUNT: '2', DRIVER: 'latest', DRIVER_MODE: 'TCC' }
- { ARCH: 'amd64', PY_VER: '3.14', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'h100', GPU_COUNT: '2', DRIVER: 'latest', DRIVER_MODE: 'MCDM' }
Expand All @@ -145,3 +145,5 @@ windows:
- { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'rtxpro6000', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM', ENV: { MODE: 'nightly-numba-cuda-mlir' } }
# nightly-cuda-core (released cuda-core from PyPI against main pathfinder/bindings)
- { ARCH: 'amd64', PY_VER: '3.14', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM', ENV: { MODE: 'nightly-cuda-core' } }
- { ARCH: 'amd64', PY_VER: '3.15', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM', ENV: { MODE: 'nightly-cuda-core' } }
- { ARCH: 'amd64', PY_VER: '3.15t', CUDA_VER: '13.3.0', LOCAL_CTK: '0', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM', ENV: { MODE: 'nightly-cuda-core' } }
Comment thread
clin1234 marked this conversation as resolved.
Outdated
7 changes: 0 additions & 7 deletions ci/tools/download-wheels
View file
Open in desktop
Comment thread
clin1234 marked this conversation as resolved.
Original file line number Diff line number Diff line change
Expand Up @@ -56,13 +56,6 @@ do
continue
fi

# TODO: remove the Python 3.15 exclusion once 3.15 is officially supported
# exclude pre-release Python (3.15) wheels from releasing
if [[ "${p}" == *python315* ]]; then
echo "Skipping pre-release Python artifact: $p"
continue
fi

# If we're not downloading "all", only process matching component
if [[ "$COMPONENT" != "all" && "$p" != ${COMPONENT}* ]]; then
continue
Expand Down
17 changes: 8 additions & 9 deletions cuda_bindings/pyproject.toml
View file
Open in desktop
Comment thread
clin1234 marked this conversation as resolved.
Original file line number Diff line number Diff line change
Expand Up @@ -44,16 +44,15 @@ all = [

[dependency-groups]
test = [
"cython>=3.2,<3.3",
"cython>=3.2",
"setuptools>=80.0.0",
# TODO: remove the Python 3.15 guard once 3.15 is officially supported
"matplotlib>=3.5.0,<=3.10.9; python_version < '3.15'",
"numpy>=1.21.1,<=2.5.0",
"pytest==9.1.0",
"pytest-benchmark==5.2.3",
"pytest-repeat==0.9.4",
"pytest-randomly==4.1.0",
"pyglet==2.1.14",
"matplotlib>=3.5.0",
"numpy>=1.21.1",
"pytest~=9.1.0",
"pytest-benchmark~=5.2.3",
"pytest-repeat~=0.9.4",
"pytest-randomly~=4.1.0",
"pyglet~=2.1.14",
]

[project.urls]
Expand Down
19 changes: 9 additions & 10 deletions cuda_core/pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -61,16 +61,15 @@ cu13 = ["cuda-bindings[all]==13.*", "cuda-toolkit==13.*"]
test = [
"cython>=3.2,<3.3",
"setuptools>=80",
"pytest==9.1.0",
"pytest-benchmark==5.2.3",
"pytest-randomly==4.1.0",
"pytest-repeat==0.9.4",
"pytest-rerunfailures==16.3",
"pytest-timeout==2.4.0",
"cloudpickle==3.1.2",
"psutil==7.2.2",
# TODO: remove the Python 3.15 guard once 3.15 is officially supported
"cffi==2.0.0; python_version < '3.15'",
Comment thread
clin1234 marked this conversation as resolved.
"pytest~=9.1.0",
"pytest-benchmark~=5.2.3",
"pytest-randomly~=4.1.0",
"pytest-repeat~=0.9.4",
"pytest-rerunfailures~=16.3",
"pytest-timeout~=2.4.0",
"cloudpickle~=3.1.2",
"psutil~=7.2.2",
Comment thread
clin1234 marked this conversation as resolved.
Outdated
"cffi~=2.0.0",
]
ml-dtypes = ["ml-dtypes>=0.5.4,<0.6.0"]
test-cu12 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda12x; python_version < '3.14'", "cuda-toolkit[cudart]==12.*"] # runtime headers needed by CuPy
Expand Down
3 changes: 1 addition & 2 deletions cuda_pathfinder/pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,7 @@ cu13 = [
"nvidia-nvshmem-cu13<3.7; sys_platform != 'win32'",
]
host = [
# TODO: remove the Python 3.15 guard once 3.15 is officially supported
"nvidia-cutlass; python_version < '3.15'",
"nvidia-cutlass",

@leofang leofang Jul 20, 2026

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Did you check that cutlass already has 3.15 support?

"nvpl-fft; platform_system == 'Linux' and platform_machine == 'aarch64'",
]

Expand Down