Skip to content

Add Windows ARM64 build support - #2312

Merged
rwgk merged 3 commits into
NVIDIA:mainfrom
isVoid:wangm/windows-arm64-bindings
Jul 13, 2026
Merged

Add Windows ARM64 build support#2312
rwgk merged 3 commits into
NVIDIA:mainfrom
isVoid:wangm/windows-arm64-bindings

Conversation

@isVoid

@isVoidisVoid commented Jul 7, 2026

Copy link
Copy Markdown
Contributor

What changed

  • Select the CUDA Toolkit lib\arm64 directory when building cuda.bindings on win-arm64.
  • Load nvcudaa64.dll from the Windows CUDA loader path on ARM64 builds.
  • Build the cuda.core AOTI shim import library with /MACHINE:ARM64 for win-arm64.

Why

These changes add native Windows ARM64 build support for cuda.bindings and the cuda.core tensor bridge AOTI shim while preserving existing x64 behavior.

Validation

  • pixi run python -m py_compile cuda_bindings/build_hooks.py
  • pixi run python -m py_compile cuda_core/setup.py cuda_bindings/build_hooks.py
  • git diff --check -- cuda_core/setup.py cuda_bindings/build_hooks.py cuda_bindings/cuda/bindings/_bindings/loader.cpp
  • pre-commit hooks during both commits

@copy-pr-bot

copy-pr-botBot commented Jul 7, 2026

Copy link
Copy Markdown
Contributor

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@isVoidisVoid added feature New feature or request cuda.bindings Everything related to the cuda.bindings module cuda.core Everything related to the cuda.core module labels Jul 7, 2026 — with ChatGPT Codex Connector
@isVoid

isVoid commented Jul 7, 2026

Copy link
Copy Markdown
ContributorAuthor

/ok to test

@github-actions

github-actionsBot commented Jul 7, 2026

Copy link
Copy Markdown
Doc Preview CI

🚀 View preview at
https://nvidia.github.io/cuda-python/pr-preview/pr-2312/

https://nvidia.github.io/cuda-python/pr-preview/pr-2312/cuda-core/

https://nvidia.github.io/cuda-python/pr-preview/pr-2312/cuda-bindings/

https://nvidia.github.io/cuda-python/pr-preview/pr-2312/cuda-pathfinder/


Preview will be ready when the GitHub Pages deployment is complete.

@isVoid
Merge remote-tracking branch 'upstream/main' into wangm/windows-arm64…
fe20915
…-bindings
# Conflicts:
#	cuda_bindings/cuda/bindings/_bindings/loader.cpp
@isVoid

isVoid commented Jul 7, 2026

Copy link
Copy Markdown
ContributorAuthor

/ok to test

@leofang
leofang self-requested a review July 7, 2026 19:41
@isVoid

isVoid commented Jul 7, 2026

Copy link
Copy Markdown
ContributorAuthor

This PR is to additionally verify that the WoA build changes does not affect the existing build system by running on the full CI matrix.

@isVoid
isVoid marked this pull request as ready for review July 7, 2026 20:48
@isVoidisVoid self-assigned this Jul 7, 2026
@isVoidisVoid added the PR review get-together Mark PRs you'd like the team to review at the weekly PR review get-together. label Jul 13, 2026
@rwgk
rwgk merged commit fcc9dec into NVIDIA:mainJul 13, 2026
112 of 113 checks passed
github-actionsBot pushed a commit that referenced this pull request Jul 14, 2026
cuda-python-bot
Clean up PR preview folders for 2 closed/merged PRs
69f7bcb
Removed preview folders for the following PRs:
- PR #2296
- PR #2312
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@rwgkrwgkrwgk approved these changes
@leofangleofangAwaiting requested review from leofang

Assignees

@isVoidisVoid

Labels

cuda.bindingsEverything related to the cuda.bindings modulecuda.coreEverything related to the cuda.core modulefeatureNew feature or requestPR review get-togetherMark PRs you'd like the team to review at the weekly PR review get-together.

Projects

None yet

Milestone

cuda.bindings 13.4.0 & 12.9.8

Development

Successfully merging this pull request may close these issues.

2 participants

@isVoid@rwgk