From 59664cc6de66cf39689c73a03e9c2fb15d1f7b3c Mon Sep 17 00:00:00 2001
From: Mark Saroufim <marksaroufim@meta.com>
Date: Fri, 6 Sep 2024 16:45:23 -0700
Subject: [PATCH] Create build-wheels-m1.yml (#822)

Co-authored-by: Eli Uriegas <1700823+seemethere@users.noreply.github.com>
---
 .github/workflows/build-wheels-m1.yml | 39 +++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 .github/workflows/build-wheels-m1.yml

diff --git a/.github/workflows/build-wheels-m1.yml b/.github/workflows/build-wheels-m1.yml
new file mode 100644
index 0000000000..49c9b8cf20
--- /dev/null
+++ b/.github/workflows/build-wheels-m1.yml
@@ -0,0 +1,39 @@
+name: Build M1 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: macos-arm64
+  build:
+    needs: generate-matrix
+    name: pytorch/ao
+    uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@main
+    with:
+      repository: pytorch/ao
+      ref: ${{ github.head_ref || github.ref_name }}
+      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
+      pre-script: packaging/pre_build_script.sh
+      # post-script: packaging/post_build_script.sh
+      package-name: torchao
+      runner-type: macos-m1-stable
+      smoke-test-script: test/smoke_test.py
+      trigger-event: ${{ github.event_name }}