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

Open
wants to merge 23 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
23 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
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
53 changes: 53 additions & 0 deletions .github/workflows/build_wheel_windows_arm64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
name: Build Windows ARM64 Wheels

on:
pull_request:
push:
branches:
- nightly
- main
- 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 }}
2 changes: 2 additions & 0 deletions packaging/pre_build_script_arm64.sh
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ cd vcpkg || exit
./bootstrap-vcpkg.sh

# Set vcpkg to only build release packages

echo "set(VCPKG_BUILD_TYPE release)" >> "$TRIPLET_FILE"

# Install dependencies using vcpkg
Expand All @@ -34,6 +35,7 @@ echo "set(VCPKG_BUILD_TYPE release)" >> "$TRIPLET_FILE"
./vcpkg install libpng[tools]:arm64-windows --x-install-root="$DEPENDENCIES_DIR"

# Copy files using cp

cp "$DEPENDENCIES_DIR/arm64-windows/lib/libpng16.lib" "$DEPENDENCIES_DIR/arm64-windows/lib/libpng.lib"
cp "$DEPENDENCIES_DIR/arm64-windows/bin/libpng16.dll" "$DEPENDENCIES_DIR/arm64-windows/bin/libpng.dll"
cp "$DEPENDENCIES_DIR/arm64-windows/bin/libpng16.pdb" "$DEPENDENCIES_DIR/arm64-windows/bin/libpng.pdb"
Expand Down
Loading