use arm runner (#15)

Author: makslevental

.github/workflows/test.yml

@@ -35,7 +35,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ ubuntu-22.04, macos-13, macos-14, macos-15 ]
+        os: [ ubuntu-22.04, ubuntu-22.04-arm, macos-13, macos-14, macos-15 ]
         py_version: [ "3.8", "3.9", "3.10", "3.11", "3.12" ]
 
         exclude:
@@ -112,6 +112,8 @@ jobs:
 
           if [ x"${{ matrix.OS }}" == x"ubuntu-22.04" ]; then
             CIBW_ARCHS=x86_64
+          elif [ x"${{ matrix.OS }}" == x"ubuntu-22.04-arm" ]; then
+            CIBW_ARCHS=aarch64
           elif [ x"${{ matrix.OS }}" == x"macos-13" ]; then
             CIBW_ARCHS=x86_64
           elif [ x"${{ matrix.OS }}" == x"macos-14" ]; then
@@ -127,78 +129,3 @@ jobs:
           cibuildwheel --output-dir wheelhouse
           pip install mmlir -f wheelhouse
           pushd test/python && python smoketest.py && popd
-
-  test-aarch64:
-
-    runs-on: ubuntu-22.04
-
-    strategy:
-      fail-fast: true
-      matrix:
-        py_version: [ "3.8", "3.9", "3.10", "3.11", "3.12" ]
-
-    steps:
-      - name: Checkout
-        uses: actions/checkout@v2
-
-      - name: Test CMake
-        uses: uraimo/run-on-arch-action@v2
-        with:
-          arch: aarch64
-          distro: ubuntu22.04
-          dockerRunArgs: --volume "${{ github.workspace }}:/workspace"
-          install: |
-            
-            apt-get update -q -y
-            apt-get install -y wget build-essential python3 \
-              python-is-python3 python3-pip cmake ninja-build unzip
-        
-          run: |
-            
-            cd /workspace
-            pip install -r requirements.txt
-            pip download mlir -f https://makslevental.github.io/wheels
-            unzip mlir-*.whl
-            rm -rf mlir-*.whl
-            
-            cmake -G Ninja \
-              -DCMAKE_BUILD_TYPE=Release \
-              -DCMAKE_PREFIX_PATH=$PWD/mlir \
-              -DLLVM_EXTERNAL_LIT=$(which lit) \
-              -DPython3_EXECUTABLE=$(which python) \
-              -B build \
-              -S $PWD
-            
-            pushd build && ninja check-minimal && popd
-            rm -rf build
-
-      - name: Test cibuildwheel
-        uses: uraimo/run-on-arch-action@v2
-        with:
-          arch: aarch64
-          distro: ubuntu22.04
-          dockerRunArgs: --volume "${{ github.workspace }}:/workspace"
-          install: |
-            
-            apt-get update -q -y
-            apt-get install -y wget build-essential cmake ninja-build
-
-            mkdir -p ~/miniconda3
-            wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh -O miniconda.sh
-            bash miniconda.sh -b -u -p /root/miniconda3
-            eval "$(/root/miniconda3/bin/conda shell.bash hook)"
-            conda init
-
-          run: |
-            
-            eval "$(/root/miniconda3/bin/conda shell.bash hook)"
-            conda create -n env -q -y -c conda-forge/label/python_rc python=${{ matrix.py_version }} 
-            conda activate env
-            
-            cd /workspace
-            
-            export CIBW_ARCHS=aarch64
-            pip install -r requirements.txt
-            pip wheel . -v -w wheelhouse --no-build-isolation
-            pip install mmlir -f wheelhouse
-            pushd test/python && python smoketest.py && popd