specify platforms per docker build

microsoft · Nov 4, 2024 · 0da1d89 · 0da1d89
1 parent cb1d40b
commit 0da1d89
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/build-image.yml b/.github/workflows/build-image.yml
@@ -28,21 +28,25 @@ jobs:
         - name: cuda12.4
           dockerfile: cuda12.4
           tags: superbench/main:cuda12.4
+          platforms: linux/amd64, linux/arm64
           runner: [self-hosted]
           build_args: "NUM_MAKE_JOBS=16"
         - name: cuda12.2
           dockerfile: cuda12.2
           tags: superbench/main:cuda12.2
+          platforms: linux/amd64
           runner: [self-hosted]
           build_args: "NUM_MAKE_JOBS=16"
         - name: cuda11.1.1
           dockerfile: cuda11.1.1
           tags: superbench/main:cuda11.1.1,superbench/superbench:latest
+          platforms: linux/amd64
           runner: ubuntu-latest
           build_args: "NUM_MAKE_JOBS=8"
         - name: rocm6.2
           dockerfile: rocm6.2.x
           tags: superbench/main:rocm6.2
+          platforms: linux/amd64
           runner: [self-hosted]
           build_args: "NUM_MAKE_JOBS=16"
     steps:
@@ -125,7 +129,7 @@ jobs:
         id: docker_build
         uses: docker/build-push-action@v2
         with:
-          platforms: linux/amd64, linux/arm64
+          platforms: ${{ matrix.platforms }}
           context: .
           file: ${{ steps.metadata.outputs.dockerfile }}
           push: ${{ github.event_name != 'pull_request' }}