run on tpus

google · Feb 26, 2024 · 2497448 · 2497448
1 parent a1b16ea
commit 2497448
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/.github/workflows/BuildDockerImages.yml b/.github/workflows/BuildDockerImages.yml
@@ -35,6 +35,9 @@ jobs:
     runs-on: ["self-hosted", "tpu", "${{ matrix.device-type }}"]
     steps:
     - uses: actions/checkout@v3
+    - name: Test build_and_upload.sh
+      run : |
+        bash build_and_upload.sh
     - name: build jax stable image
       run : |
         project=tpu-prod-env-multipod
@@ -43,8 +46,6 @@ jobs:
         bash docker_build_dependency_image.sh LOCAL_IMAGE_NAME=$local_image_name MODE=stable &&
         docker tag ${local_image_name} gcr.io/$project/${cloud_image_name}:latest
         docker push gcr.io/$project/${cloud_image_name}:latest
-    - name: Test build_and_upload.sh
-      run : |
-        bash build_and_upload.sh
+