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,20 +56,16 @@ 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: |
           UPLOAD_IMAGE_TAG=gcr.io/cloud-tpu-inference-test/${LOCAL_IMAGE_TAG}
-          docker tag ${LOCAL_IMAGE_TAG} ${UPLOAD_IMAGE_TAG}
+          docker pull gcr.io/cloud-tpu-inference-test/jetstream-maxtext-stable-stack/tpu:nightly
+          docker pull alpine
+          docker tag alpine ${UPLOAD_IMAGE_TAG}
           docker push ${UPLOAD_IMAGE_TAG}
-          NIGHTLY_TAG=${UPLOAD_IMAGE_TAG%:*}:nightly
-          NIGHTLY_TAG_DATE=${NIGHTLY_TAG}-$(date +"%Y%m%d")
-          docker tag ${LOCAL_IMAGE_TAG} ${NIGHTLY_TAG}
-          docker tag ${LOCAL_IMAGE_TAG} ${NIGHTLY_TAG_DATE}
-          docker push ${NIGHTLY_TAG}
-          docker push ${NIGHTLY_TAG_DATE}
 
   benchmark_report:
     name: Benchmark Report