Few other places TE needed to be pinned

NVIDIA · Nov 16, 2023 · de18428 · de18428
1 parent 36157db
commit de18428
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/.github/workflows/_build_t5x.yaml b/.github/workflows/_build_t5x.yaml
@@ -32,7 +32,9 @@ on:
         type: string
         description: Git commit, tag, or branch for TE
         required: false
-        default: main
+        # TODO: This is a temporary pinning of TE as the API in TE no longer matches the TE patch
+        #   This should be reverted to main ASAP
+        default: 7976bd003fcf084dd068069b92a9a79b1743316a
     outputs:
       DOCKER_TAGS:
         description: "Tags of the image built"

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -25,7 +25,9 @@ on:
         description: 'TE source: <repo>#<branch|tag|commit>'
         type: string
         required: true
-        default: 'https://github.com/NVIDIA/TransformerEngine.git#main'
+        # TODO: This is a temporary pinning of TE as the API in TE no longer matches the TE patch
+        #   This should be reverted to main ASAP
+        default: 'https://github.com/NVIDIA/TransformerEngine.git#7976bd003fcf084dd068069b92a9a79b1743316a'
       SRC_T5X:
         description: 'T5X source: <repo>#<branch|tag|commit>'
         type: string
@@ -94,7 +96,9 @@ jobs:
           # default values are for `pull_request`` event types
           parse_git_src JAX    "${{ inputs.SRC_JAX }}"    "https://github.com/google/jax.git#main"
           parse_git_src XLA    "${{ inputs.SRC_XLA }}"    "https://github.com/openxla/xla.git#main"
-          parse_git_src TE     "${{ inputs.SRC_TE }}"     "https://github.com/NVIDIA/TransformerEngine.git#main"
+          # TODO: This is a temporary pinning of TE as the API in TE no longer matches the TE patch
+          #   This should be reverted to main ASAP
+          parse_git_src TE     "${{ inputs.SRC_TE }}"     "https://github.com/NVIDIA/TransformerEngine.git#7976bd003fcf084dd068069b92a9a79b1743316a"
           parse_git_src T5X    "${{ inputs.SRC_T5X }}"    "https://github.com/google-research/t5x.git#main"
           parse_git_src PAXML  "${{ inputs.SRC_PAXML }}"  "https://github.com/google/paxml.git#main"
           parse_git_src PRAXIS "${{ inputs.SRC_PRAXIS }}" "https://github.com/google/praxis.git#main"