CI/CD - Add image build on arm64 arch (#690)

Add image build on arm64 arch.

CI/CD - Add image build on arm64 arch (#690)
Add image build on arm64 arch.
300df46b · Yifan Xiong · GitHub · 64edc9c5 · 300df46b
Unverified Commit 300df46b authored Mar 07, 2025 by Yifan Xiong Committed by GitHub Mar 07, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 4 deletions

.github/workflows/build-image.yml .github/workflows/build-image.yml +10 -4

No files found.
--- a/.github/workflows/build-image.yml
+++ b/.github/workflows/build-image.yml
@@ -25,17 +25,23 @@ jobs:
    strategy:
      matrix:
        include:
+        - name: cuda12.4 arm64
+          dockerfile: cuda12.4
+          tags: superbench/main:cuda12.4
+          platforms: linux/arm64
+          runner: [self-hosted, linux/arm64]
+          build_args: "NUM_MAKE_JOBS=8"
        - name: cuda12.4
          dockerfile: cuda12.4
          tags: superbench/main:cuda12.4
-          platforms: linux/amd64 # TODO: linux/arm64
+          platforms: linux/amd64
-          runner: [self-hosted]
+          runner: [self-hosted, linux/amd64]
          build_args: "NUM_MAKE_JOBS=16"
        - name: cuda12.2
          dockerfile: cuda12.2
          tags: superbench/main:cuda12.2
          platforms: linux/amd64
-          runner: [self-hosted]
+          runner: [self-hosted, linux/amd64]
          build_args: "NUM_MAKE_JOBS=16"
        - name: cuda11.1.1
          dockerfile: cuda11.1.1
@@ -47,7 +53,7 @@ jobs:
          dockerfile: rocm6.2.x
          tags: superbench/main:rocm6.2
          platforms: linux/amd64
-          runner: [self-hosted]
+          runner: [self-hosted, linux/amd64]
          build_args: "NUM_MAKE_JOBS=16"
    steps:
      - name: Checkout