ci: Add rosetta-maxtext build

NVIDIA · Apr 19, 2024 · 4fc41f3 · 4fc41f3
1 parent 5acefb4
commit 4fc41f3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
@@ -90,6 +90,16 @@ jobs:
       DOCKERFILE: .github/container/Dockerfile.maxtext.amd64
     secrets: inherit
 
+  build-rosetta-maxtext:
+    needs: build-upstream-maxtext
+    uses: ./.github/workflows/_build_rosetta.yaml
+    with:
+      ARCHITECTURE: ${{ inputs.ARCHITECTURE }}
+      BUILD_DATE: ${{ inputs.BUILD_DATE }}
+      BASE_IMAGE: ${{ needs.build-upstream-maxtext.outputs.DOCKER_TAG_MEALKIT }}
+      BASE_LIBRARY: maxtext
+    secrets: inherit
+
   build-levanter:
     needs: [build-jax]
     uses: ./.github/workflows/_build.yaml