From 6fee7c11d96fd5d664f44a259167dd2b42fd7923 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Sat, 20 Apr 2024 21:36:23 +0200 Subject: [PATCH] Update rosetta/Dockerfile.maxtext Co-authored-by: Terry Kong --- .github/workflows/_ci.yaml | 4 +++- rosetta/Dockerfile.maxtext | 4 +--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml index 8c7ef1f4f..1295f5bc6 100644 --- a/.github/workflows/_ci.yaml +++ b/.github/workflows/_ci.yaml @@ -86,7 +86,7 @@ jobs: BADGE_FILENAME: badge-maxtext-build BUILD_DATE: ${{ inputs.BUILD_DATE }} BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }} - CONTAINER_NAME: maxtext + CONTAINER_NAME: upstream-maxtext DOCKERFILE: .github/container/Dockerfile.maxtext.amd64 secrets: inherit @@ -184,8 +184,10 @@ jobs: - build-levanter - build-upstream-t5x - build-upstream-pax + - build-upstream-maxtext - build-rosetta-t5x - build-rosetta-pax + - build-rosetta-maxtext - build-grok outputs: TAGS: ${{ steps.collect-tags.outputs.TAGS }} diff --git a/rosetta/Dockerfile.maxtext b/rosetta/Dockerfile.maxtext index 936fbefa1..6c69ed93a 100644 --- a/rosetta/Dockerfile.maxtext +++ b/rosetta/Dockerfile.maxtext @@ -1,5 +1,5 @@ # syntax=docker/dockerfile:1-labs -ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-pax +ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-maxtext ARG GIT_USER_EMAIL=jax@nvidia.com ARG GIT_USER_NAME=NVIDIA # If set to "true", then will pull new local patches, the manifest.yaml and create-distribution.sh (in case it was updated). @@ -35,8 +35,6 @@ if [[ "${UPDATE_PATCHES}" == "true" ]]; then cp -r /mnt/jax-toolbox/.github/container/patches ${MANIFEST_DIR}/ cp /mnt/jax-toolbox/.github/container/manifest.yaml ${MANIFEST_DIR}/manifest.yaml cp /mnt/jax-toolbox/.github/container/create-distribution.sh ${MANIFEST_DIR}/create-distribution.sh - # TODO: remove - cp /mnt/jax-toolbox/.github/container/pip-finalize.sh /usr/local/bin/ fi cp -r /mnt/jax-toolbox/rosetta /opt/rosetta