From 84665f5c4169a21fc5a50747978696e6fedec2b5 Mon Sep 17 00:00:00 2001 From: Terry Kong Date: Sun, 13 Oct 2024 00:04:10 -0700 Subject: [PATCH] ci[maxtext]: adds rosetta directory to maxtext containers - makes xla flag env files available to users Signed-off-by: Terry Kong --- .github/container/Dockerfile.maxtext.amd64 | 5 +++-- .github/container/Dockerfile.maxtext.arm64 | 3 ++- .github/workflows/_ci.yaml | 1 + 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.github/container/Dockerfile.maxtext.amd64 b/.github/container/Dockerfile.maxtext.amd64 index 8289a6099..96f06f022 100644 --- a/.github/container/Dockerfile.maxtext.amd64 +++ b/.github/container/Dockerfile.maxtext.amd64 @@ -21,7 +21,8 @@ EOF ## Add test script to the path ############################################################################### -ADD test-maxtext.sh /usr/local/bin +ADD .github/container/test-maxtext.sh /usr/local/bin +ADD rosetta /opt/rosetta ############################################################################### ## Install accumulated packages from the base image and the previous stage @@ -31,4 +32,4 @@ FROM mealkit as final RUN pip-finalize.sh -WORKDIR ${SRC_PATH_MAXTEXT} \ No newline at end of file +WORKDIR ${SRC_PATH_MAXTEXT} diff --git a/.github/container/Dockerfile.maxtext.arm64 b/.github/container/Dockerfile.maxtext.arm64 index bd64c5b6d..2ff52cc4a 100644 --- a/.github/container/Dockerfile.maxtext.arm64 +++ b/.github/container/Dockerfile.maxtext.arm64 @@ -62,7 +62,8 @@ EOF ## Add test script to the path ############################################################################### -ADD test-maxtext.sh /usr/local/bin +ADD .github/container/test-maxtext.sh /usr/local/bin +ADD rosetta /opt/rosetta ############################################################################### ## Install accumulated packages from the base image and the previous stage diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml index 426764323..91089d9d6 100644 --- a/.github/workflows/_ci.yaml +++ b/.github/workflows/_ci.yaml @@ -108,6 +108,7 @@ jobs: BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }} CONTAINER_NAME: maxtext DOCKERFILE: .github/container/Dockerfile.maxtext.${{ inputs.ARCHITECTURE }} + DOCKER_CONTEXT: . EXTRA_BUILD_ARGS: | URLREF_MAXTEXT=${{ fromJson(inputs.SOURCE_URLREFS).MAXTEXT }} secrets: inherit