Aarch runner

.github/workflows/python.yml

@@ -88,10 +88,11 @@ jobs:
 
           if [[ $TAGGED_OR_MAIN ]]; then
             # MacOS build is really slow (>30 mins); uses up a lot of CI minutes
-            matrix+=('{"platform": "macos", "runs_on": "macos-latest"},')
+            matrix+=('{"platform": "macos", "runs_on": "macos-latest", "pip": "pip", "python": "python3"},')
           fi
-          matrix+=('{"platform": "windows", "runs_on": "windows-latest"},')
-          matrix+=('{"platform": "linux", "runs_on": "ubuntu-latest", container: {"image": "rerunio/ci_docker:0.5"}}')
+          matrix+=('{"platform": "windows", "runs_on": "windows-latest", "pip": "pip", "python": "python"},')
+          matrix+=('{"platform": "linux", "runs_on": "ubuntu-latest", container: {"image": "rerunio/ci_docker:0.5"}, "pip": "pip", "python": "python3"},')
+          matrix+=('{"platform": "aarch64", "runs_on": ["self-hosted", "linux", "ARM64"], container: {"image": "quay.io/pypa/manylinux_2_28_aarch64"}, "pip": "python3.8 -m pip", "python": "python3.8"}')
 
           echo "Matrix values: ${matrix[@]}"
 
@@ -109,6 +110,10 @@ jobs:
     container: ${{ matrix.container }}
 
     steps:
+      - name: Install rust
+        if: matrix.platform == 'aarch64'
+        run: curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+
       - uses: actions/checkout@v3
 
       # These should already be in the docker container, but run for good measure. A no-op install
@@ -122,6 +127,20 @@ jobs:
           version: 2.0 # Increment this to pull newer packages
           execute_install_scripts: true
 
+      - name: Cache DNF packages
+        if: matrix.platform == 'aarch64'
+        uses: actions/cache@v2
+        with:
+          path: |
+            /var/cache/dnf
+          key: ${{ runner.os }}-dnf-aarch64
+
+      - name: AARCH64 install packages
+        if: matrix.platform == 'aarch64'
+        run: |
+          dnf makecache --refresh
+          dnf install -y gtk3-devel openssl-devel
+
       - name: Set up cargo cache
         uses: Swatinem/rust-cache@v2
         with:
@@ -134,7 +153,7 @@ jobs:
       # The pip-cache setup logic doesn't work in the ubuntu docker container
       # That's probably fine since we bake these deps into the container already
       - name: Setup python
-        if: matrix.platform != 'linux'
+        if: matrix.platform != 'linux' && matrix.platform != 'aarch64'
         uses: actions/setup-python@v4
         with:
           python-version: ${{ env.PYTHON_VERSION }}
@@ -144,7 +163,14 @@ jobs:
       # These should already be in the docker container, but run for good measure. A no-op install
       # should be fast, and this way things don't break if we add new packages without rebuilding
       # docker
-      - run: pip install -r rerun_py/requirements-build.txt
+      - name: "Install build requirements"
+        run: |
+          if [ "${{ matrix.platform }}" = "aarch64" ]; then
+            ${{ matrix.pip }} install -r rerun_py/requirements-build.txt && echo "/opt/_internal/cpython-3.8.16/bin" >> $GITHUB_PATH
+          else
+            ${{ matrix.pip }} install -r rerun_py/requirements-build.txt
+          fi
+        shell: bash
 
       # ----------------------------------------------------------------------------------
       # Install prerequisites for building the web-viewer Wasm
@@ -196,6 +222,7 @@ jobs:
           python3 scripts/version_util.py --check_version
 
       - name: Build Wheel
+        if: matrix.platform != 'aarch64'
         uses: PyO3/maturin-action@v1
         with:
           maturin-version: "0.14.10"
@@ -209,19 +236,32 @@ jobs:
             --features pypi
             --universal2
             --out pre-dist
+            -i ${{ matrix.python }}
+      - name: Build Wheel (aarch64)
+        if: matrix.platform == 'aarch64'
+        run: python3.8 -m maturin build --manifest-path rerun_py/Cargo.toml --release --no-default-features --features pypi --universal2 --out pre-dist
 
       - name: Install built wheel
         run: |
-          pip install depthai_viewer --find-links pre-dist --force-reinstall
+          ${{ matrix.pip }} install depthai_viewer --find-links pre-dist --force-reinstall
 
       - name: Run tests
-        run: cd rerun_py/tests && pytest
+        run: |
+          if [ "${{ matrix.platform }}" = "aarch64" ]; then
+            cd rerun_py/tests && python3.8 -m pytest
+          else
+            cd rerun_py/tests && pytest
+          fi
 
       - name: Unpack the wheel
         shell: bash
         run: |
           mkdir unpack-dist
-          wheel unpack pre-dist/*.whl --dest unpack-dist
+          if [ "${{ matrix.platform }}" == "aarch64" ]; then
+            python3.8 -m wheel unpack pre-dist/*.whl --dest unpack-dist
+          else
+            wheel unpack pre-dist/*.whl --dest unpack-dist
+          fi
 
       - name: Get the folder name
         shell: bash
@@ -232,7 +272,11 @@ jobs:
         shell: bash
         run: |
           mkdir dist
-          wheel pack unpack-dist/${{ env.pkg_folder }} --dest dist/
+          if [ "${{ matrix.platform }}" == "aarch64" ]; then
+            python3.8 -m wheel pack unpack-dist/${{ env.pkg_folder }} --dest dist/
+          else
+            wheel pack unpack-dist/${{ env.pkg_folder }} --dest dist/
+          fi
 
       - name: Upload wheels
         uses: actions/upload-artifact@v3