docker: add arm platforms

devitocodes · Aug 8, 2023 · 1d31fd1 · 1d31fd1
1 parent a1506b6
commit 1d31fd1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.github/workflows/docker-bases.yml b/.github/workflows/docker-bases.yml
@@ -59,6 +59,7 @@ jobs:
           target: 'gcc'
           build-args: 'arch=gcc'
           tags: 'devitocodes/bases:cpu-gcc'
+          platforms: linux/amd64,linux/arm64,linux/arm/v6,linux/arm/v7
 
 #######################################################
 ############## Intel OneApi CPU #######################

diff --git a/.github/workflows/docker-devito.yml b/.github/workflows/docker-devito.yml
@@ -23,25 +23,29 @@ jobs:
             flag: '--gpus all'
             test: 'tests/test_gpu_openacc.py tests/test_gpu_common.py'
             runner: ["self-hosted", "nvidiagpu"]
+            platforms: linux/amd64
 
           # Runtime gpu flags from https://hub.docker.com/r/rocm/tensorflow/
           - base: 'bases:amd'
             tag: 'amd'
             flag: '--network=host --device=/dev/kfd --device=/dev/dri --ipc=host --group-add video --group-add $(getent group render | cut -d: -f3) --cap-add=SYS_PTRACE --security-opt seccomp=unconfined'
             test: 'tests/test_gpu_openmp.py'
             runner: ["self-hosted", "amdgpu"]
+            platforms: linux/amd64
 
           - base: 'bases:cpu-gcc'
             tag: "gcc"
             flag: ''
             test: 'tests/test_operator.py'
             runner: ubuntu-latest
+            platforms: linux/amd64,linux/arm64,linux/arm/v6,linux/arm/v7
 
           - base: 'bases:cpu-icx'
             tag: "icx"
             flag: ''
             test: 'tests/test_operator.py'
             runner: ubuntu-latest
+            platforms: linux/amd64
 
     steps:
       - name: Checkout devito
@@ -100,6 +104,7 @@ jobs:
           push: true
           tags: ${{ steps.meta.outputs.tags }}
           build-args: base=devitocodes/${{ matrix.base }}
+          platforms: ${{ matrix.platforms }}
 
       - name: Remove dangling layers
         run: docker system prune -f