Skip to content

Commit 87d2327

Browse files
committed
Add aarch64 builds on Nova (#2024)
Summary: Pull Request resolved: #2024 Add arm builds through Nova to be hosted on pytorch.org Reviewed By: osalpekar Differential Revision: D49333436 fbshipit-source-id: 5d8c192afe10db5ed9b5c6bb35f823d66794e81a
1 parent fe8d312 commit 87d2327

File tree

1 file changed

+60
-0
lines changed

1 file changed

+60
-0
lines changed
Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
name: Build Aarch64 Linux Wheels
2+
3+
on:
4+
pull_request:
5+
push:
6+
branches:
7+
- nightly
8+
- main
9+
# Release candidate branch look like: v1.11.0-release
10+
- v[0-9]+.[0-9]+.[0-9]+-release+
11+
tags:
12+
# Release candidate tag look like: v1.11.0-rc1
13+
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
14+
- v[0-9]+.[0-9]+.[0-9]+
15+
workflow_dispatch:
16+
17+
concurrency:
18+
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
19+
cancel-in-progress: true
20+
21+
jobs:
22+
generate-matrix:
23+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
24+
with:
25+
package-type: wheel
26+
os: linux-aarch64
27+
test-infra-repository: pytorch/test-infra
28+
test-infra-ref: main
29+
with-cuda: disable
30+
build:
31+
needs: generate-matrix
32+
strategy:
33+
fail-fast: false
34+
matrix:
35+
include:
36+
- repository: pytorch/FBGEMM
37+
smoke-test-script: ""
38+
pre-script: ../.github/scripts/nova_prescript.bash
39+
post-script: ../.github/scripts/nova_postscript.bash
40+
env-var-script: .github/scripts/nova_dir.bash
41+
package-name: fbgemm_gpu
42+
name: ${{ matrix.repository }}
43+
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
44+
with:
45+
repository: ${{ matrix.repository }}
46+
ref: ""
47+
test-infra-repository: pytorch/test-infra
48+
test-infra-ref: main
49+
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
50+
pre-script: ${{ matrix.pre-script }}
51+
post-script: ${{ matrix.post-script }}
52+
package-name: ${{ matrix.package-name }}
53+
env-var-script: ${{ matrix.env-var-script }}
54+
smoke-test-script: ${{ matrix.smoke-test-script }}
55+
trigger-event: ${{ github.event_name }}
56+
architecture: aarch64
57+
setup-miniconda: false
58+
secrets:
59+
AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }}
60+
AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }}

0 commit comments

Comments
 (0)