Skip to content

Add Windows ARM64 build support#2312

Open
isVoid wants to merge 3 commits into
NVIDIA:mainfrom
isVoid:wangm/windows-arm64-bindings
Open

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

Conversation

@isVoid

@isVoid isVoid 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-bot Bot 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.

@isVoid isVoid 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
Contributor Author

/ok to test

@github-actions

github-actions Bot commented Jul 7, 2026

Copy link
Copy Markdown

…-bindings

# Conflicts:
#	cuda_bindings/cuda/bindings/_bindings/loader.cpp
@isVoid

isVoid commented Jul 7, 2026

Copy link
Copy Markdown
Contributor Author

/ok to test

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

isVoid commented Jul 7, 2026

Copy link
Copy Markdown
Contributor Author

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
@isVoid isVoid self-assigned this Jul 7, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cuda.bindings Everything related to the cuda.bindings module cuda.core Everything related to the cuda.core module feature New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant