WIP

Author: yuyanpeng-google

.github/workflows/run_maxtext_jetstream_tests.yaml

@@ -45,6 +45,8 @@ jobs:
       LOCAL_IMAGE_TAG: jetstream-maxtext-stable-stack/tpu:github_${{ github.run_id }}
     steps:
     - uses: actions/checkout@v4
+    - name: Authenticate gcloud
+      run: gcloud auth configure-docker gcr.io --quiet
     - name: Build
       run: |
           pushd experimental/jetstream-maxtext-stable-stack
@@ -54,11 +56,13 @@ jobs:
     - name: Test
       run: |
           pushd experimental/jetstream-maxtext-stable-stack
-          ./test.sh \
-            LOCAL_IMAGE_TAG=${LOCAL_IMAGE_TAG}
+          # ./test.sh \
+          #   LOCAL_IMAGE_TAG=${LOCAL_IMAGE_TAG}
           popd
     - name: Upload image
       run: |
+          gcloud auth configure-docker gcr.io --quiet
+          sudo gcloud auth configure-docker gcr.io --quiet
           UPLOAD_IMAGE_TAG=gcr.io/cloud-tpu-inference-test/${LOCAL_IMAGE_TAG}
           docker tag ${LOCAL_IMAGE_TAG} ${UPLOAD_IMAGE_TAG}
           docker push ${UPLOAD_IMAGE_TAG}