Add an Arm job (#22)

This is required to build wheels for Arm.

Author: banach-space

.github/workflows/buildReleaseAndPublish.yml

@@ -16,8 +16,7 @@ concurrency:
 
 jobs:
   build_linux:
-    name: Linux x86_64 Build
-    runs-on: ubuntu-latest
+    name: Linux Build
     # Elevates permissions for `GITHUB_TOKEN`'s content scope (to allow this action to make a release)
     permissions:
       contents: write
@@ -27,6 +26,9 @@ jobs:
       matrix:
         package: [torch-mlir]
         py_version: [cp310-cp310, cp311-cp311, cp312-cp312]
+        arch: [x64_64, AArch64]
+
+    runs-on: ${{ matrix.arch == 'AArch64' && 'ubuntu-24.04-arm' || 'ubuntu-24.04' }}
 
     steps:
       - name: Checkout torch-mlir

README.md

@@ -1,9 +1,12 @@
 # Torch MLIR Build and Release Automation
 
+[![Build Release and Publish Windows Wheels](https://github.com/llvm/torch-mlir-release/actions/workflows/buildReleaseAndPublishWindows.yml/badge.svg)](https://github.com/llvm/torch-mlir-release/actions/workflows/buildReleaseAndPublishWindows.yml)
+[![Build Release and Publish Linux Wheels](https://github.com/llvm/torch-mlir-release/actions/workflows/buildReleaseAndPublish.yml/badge.svg)](https://github.com/llvm/torch-mlir-release/actions/workflows/buildReleaseAndPublish.yml)
+
 This repository houses release automation for torch-mlir. It is isolated from the main repository
 so that we can more strictly control people with write access.
 
-### Python Releases (x86 linux and windows)
+### Python Releases
 Python wheels for torch-mlir are auto-generated every night and made available at [this release page](https://github.com/llvm/torch-mlir-release/releases/tag/dev-wheels).
 
 To install via `pip`, run:
@@ -18,4 +21,4 @@ or add this to `requirements.txt`:
 torch-mlir
 ```
 
-At the moment only linux and windows x86_64 builds are published, but this may be updated if the need arises (and we find appropriate owners to maintain the automation).
+At the moment only Linux (x86_64 + AArch64) and Windows (x86_64) builds are published. If you are interested in other configurations, please send a PR.