test aarch64

Author: makslevental

.github/workflows/test.yml

@@ -13,7 +13,7 @@ on:
 
 jobs:
 
-  test-against-mlir-bindings:
+  mlir-bindings:
 
     runs-on: ${{ matrix.os }}
 
@@ -57,7 +57,51 @@ jobs:
         run: |
           python examples/mwe.py
 
-  test-against-torch-mlir-bindings:
+  mlir-bindings-aarch64:
+
+    runs-on: ${{ matrix.os }}
+
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ ubuntu-22.04 ]
+        py_version: [ "3.10", "3.11" ]
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: Install, configure, and test
+        uses: uraimo/run-on-arch-action@v2
+        with:
+          arch: aarch64
+          distro: ubuntu20.04
+          dockerRunArgs: --volume "${{ github.workspace }}:/workspace"
+          install: |
+            
+            apt-get update -q -y
+            apt-get install -y wget
+            
+            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
+            conda install -q -y python=${{ matrix.py_version }}
+
+          run: |
+            
+            eval "$(/root/miniconda3/bin/conda shell.bash hook)"
+            
+            cd /workspace
+            
+            pip install .[test,mlir] -f https://makslevental.github.io/wheels
+            configure-mlir-python-utils -y mlir
+            
+            pytest --capture=tee-sys --ignore-glob=*test_smoke* tests
+            python examples/mwe.py
+
+  torch-mlir-bindings:
 
     runs-on: ${{ matrix.os }}
 
@@ -97,7 +141,7 @@ jobs:
           fi
 
 
-  test-against-jax-bindings:
+  jax-bindings:
 
     runs-on: ${{ matrix.os }}
 

pyproject.toml

@@ -17,7 +17,7 @@ dependencies = [
 test = ["pytest", "mlir-native-tools", "astpretty"]
 torch-mlir = ["torch-mlir-core"]
 jax = ["jax[cpu]", ]
-mlir = ["mlir-python-bindings"]
+mlir = ["mlir-python-bindings==18.0.0.2023090322+19d286bc"]
 
 [project.scripts]
 configure-mlir-python-utils = "mlir_utils:_configuration.configuration.configure_host_bindings"