deps: add support for arm64 for gcc and nvidia

Author: mloubout

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

@@ -66,12 +66,18 @@ jobs:
       runs-on: ${{ matrix.os }}
       strategy:
         matrix:
-          name: [gcc, icx]
+          name: [gcc, gcc-arm, icx]
           include:
             - name: gcc
               arch: gcc
               os: ubuntu-latest
               mpiflag: ""
+
+            - name: gcc-arm
+              arch: gcc
+              os: ubuntu-24.04-arm
+              mpiflag: ""
+
             - name: icx
               arch: icx
               os: ubuntu-latest

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

@@ -38,6 +38,7 @@ jobs:
            pytest-ubuntu-py310-gcc9-omp,
            pytest-osx-py312-clang-omp,
            pytest-docker-py310-gcc-omp,
+           pytest-docker-py310-gcc-omp-arm64,
            pytest-docker-py310-icx-omp,
            pytest-ubuntu-py313-gcc14-omp
         ]
@@ -99,6 +100,13 @@ jobs:
           language: "openmp"
           sympy: "1.13"
 
+        - name: pytest-docker-py310-gcc-omp
+          python-version: '3.10'
+          os: ubuntu-24.04-arm
+          arch: "gcc"
+          language: "openmp"
+          sympy: "1.13"
+
         - name: pytest-docker-py310-icx-omp
           python-version: '3.10'
           os: ubuntu-latest

docker/README.md

@@ -75,6 +75,16 @@ In addition, the following legacy tags are available:
 - `devito:gpu-*` that corresponds to `devito:nvidia-nvc-*`
 
 
+### [Devito] on ARM64
+
+We provide arm64 compatible images for both gcc and nvidia compilers. These images have the same tag as their x86_64 images and will automatically pull the correct architecture based on your system. Namely the base images supporting arm64 are
+
+- `devitocodes/bases:cpu-gcc14` for the gcc 14 base image.
+- `devitocodes/bases:cpu-gcc` for the ubuntu 22.04 default gcc base image.
+- `devitocodes/bases:nvidia-nvc` for the nvidia openacc base image.
+- `devitocodes/bases:nvidia-nvcc` for the nvidia cuda base image.
+
+
 ## Build a [Devito] image
 
 To build the images yourself, you only need to run the standard build command using the provided Dockerfile. The main difference between the CPU and GPU images will be the base image used.