Skip to content

Arm64 workflow enablement #8947

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.

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

Merged
merged 27 commits into from
Jun 26, 2025
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
849879f
add windows arm64 workflow
alinpahontu2912 Feb 28, 2025
c72d258
Merge branch 'pytorch:main' into arm64_enablement
alinpahontu2912 Mar 3, 2025
839d069
deafult values for build and trigger workflow
alinpahontu2912 Mar 3, 2025
3698e04
Merge branch 'arm64_enablement' of https://github.com/alinpahontu2912…
alinpahontu2912 Mar 3, 2025
f060536
remove hardcode branch name
alinpahontu2912 Mar 3, 2025
38e4922
update workflow
alinpahontu2912 Mar 3, 2025
c97f7e5
test rerun workflow
alinpahontu2912 Mar 3, 2025
94add51
update workflow
alinpahontu2912 Mar 5, 2025
59f00e4
update workflow
alinpahontu2912 Mar 5, 2025
15fb649
fix error
alinpahontu2912 Mar 5, 2025
56c7573
fix inputs and add defaults
alinpahontu2912 Mar 5, 2025
7c67cef
Merge branch 'main' into arm64_enablement
alinpahontu2912 Mar 5, 2025
1305263
remove unwanted inputs
alinpahontu2912 Mar 31, 2025
a14c5e0
remove unnecessary inputs and reformat
alinpahontu2912 Apr 2, 2025
d256624
Merge branch 'arm64_enablement' of https://github.com/alinpahontu2912…
alinpahontu2912 Apr 2, 2025
4ec4073
Merge branch 'pytorch:main' into arm64_enablement
alinpahontu2912 Apr 9, 2025
18d1f9d
Merge branch 'pytorch:main' into arm64_enablement
alinpahontu2912 Apr 11, 2025
f3e05c3
test trigger arm64 workflow
alinpahontu2912 Apr 11, 2025
e369831
use correct refs
alinpahontu2912 Apr 11, 2025
6e94da7
update workflow and scripts for arm64
alinpahontu2912 May 6, 2025
d4daa6a
Merge branch 'main' into arm64_enablement
alinpahontu2912 May 6, 2025
5e61df8
add link for stable release torch package until pypi download is avai…
alinpahontu2912 May 20, 2025
1925d00
Merge branch 'main' into arm64_enablement
alinpahontu2912 May 23, 2025
92b2649
Merge branch 'main' into arm64_enablement
alinpahontu2912 Jun 25, 2025
bf3a078
remove not needed env variables
alinpahontu2912 Jun 25, 2025
4be3d3b
Invoke only on nightly or if .yml is modified
malfet Jun 26, 2025
f842b49
Update .github/workflows/build_wheel_windows_arm64.yml
malfet Jun 26, 2025
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
54 changes: 54 additions & 0 deletions .github/workflows/build_wheel_windows_arm64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
name: Build Windows ARM64 Wheels

on:
pull_request:
paths:
- .github/workflows/build_wheel_windows_arm64.yml
push:
branches:
- nightly
- release/*
tags:
# NOTE: Binary build pipelines should only get triggered on release candidate builds
# Release candidate tags look like: v1.11.0-rc1
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
workflow_dispatch:

permissions:
id-token: write
contents: read

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
with:
package-type: wheel
os: windows-arm64
test-infra-repository: pytorch/test-infra
test-infra-ref: main
with-cuda: disable

build:
needs: generate-matrix
strategy:
fail-fast: false
matrix:
include:
- repository: pytorch/vision
smoke-test-script: test/smoke_test.py
pre-script: packaging/pre_build_script_arm64.sh
package-name: torchvision
architecture: "arm64"
name: ${{ matrix.repository }}
uses: pytorch/test-infra/.github/workflows/build_wheels_windows.yml@main
with:
repository: ${{ matrix.repository }}
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
pre-script: ${{ matrix.pre-script }}
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
package-name: ${{ matrix.package-name }}
smoke-test-script: ${{ matrix.smoke-test-script }}
trigger-event: ${{ github.event_name }}
architecture: ${{ matrix.architecture }}
3 changes: 0 additions & 3 deletions packaging/pre_build_script_arm64.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,6 @@ echo "Building vision dependencies and wheel started."
export SRC_PATH="$GITHUB_WORKSPACE/$SRC_DIR"
export CMAKE_BUILD_TYPE="$BUILD_TYPE"
export VCVARSALL_PATH="$DEPENDENCIES_DIR/VSBuildTools/VC/Auxiliary/Build/vcvarsall.bat"
export CONDA_PREFIX="$DEPENDENCIES_DIR"
export PATH="$PATH:$CONDA_PREFIX/Library/bin"
export DISTUTILS_USE_SDK=1
export TRIPLET_FILE="triplets/arm64-windows.cmake"
export PYTORCH_VERSION="$PYTORCH_VERSION"
export CHANNEL="$CHANNEL"
Expand Down
Loading