Merge branch 'main' into vkozlov/fix-levanter-tests-3

NVIDIA · Oct 2, 2024 · 19e9c1d · 19e9c1d
2 parents 786a4d4 + afd9f7f
commit 19e9c1d
Show file tree

Hide file tree

Showing 4 changed files with 157 additions and 134 deletions.
diff --git a/.github/container/Dockerfile.maxtext.amd64 b/.github/container/Dockerfile.maxtext.amd64
@@ -17,13 +17,6 @@ git-clone.sh ${URLREF_MAXTEXT} ${SRC_PATH_MAXTEXT}
 echo "-r ${SRC_PATH_MAXTEXT}/requirements.txt" >> /opt/pip-tools.d/requirements-maxtext.in
 EOF
 
-###############################################################################
-## Apply patch
-###############################################################################
-
-ADD maxtext-mha.patch /opt
-RUN cd "${SRC_PATH_MAXTEXT}" && patch -p1 < /opt/maxtext-mha.patch && git diff
-
 ###############################################################################
 ## Add test script to the path
 ###############################################################################

diff --git a/.github/container/Dockerfile.maxtext.arm64 b/.github/container/Dockerfile.maxtext.arm64
@@ -58,13 +58,6 @@ git-clone.sh ${URLREF_MAXTEXT} ${SRC_PATH_MAXTEXT}
 echo "-r ${SRC_PATH_MAXTEXT}/requirements.txt" >> /opt/pip-tools.d/requirements-maxtext.in
 EOF
 
-###############################################################################
-## Apply patch
-###############################################################################
-
-ADD maxtext-mha.patch /opt
-RUN cd "${SRC_PATH_MAXTEXT}" && patch -p1 < /opt/maxtext-mha.patch && git diff
-
 ###############################################################################
 ## Add test script to the path
 ###############################################################################

diff --git a/.github/container/maxtext-mha.patch b/.github/container/maxtext-mha.patch