ci: add ARM64 support with native builds

  Adds comprehensive ARM64 (AWS Graviton, Apple Silicon) testing and
  Docker support using GitHub's free ARM64 runners with native compilation.

  CI Testing (Native ARM64):
  - Add 2 ARM64 configs to pytest-core-nompi (gcc-13, gcc-14 with OpenMP)
  - Add ARM64 Docker test to pytest-core-mpi
  - Tests run natively on ubuntu-24.04-arm runners
  - Covers base and adjoint test sets on ARM64

  Docker Base Images (Native Builds):
  - Separate jobs for native compilation on each platform:
    - deploy-gcc-amd64-bases: ubuntu-24.04 → linux/amd64
    - deploy-gcc-arm64-bases: ubuntu-24.04-arm → linux/arm64
  - Single Dockerfile.cpu builds both architectures (no duplication)
  - Tags: cpu-gcc (x86_64) and arm64-gcc (ARM64) for backwards compatibility
  - Builds OpenMPI 5.0.x from source for GCC 14 compatibility

Tweak

Author: ggorman

.github/actionlint.yaml

@@ -0,0 +1,8 @@
+# actionlint configuration file
+# https://github.com/rhysd/actionlint/blob/main/docs/config.md
+
+self-hosted-runner:
+  # Custom labels for self-hosted runners
+  labels:
+    - nvidiagpu  # Self-hosted runner for NVIDIA GPU builds
+    - amdgpu     # Self-hosted runner for AMD GPU builds

.github/workflows/docker-bases.yml

@@ -16,6 +16,9 @@ on:
       cpu:
         type: boolean
         default: false
+      arm64:
+        type: boolean
+        default: false
       nvidia:
         type: boolean
         default: false
@@ -34,12 +37,12 @@ on:
 
 jobs:
   #######################################################
-  ############## Basic gcc CPU ##########################
+  ############## GCC x86_64 Base ########################
   #######################################################
-  deploy-cpu-bases:
+  deploy-gcc-amd64-bases:
     if: inputs.cpu
-    name: "cpu-base"
-    runs-on: ubuntu-latest
+    name: "gcc-base-amd64"
+    runs-on: ubuntu-24.04
     env:
       DOCKER_BUILDKIT: "1"
 
@@ -54,9 +57,6 @@ jobs:
       - name: Check event name
         run: echo ${{ github.event_name }}
 
-      - name: Set up QEMU
-        uses: docker/setup-qemu-action@v3
-
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
@@ -66,15 +66,56 @@ jobs:
           username: ${{ secrets.DOCKER_USERNAME }}
           password: ${{ secrets.DOCKER_PASSWORD }}
 
-      - name: GCC image
+      - name: Build and push x86_64 GCC image
         uses: docker/build-push-action@v6
         with:
           context: .
           file: "./docker/Dockerfile.cpu"
           push: true
+          platforms: linux/amd64
           build-args: "gcc=${{ matrix.gcc }}"
           tags: "devitocodes/bases:cpu-gcc${{ matrix.gcc }}"
 
+  #######################################################
+  ############## GCC ARM64 Base #########################
+  #######################################################
+  deploy-gcc-arm64-bases:
+    if: inputs.arm64
+    name: "gcc-base-arm64"
+    runs-on: ubuntu-24.04-arm
+    env:
+      DOCKER_BUILDKIT: "1"
+
+    strategy:
+      matrix:
+        gcc: [""]  # Only build default GCC (building from source too slow)
+
+    steps:
+      - name: Checkout devito
+        uses: actions/checkout@v5
+
+      - name: Check event name
+        run: echo ${{ github.event_name }}
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Login to DockerHub
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKER_USERNAME }}
+          password: ${{ secrets.DOCKER_PASSWORD }}
+
+      - name: Build and push ARM64 GCC image
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: "./docker/Dockerfile.cpu"
+          push: true
+          platforms: linux/arm64
+          build-args: "gcc=${{ matrix.gcc }}"
+          tags: "devitocodes/bases:arm64-gcc${{ matrix.gcc }}"
+
   #######################################################
   ############## Intel OneApi CPU #######################
   #######################################################

.github/workflows/docker-devito.yml

@@ -46,6 +46,12 @@ jobs:
             test: 'tests/test_operator.py'
             runner: ubuntu-latest
 
+          - base: 'bases:arm64-gcc'
+            tag: 'arm64-gcc'
+            flag: '--init -t'
+            test: 'tests/test_operator.py'
+            runner: ubuntu-24.04-arm  # GitHub's free ARM64 runner
+
     steps:
       - name: Checkout devito
         uses: actions/checkout@v5

.github/workflows/pytest-core-mpi.yml

@@ -66,25 +66,32 @@ jobs:
       runs-on: ${{ matrix.os }}
       strategy:
         matrix:
-          name: [gcc, icx]
+          name: [gcc, icx, gcc-arm64]
           include:
             - name: gcc
               arch: gcc
+              base_prefix: cpu
               os: ubuntu-latest
               mpiflag: ""
             - name: icx
               arch: icx
+              base_prefix: cpu
               os: ubuntu-latest
               # Need safe math for icx due to inaccuracy with mpi+sinc interpolation
               mpiflag: "-e DEVITO_SAFE_MATH=1"
+            - name: gcc-arm64
+              arch: gcc
+              base_prefix: arm64
+              os: ubuntu-24.04-arm
+              mpiflag: ""
 
       steps:
       - name: Checkout devito
         uses: actions/checkout@v5
 
       - name: Build docker image
         run: |
-          docker build . --file docker/Dockerfile.devito --tag devito_img --build-arg base=devitocodes/bases:cpu-${{ matrix.arch }}
+          docker build . --file docker/Dockerfile.devito --tag devito_img --build-arg base=devitocodes/bases:${{ matrix.base_prefix }}-${{ matrix.arch }}
 
       - name: Test with pytest
         run: |

.github/workflows/pytest-core-nompi.yml

@@ -39,7 +39,9 @@ jobs:
            pytest-osx-py312-clang-omp,
            pytest-docker-py310-gcc-omp,
            pytest-docker-py310-icx-omp,
-           pytest-ubuntu-py313-gcc14-omp
+           pytest-ubuntu-py313-gcc14-omp,
+           pytest-ubuntu-py312-gcc14-omp-arm64,
+           pytest-ubuntu-py311-gcc13-omp-arm64
         ]
         set: [base, adjoint]
         include:
@@ -113,6 +115,20 @@ jobs:
           language: "openmp"
           sympy: "1.14"
 
+        - name: pytest-ubuntu-py312-gcc14-omp-arm64
+          python-version: '3.12'
+          os: ubuntu-24.04-arm
+          arch: "gcc-14"
+          language: "openmp"
+          sympy: "1.14"
+
+        - name: pytest-ubuntu-py311-gcc13-omp-arm64
+          python-version: '3.11'
+          os: ubuntu-24.04-arm
+          arch: "gcc-13"
+          language: "openmp"
+          sympy: "1.13"
+
         - set: base
           test-set: 'not adjoint'
 

docker/README.md

@@ -8,11 +8,12 @@ Devito provides several images that target different architectures and compilers
 
 ### [Devito] on CPU
 
-We provide two CPU images:
-- `devito:gcc-*` with the standard GNU gcc compiler.
-- `devito:icx-*` with the Intel C compiler for Intel architectures.
+We provide CPU images for different architectures and compilers:
+- `devito:gcc-*` with the standard GNU gcc compiler (x86_64).
+- `devito:icx-*` with the Intel C compiler for Intel architectures (x86_64).
+- `devito:arm64-gcc-*` with the standard GNU gcc compiler for ARM64 architectures (AWS Graviton, Apple Silicon, etc.).
 
-These images provide a working environment for any CPU architecture and come with [Devito], `gcc/icx` and `mpi` preinstalled, and utilities such as `jupyter` for usability and exploration of the package.
+These images provide a working environment for their respective CPU architectures and come with [Devito], `gcc/icx` and `mpi` preinstalled, and utilities such as `jupyter` for usability and exploration of the package.
 
 To run this image locally, you will first need to install `docker`. Then, the following commands will get you started:
 
@@ -38,6 +39,38 @@ In addition, the following legacy tags are available:
 - `devito:cpu-*` that corresponds to `devito:gcc-*`
 
 
+### [Devito] on ARM64
+
+We provide ARM64-specific images optimized for ARM64 processors:
+- `devito:arm64-gcc-*` with the standard GNU gcc compiler for ARM64 architectures.
+
+These images support various ARM64 platforms including:
+- AWS Graviton2/3/4 instances
+- Apple Silicon (M1/M2/M3) via Docker Desktop
+- ARM-based cloud instances
+
+Devito automatically detects the specific ARM64 variant at runtime and applies appropriate optimizations through its JIT compiler.
+
+To run on ARM64 systems:
+
+```bash
+# Pull image and start a bash shell
+docker run --rm -it -p 8888:8888 devitocodes/devito:arm64-gcc-latest /bin/bash
+
+# or start a Jupyter notebook server on port 8888
+docker run --rm -it -p 8888:8888 devitocodes/devito:arm64-gcc-latest
+
+# Run an example
+docker run --rm -it devitocodes/devito:arm64-gcc-latest python examples/seismic/acoustic/acoustic_example.py
+```
+
+On AWS Graviton instances with user context:
+
+```bash
+docker run --rm -it -v `pwd`:`pwd` -w `pwd` -u $(id -u):$(id -g) devitocodes/devito:arm64-gcc-latest python examples/seismic/acoustic/acoustic_example.py
+```
+
+
 ### [Devito] on GPU
 
 Second, we provide three images to run [Devito] on GPUs, tagged `devito:nvidia-nvc-*`, and `devito:amd-*`.