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 11 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 all commits
Commits
Show all changes
11 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
249d73c
Merge branch 'main' into 315-support
clin1234 Oct 11, 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
4 changes: 4 additions & 0 deletions ci/test-matrix.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,10 @@ linux:
- { ARCH: 'arm64', PY_VER: '3.14t', CUDA_VER: '13.4.2', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '2', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
# amd64 with h200 (nightly-only, as there is only one server with h200s)
- { ARCH: 'amd64', PY_VER: '3.14', CUDA_VER: '13.4.2', LOCAL_CTK: '0', GPU: 'h200', GPU_COUNT: '2', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'arm64', PY_VER: '3.15', CUDA_VER: '13.4.2', LOCAL_CTK: '1', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'arm64', PY_VER: '3.15t', CUDA_VER: '13.4.2', LOCAL_CTK: '1', GPU: 'l4', GPU_COUNT: '1', DRIVER: 'latest', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'amd64', PY_VER: '3.15', CUDA_VER: '13.4.2', LOCAL_CTK: '0', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM', ENV: { MODE: 'nightly-standard' } }
- { ARCH: 'amd64', PY_VER: '3.15t', CUDA_VER: '13.4.2', LOCAL_CTK: '0', GPU: 'a100', GPU_COUNT: '1', DRIVER: 'latest', DRIVER_MODE: 'MCDM', ENV: { MODE: 'nightly-standard' } }
Comment on lines +121 to +122

@coderabbitai coderabbitai Bot Oct 11, 2026

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

🎯 Functional Correctness | 🟡 Minor | ⚡ Quick win

suggestion: Route the amd64 entries to a nightly-standard job.

ci-nightly.yml runs nightly-standard only for linux-aarch64. The shared Linux workflow filters matrix entries by the requested architecture. These amd64 rows are therefore never selected and add no amd64 Python 3.15 or 3.15t test coverage. Add a linux-64 nightly-standard job or remove the unused rows. (raw.githubusercontent.com)

As per path instructions: “focus on matrix correctness” and “targeted build and test behavior.”

Source: Path instructions


windows:
pull-request:
Expand Down
2 changes: 1 addition & 1 deletion 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
@@ -1,6 +1,6 @@
#!/usr/bin/env bash

# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# SPDX-License-Identifier: Apache-2.0

Expand Down
Loading