Update build-image.yml

microsoft · Sep 19, 2024 · 14fc577 · 14fc577
1 parent 9f3be29
commit 14fc577
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/build-image.yml b/.github/workflows/build-image.yml
@@ -40,11 +40,16 @@ jobs:
           tags: superbench/main:cuda11.1.1,superbench/superbench:latest
           runner: ubuntu-latest
           build_args: "NUM_MAKE_JOBS=8"
+        - name: rocm5.7
+          dockerfile: rocm5.7.x
+          tags: superbench/main:rocm5.7
+          runner: [self-hosted, rocm-build]
+          build_args: "NUM_MAKE_JOBS=16"
         - name: rocm6.2
           dockerfile: rocm6.2.x
           tags: superbench/main:rocm6.2
           runner: [self-hosted, rocm-build]
-          build_args: "NUM_MAKE_JOBS=64"
+          build_args: "NUM_MAKE_JOBS=16"
     steps:
       - name: Checkout
         uses: actions/checkout@v2