added functionality tests for upstream-maxtext

Author: Md Khan

.github/workflows/_ci.yaml

@@ -140,6 +140,7 @@ jobs:
       - build-pax
       - build-rosetta-t5x
       - build-rosetta-pax
+      - build-maxtext
     outputs:
       CONTAINER_TAGS: ${{ steps.save-docker-tags.outputs.CONTAINER_TAGS }}
     steps:
@@ -159,7 +160,9 @@ jobs:
             "tag-t5x-mealkit": "${{ needs.build-rosetta-t5x.outputs.DOCKER_TAG_MEALKIT }}", \
             "tag-t5x-final": "${{ needs.build-rosetta-t5x.outputs.DOCKER_TAG_FINAL }}", \
             "tag-pax-mealkit": "${{ needs.build-rosetta-pax.outputs.DOCKER_TAG_MEALKIT }}", \
-            "tag-pax-final": "${{ needs.build-rosetta-pax.outputs.DOCKER_TAG_FINAL }}" \
+            "tag-pax-final": "${{ needs.build-rosetta-pax.outputs.DOCKER_TAG_FINAL }}", \
+            "tag-upstream-maxtext-mealkit": "${{ needs.build-maxtext.outputs.DOCKER_TAG_MEALKIT }}", \
+            "tag-upstream-maxtext-final": "${{ needs.build-maxtext.outputs.DOCKER_TAG_FINAL }}" \
             }
           EOF
 
@@ -243,7 +246,15 @@ jobs:
     with:
       PAX_IMAGE: ${{ needs.build-rosetta-pax.outputs.DOCKER_TAG_FINAL }}
     secrets: inherit
-  
+
+  test-upstream-maxtext:
+    needs: build-maxtext
+    if: inputs.ARCHITECTURE == 'amd64' # no images for arm64
+    uses: ./.github/workflows/_test_maxtext.yaml
+    with:
+      MAXTEXT_IMAGE: ${{ needs.build-maxtext.outputs.DOCKER_TAG_FINAL }}
+    secrets: inherit
+
   ## TODO: uncomments when PR #463
   # test-levanter:
   #   needs: build-levanter

.github/workflows/_sandbox.yaml

@@ -1,10 +1,41 @@
-name: ~Sandbox
+name: "~Sandbox"
 
 on:
   workflow_dispatch:
 
 jobs:
+  sandbox:
+    runs-on: ubuntu-22.04
+    steps:
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v2
+        with:
+          registry: ghcr.io
+          username: ${{ github.repository_owner }}
+          password: ${{ secrets.GITHUB_TOKEN }}
 
-  test-maxtext:
-    uses: ./.github/workflows/_test_maxtext.yaml
-    secrets: inherit
+      - name: Print usage
+        run: |
+          cat << EOF
+          This is an empty workflow file located in the main branch of your
+          repository. It serves as a testing ground for new GitHub Actions on
+          development branches before merging them to the main branch. By
+          defining and overloading this workflow on your development branch,
+          you can test new actions without affecting your main branch, ensuring
+          a smooth integration process once the changes are ready to be merged.
+
+          Usage:
+          
+          1. In your development branch, modify the sandbox.yml workflow file
+             to include the new actions you want to test. Make sure to commit
+             the changes to the development branch.
+          2. Navigate to the 'Actions' tab in your repository, select the
+             '~Sandbox' workflow, and choose your development branch from the
+             branch dropdown menu. Click on 'Run workflow' to trigger the
+             workflow on your development branch.
+          3. Once you have tested and verified the new actions in the Sandbox
+             workflow, you can incorporate them into your main workflow(s) and
+             merge the development branch into the main branch. Remember to
+             revert the changes to the sandbox.yml file in the main branch to
+             keep it empty for future testing.
+          EOF