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
Changes from 12 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
70 changes: 70 additions & 0 deletions .github/workflows/build_wheel_windows_arm64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,70 @@
name: Build Windows Arm64 Torchvision Wheel

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:
inputs:
python_version:
description: "Python version"
required: true
type: choice
options:
- Python312
- Python311
default: Python312
msvc_version:
description: "MSVC Version"
required: true
type: choice
options:
- 'latest'
- '14.40'
- '14.36'
default: 'latest'
build_type:
description: "Build type"
required: true
type: choice
options:
- Release
# - Debug (not supported for wheel)
default: Release
repository_name:
description: "torchvision repository (username/repo)"
required: true
type: string
default: "pytorch/vision"
repository_branch:
description: "repository branch"
required: true
type: string
default: "main"
build_version:
description: "Optional: build version for wheel file name (e.g. 2.6.0); leave empty for default naming"
required: false
type: string
default: ""

permissions:
id-token: write
contents: read

jobs:
trigger-test-infra:
uses: alinpahontu2912/test-infra/.github/workflows/build_wheels_windows_arm64.yml@winarm64_wheels
with:
python_version: ${{ inputs.python_version || 'Python312' }}
msvc_version: ${{ inputs.msvc_version || 'latest' }}
build_type: ${{ inputs.build_type || 'Release' }}
repository_name: ${{ inputs.repository_name || 'pytorch/vision' }}
repository_branch: ${{ inputs.repository_branch || 'main' }}
build_version: ${{ inputs.build_version || '' }}
Loading