Update _sandbox.yaml

NVIDIA · Nov 17, 2023 · dd030b5 · dd030b5
1 parent 833dd1f
commit dd030b5
Showing 1 changed file with 4 additions and 66 deletions.
diff --git a/.github/workflows/_sandbox.yaml b/.github/workflows/_sandbox.yaml
@@ -116,16 +116,6 @@ jobs:
       REF_XLA: ${{ needs.metadata.outputs.REF_XLA }}
     secrets: inherit
 
-  build-te:
-    needs: [metadata, build-jax]
-    uses: ./.github/workflows/_build_te.yaml
-    with:
-      BUILD_DATE: ${{ needs.metadata.outputs.BUILD_DATE }}
-      BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAGS }}
-      REPO_TE: ${{ needs.metadata.outputs.REPO_TE }}
-      REF_TE: ${{ needs.metadata.outputs.REF_TE }}
-    secrets: inherit
-
   build-t5x:
     needs: [metadata, build-jax]
     uses: ./.github/workflows/_build_t5x.yaml
@@ -136,39 +126,9 @@ jobs:
       REF_T5X: ${{ needs.metadata.outputs.REF_T5X }}
     secrets: inherit
 
-  build-pax:
-    needs: [metadata, build-jax]
-    uses: ./.github/workflows/_build_pax.yaml
-    with:
-      BUILD_DATE: ${{ needs.metadata.outputs.BUILD_DATE }}
-      BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAGS }}
-      REPO_PAXML: ${{ needs.metadata.outputs.REPO_PAXML }}
-      REF_PAXML: ${{ needs.metadata.outputs.REF_PAXML }}
-      REPO_PRAXIS: ${{ needs.metadata.outputs.REPO_PRAXIS }}
-      REF_PRAXIS: ${{ needs.metadata.outputs.REF_PRAXIS }}
-    secrets: inherit
-
-  build-rosetta-t5x:
-    uses: ./.github/workflows/_build_rosetta.yaml
-    needs: [metadata, build-t5x]
-    with:
-      BUILD_DATE: ${{ needs.metadata.outputs.BUILD_DATE }}
-      BASE_IMAGE: ${{ needs.build-t5x.outputs.DOCKER_TAGS }}
-      BASE_LIBRARY: t5x
-      PLATFORMS: '["amd64"]'
-    secrets: inherit
-
-  build-rosetta-pax:
-    uses: ./.github/workflows/_build_rosetta.yaml
-    needs: [metadata, build-pax]
-    with:
-      BUILD_DATE: ${{ needs.metadata.outputs.BUILD_DATE }}
-      BASE_IMAGE: ${{ needs.build-pax.outputs.DOCKER_TAGS }}
-      BASE_LIBRARY: pax
-    secrets: inherit
 
   build-summary:
-    needs: [build-base, build-jax, build-te, build-t5x, build-pax, build-rosetta-t5x, build-rosetta-pax]
+    needs: [build-base, build-jax, build-t5x]
     # needs: [build-base, build-jax, build-te, build-t5x, build-pax, build-pax-aarch64, build-rosetta-t5x, build-rosetta-pax]
     if: always()
     runs-on: ubuntu-22.04
@@ -183,15 +143,12 @@ jobs:
           | ------------ | -------------------------------------------------- |
           | Base         | ${{ needs.build-base.outputs.DOCKER_TAGS }}        |
           | JAX          | ${{ needs.build-jax.outputs.DOCKER_TAGS }}         |
-          | JAX-TE       | ${{ needs.build-te.outputs.DOCKER_TAGS }}          |
           | T5X          | ${{ needs.build-t5x.outputs.DOCKER_TAGS }}         |
-          | PAX          | ${{ needs.build-pax.outputs.DOCKER_TAGS }}         |
-          | ROSETTA(t5x) | ${{ needs.build-rosetta-t5x.outputs.DOCKER_TAGS }} |
-          | ROSETTA(pax) | ${{ needs.build-rosetta-pax.outputs.DOCKER_TAGS }} |
+
           EOF
 
   retrofit-containers:
-    needs: [build-base, build-jax, build-te, build-t5x, build-pax, build-rosetta-t5x, build-rosetta-pax]
+    needs: [build-base, build-jax, build-t5x]
     if: always()
     runs-on: ubuntu-22.04
     env:
@@ -270,37 +227,18 @@ jobs:
       JAX_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAGS }}
     secrets: inherit
 
-  test-te:
-    needs: build-te
-    uses: ./.github/workflows/_test_te.yaml
-    with:
-      JAX_TE_IMAGE: ${{ needs.build-te.outputs.DOCKER_TAGS }}
-    secrets: inherit
   test-t5x:
     needs: build-t5x
     uses: ./.github/workflows/_test_t5x.yaml
     with:
       T5X_IMAGE: ${{ needs.build-t5x.outputs.DOCKER_TAGS }}
     secrets: inherit
 
-  test-pax:
-    needs: build-pax
-    uses: ./.github/workflows/_test_pax.yaml
-    with:
-      PAX_IMAGE: ${{ needs.build-pax.outputs.DOCKER_TAGS }}
-    secrets: inherit
-
-  test-vit:
-    needs: build-rosetta-t5x
-    uses: ./.github/workflows/_test_vit.yaml
-    with:
-      ROSETTA_T5X_IMAGE: ${{ needs.build-rosetta-t5x.outputs.DOCKER_TAGS }}
-    secrets: inherit 
 
   finalize:
     if: always()
     # TODO: use dynamic matrix to make dependencies self-updating
-    needs: [build-summary, test-distribution, test-jax, test-te, test-t5x, test-pax]
+    needs: [build-summary, test-distribution, test-jax, test-t5x]
     uses: ./.github/workflows/_finalize.yaml
     with:
       PUBLISH_BADGE: false