| 1 | #!/bin/bash |
| 2 | |
| 3 | set -ex |
| 4 | |
| 5 | # Get tag variant from argument, default to empty if not provided, should be something like "cu130". |
| 6 | # Due to limits in cleanup script, we must move variants to use separate tags like "cu130-nightly", |
| 7 | # otherwise they will be cleaned up together with the main "nightly" tags. |
| 8 | |
| 9 | TAG_VARIANT="$1" |
| 10 | if [ -n "$TAG_VARIANT" ]; then |
| 11 | ORIG_TAG_SUFFIX="-$TAG_VARIANT" |
| 12 | TAG_NAME="$TAG_VARIANT-nightly" |
| 13 | else |
| 14 | ORIG_TAG_SUFFIX="" |
| 15 | TAG_NAME="nightly" |
| 16 | fi |
| 17 | |
| 18 | ORIG_TAG_NAME="$BUILDKITE_COMMIT" |
| 19 | |
| 20 | echo "Pushing original tag $ORIG_TAG_NAME$ORIG_TAG_SUFFIX to new nightly tag name: $TAG_NAME" |
| 21 | |
| 22 | # pull original arch-dependent images from AWS ECR Public |
| 23 | aws ecr-public get-login-password --region us-east-1 | docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7 |
| 24 | docker pull public.ecr.aws/q9t5s3a7/vllm-release-repo:"$ORIG_TAG_NAME"-x86_64"$ORIG_TAG_SUFFIX" |
| 25 | docker pull public.ecr.aws/q9t5s3a7/vllm-release-repo:"$ORIG_TAG_NAME"-aarch64"$ORIG_TAG_SUFFIX" |
| 26 | # tag arch-dependent images |
| 27 | docker tag public.ecr.aws/q9t5s3a7/vllm-release-repo:"$ORIG_TAG_NAME"-x86_64"$ORIG_TAG_SUFFIX" vllm/vllm-openai:"$TAG_NAME"-x86_64 |
| 28 | docker tag public.ecr.aws/q9t5s3a7/vllm-release-repo:"$ORIG_TAG_NAME"-aarch64"$ORIG_TAG_SUFFIX" vllm/vllm-openai:"$TAG_NAME"-aarch64 |
| 29 | # push arch-dependent images to DockerHub |
| 30 | docker push vllm/vllm-openai:"$TAG_NAME"-x86_64 |
| 31 | docker push vllm/vllm-openai:"$TAG_NAME"-aarch64 |
| 32 | # push arch-independent manifest to DockerHub |
| 33 | docker manifest create vllm/vllm-openai:"$TAG_NAME" vllm/vllm-openai:"$TAG_NAME"-x86_64 vllm/vllm-openai:"$TAG_NAME"-aarch64 --amend |
| 34 | docker manifest create vllm/vllm-openai:"$TAG_NAME"-"$BUILDKITE_COMMIT" vllm/vllm-openai:"$TAG_NAME"-x86_64 vllm/vllm-openai:"$TAG_NAME"-aarch64 --amend |
| 35 | docker manifest push vllm/vllm-openai:"$TAG_NAME" |
| 36 | docker manifest push vllm/vllm-openai:"$TAG_NAME"-"$BUILDKITE_COMMIT" |