Expand wheel build to arm64 and Windows by clin1234 · Pull Request #2387 · NVIDIA/cuda-python

clin1234 · 2026-07-18T16:43:15Z

Description

closes #2163

Checklist

New or existing tests cover these changes.
The documentation is up to date with these changes.

copy-pr-bot · 2026-07-18T16:43:18Z

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

leofang

Thanks for your contribution! I don't think we want to burn resources this much for pre-release software on a per-PR basis. Could you keep 3.15/3.15t to nightly-standard rows only?

Removed configurations for Python version 3.15 and 3.15t from the test matrix.

leofang · 2026-07-20T17:57:06Z

 host = [
-    # TODO: remove the Python 3.15 guard once 3.15 is officially supported
-    "nvidia-cutlass; python_version < '3.15'",
+    "nvidia-cutlass",


Did you check that cutlass already has 3.15 support?

Co-authored-by: Leo Fang <leo80042@gmail.com>

Added new configurations for Python 3.15 and 3.15t with CUDA 13.3.0 for AMD64 and ARM64 architectures.

clin1234 · 2026-08-11T12:31:14Z

@leofang, please review

coderabbitai · 2026-10-11T09:27:56Z

Summary

Summary by CodeRabbit

Tests
- Expanded nightly test coverage for CUDA 13.5 on arm64 and amd64 systems, including Python 3.15 and 3.15t configurations. The matrix includes A100 or L4 GPU configurations across these platform and Python combinations. These changes affect automated testing and do not introduce user-facing product features.

The Linux nightly-standard matrix adds five CUDA 13.5.0 test rows: four for Python 3.15 or 3.15t, and one for Python 3.14.

Changes: Python 3.15 CI

Layer / File(s)	Summary
Add Python 3.15 nightly matrix entries `ci/test-matrix.yml`	The matrix adds four Python 3.15 or 3.15t rows and one Python 3.14 row for Linux nightly-standard tests with CUDA 13.5.0.

Assessment against linked issues

Objective	Addressed	Explanation
Build and test Python 3.15 pre-releases in CI [`#2163`]	✅

Out-of-scope changes

Code Change	Explanation
Add a Python 3.14/H200 matrix row (ci/test-matrix.yml:126-130)	The linked objective concerns Python 3.15 CI support, not Python 3.14 coverage.

Change: Other

Merge Risk: Low · up to 36b6b

The new amd64 configurations do not run in nightly-standard CI. Enable them or accept the coverage gap before merging.

- [ ] Autofix · Keep fixing CodeRabbit findings and required CI, and resolving merge conflicts

coderabbitai

Actionable comments posted: 1

ℹ️ Review info

⚙️ Run configuration

Configuration used: Repository: NVIDIA/cuda-python/.coderabbit.yaml
Review profile: CHILL
Plan: Advanced
Run ID: c5d5f34c-1a63-404f-b6a6-5133a9df1594

📥 Commits

Reviewing files that changed from the base of the PR and between c7e6f09 and 249d73c.

📒 Files selected for processing (2)

ci/test-matrix.yml
ci/tools/download-wheels

Included review availability: This review used your included allowance. Your plan provides up to 8 included reviews per hour; 7 remain after this review.

coderabbitai · 2026-10-11T09:27:59Z

+    - { 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' } }


🎯 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

clin1234

Looks good

_{Commented in CodeRabbit Change Stack}

coderabbitai

Additional comments: 1

ci/test-matrix.yml-126-127: 1

126-127: Minor · Performance & Scalability · Quick win

Unverified finding
Verification ran but could not confirm this finding. It is shown for review, not as a verified issue.

important: Avoid repeating the arm64 GPU jobs.

Lines 91 and 93 already run these configurations in the scheduled full CI. The nightly workflow repeats the standard tests on the same runner types, adding two GPU jobs without a distinct configuration. Remove these rows or use nightly-only configurations. (raw.githubusercontent.com)

As per path instructions: “avoid unnecessary expensive jobs.”

Source: Path instructions

Review info

Run configuration

Source mode: Auto

Repository: NVIDIA/cuda-python

Branch: main

File: .coderabbit.yaml

Commit: 8b66bdc

Layers: Repository YAML

Review profile: CHILL

Plan: Advanced

Run ID: `dd6e17c1-97f1-4d38-a76d-ff37aa6f16bb`

Commits

Reviewing files that changed from the base of the PR and between `249d73c` and `36b6bb9`.

Files selected for processing: 1

`ci/test-matrix.yml`

Included review availability: This review used your included allowance. Your plan provides up to 8 included reviews per hour; 7 remain after this review.

github-actions Bot added CI/CD CI/CD infrastructure cuda.bindings Everything related to the cuda.bindings module cuda.core Everything related to the cuda.core module cuda.pathfinder Everything related to the cuda.pathfinder module labels Jul 18, 2026

clin1234 changed the title ~~315 support~~ Expand wheel build to arm64 and Windows Jul 18, 2026

leofang requested changes Jul 20, 2026

View reviewed changes

leofang assigned clin1234 and leofang Jul 20, 2026

leofang added P2 Low priority - Nice to have and removed cuda.bindings Everything related to the cuda.bindings module cuda.core Everything related to the cuda.core module cuda.pathfinder Everything related to the cuda.pathfinder module labels Jul 20, 2026

unknown added 5 commits July 20, 2026 13:35

Expand test matrix for Python 3.15

5b541a5

Don't exclude 3.15 wheels from downloaded artifacts

79a0eed

Attempt to build 3.15 wheels on all platforms

07e037b

Fix pytest param

6189daf

scipy not supported on 3.15 yet

79a6760

clin1234 force-pushed the 315-support branch from fc558a6 to 79a6760 Compare July 20, 2026 17:35

github-actions Bot added cuda.bindings Everything related to the cuda.bindings module cuda.core Everything related to the cuda.core module cuda.pathfinder Everything related to the cuda.pathfinder module labels Jul 20, 2026

Restrict Python 3.15 configurations to nightly wheel builds

60ee552

Removed configurations for Python version 3.15 and 3.15t from the test matrix.

leofang requested changes Jul 20, 2026

View reviewed changes

clin1234 and others added 4 commits July 20, 2026 14:32

Apply suggestion from @leofang

b32546e

Co-authored-by: Leo Fang <leo80042@gmail.com>

Add Python 3.15 configurations to test matrix

ecddeb6

Added new configurations for Python 3.15 and 3.15t with CUDA 13.3.0 for AMD64 and ARM64 architectures.

Revert TOML changes

44af69f

lol

3c7bf6a

clin1234 requested a review from leofang October 11, 2026 09:21

Merge branch 'main' into 315-support

249d73c

clin1234 marked this pull request as ready for review October 11, 2026 09:23

clin1234 marked this pull request as draft October 11, 2026 09:25

coderabbitai Bot reviewed Oct 11, 2026

View reviewed changes

clin1234 commented Oct 11, 2026

View reviewed changes

Merge branch 'main' into 315-support

36b6bb9

clin1234 marked this pull request as ready for review November 10, 2026 22:52

coderabbitai Bot reviewed Nov 10, 2026

View reviewed changes

		- { 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' } }

Conversation

clin1234 commented Jul 18, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Description

Checklist

Uh oh!

copy-pr-bot Bot commented Jul 18, 2026

Uh oh!

leofang left a comment

Choose a reason for hiding this comment

Uh oh!

leofang Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

clin1234 commented Aug 11, 2026

Uh oh!

coderabbitai Bot commented Oct 11, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Summary by CodeRabbit

Assessment against linked issues

Out-of-scope changes

Uh oh!

coderabbitai Bot left a comment

Choose a reason for hiding this comment

Uh oh!

coderabbitai Bot Oct 11, 2026

Choose a reason for hiding this comment

Uh oh!

clin1234 left a comment

Choose a reason for hiding this comment

Uh oh!

coderabbitai Bot left a comment

Choose a reason for hiding this comment

Uh oh!

This branch has not been deployed

Reviewers

Uh oh!

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants

clin1234 commented Jul 18, 2026 •

edited

Loading

coderabbitai Bot commented Oct 11, 2026 •

edited

Loading