Skip to content

Commit

Permalink
Add aarch64 builds on Nova (pytorch#2024)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#2024

Add arm builds through Nova to be hosted on pytorch.org

Reviewed By: osalpekar

Differential Revision: D49333436

fbshipit-source-id: 5d8c192afe10db5ed9b5c6bb35f823d66794e81a
  • Loading branch information
spcyppt authored and facebook-github-bot committed Sep 15, 2023
1 parent bbc676e commit bc0b58d
Showing 1 changed file with 60 additions and 0 deletions.
60 changes: 60 additions & 0 deletions .github/workflows/build_wheels_aarch64_linux.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
name: Build Aarch64 Linux Wheels

on:
pull_request:
push:
branches:
- nightly
- main
# Release candidate branch look like: v1.11.0-release
- v[0-9]+.[0-9]+.[0-9]+-release+
tags:
# Release candidate tag look like: v1.11.0-rc1
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
- v[0-9]+.[0-9]+.[0-9]+
workflow_dispatch:

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
cancel-in-progress: true

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
with:
package-type: wheel
os: linux-aarch64
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/FBGEMM
smoke-test-script: ""
pre-script: ../.github/scripts/nova_prescript.bash
post-script: ../.github/scripts/nova_postscript.bash
env-var-script: .github/scripts/nova_dir.bash
package-name: fbgemm_gpu
name: ${{ matrix.repository }}
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
with:
repository: ${{ matrix.repository }}
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
pre-script: ${{ matrix.pre-script }}
post-script: ${{ matrix.post-script }}
package-name: ${{ matrix.package-name }}
env-var-script: ${{ matrix.env-var-script }}
smoke-test-script: ${{ matrix.smoke-test-script }}
trigger-event: ${{ github.event_name }}
architecture: aarch64
setup-miniconda: false
secrets:
AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }}
AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }}

0 comments on commit bc0b58d

Please sign in to comment.