wangyuandHongsheng Liu
31384cb39c
Signed-off-by: wangyu <410167048@qq.com> Signed-off-by: [Your Name] [Your Email] Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
88 lines
2.4 KiB
Bash
Executable File
88 lines
2.4 KiB
Bash
Executable File
#!/bin/bash
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# Publish Docker images from ECR staging to DockerHub.
|
|
#
|
|
# Usage:
|
|
# publish-release-images.sh # release mode (default)
|
|
# publish-release-images.sh release
|
|
# publish-release-images.sh nightly
|
|
#
|
|
# Release tags: latest, v<version>
|
|
# Nightly tags: nightly, nightly-<commit>
|
|
|
|
set -euo pipefail
|
|
|
|
MODE="${1:-release}"
|
|
|
|
DOCKERHUB_REPO="vllm/vllm-omni"
|
|
ECR_REPO="public.ecr.aws/q9t5s3a7/vllm-omni-release-repo"
|
|
COMMIT="$BUILDKITE_COMMIT"
|
|
PRIMARY_TAGS=()
|
|
EXTRA_MANIFEST_TAGS=()
|
|
|
|
case "${MODE}" in
|
|
release)
|
|
RELEASE_VERSION=$(buildkite-agent meta-data get release-version --default "" | sed 's/^v//')
|
|
if [ -z "${RELEASE_VERSION}" ]; then
|
|
echo "ERROR: release-version metadata not set"
|
|
exit 1
|
|
fi
|
|
PRIMARY_TAGS=("latest" "v${RELEASE_VERSION}")
|
|
;;
|
|
nightly)
|
|
# Commit-pinned manifest reuses nightly-{x86_64,aarch64} arch images.
|
|
PRIMARY_TAGS=("nightly")
|
|
EXTRA_MANIFEST_TAGS=("nightly-${COMMIT}")
|
|
;;
|
|
*)
|
|
echo "Usage: $0 {release|nightly}"
|
|
exit 2
|
|
;;
|
|
esac
|
|
|
|
echo "========================================"
|
|
echo "Publishing ${MODE} images to ${DOCKERHUB_REPO}"
|
|
echo " Commit: ${COMMIT}"
|
|
echo " Primary tags: ${PRIMARY_TAGS[*]}"
|
|
if [ ${#EXTRA_MANIFEST_TAGS[@]} -gt 0 ]; then
|
|
echo " Extra manifests: ${EXTRA_MANIFEST_TAGS[*]}"
|
|
fi
|
|
echo "========================================"
|
|
|
|
aws ecr-public get-login-password --region us-east-1 | \
|
|
docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7
|
|
|
|
SRC_X86="${ECR_REPO}:${COMMIT}-x86_64"
|
|
SRC_ARM="${ECR_REPO}:${COMMIT}-aarch64"
|
|
|
|
docker pull "${SRC_X86}"
|
|
docker pull "${SRC_ARM}"
|
|
|
|
publish_manifest() {
|
|
local tag="$1"
|
|
local arch_tag_base="$2"
|
|
docker manifest rm "${DOCKERHUB_REPO}:${tag}" || true
|
|
docker manifest create \
|
|
"${DOCKERHUB_REPO}:${tag}" \
|
|
"${DOCKERHUB_REPO}:${arch_tag_base}-x86_64" \
|
|
"${DOCKERHUB_REPO}:${arch_tag_base}-aarch64"
|
|
docker manifest push "${DOCKERHUB_REPO}:${tag}"
|
|
}
|
|
|
|
for tag in "${PRIMARY_TAGS[@]}"; do
|
|
docker tag "${SRC_X86}" "${DOCKERHUB_REPO}:${tag}-x86_64"
|
|
docker tag "${SRC_ARM}" "${DOCKERHUB_REPO}:${tag}-aarch64"
|
|
docker push "${DOCKERHUB_REPO}:${tag}-x86_64"
|
|
docker push "${DOCKERHUB_REPO}:${tag}-aarch64"
|
|
publish_manifest "${tag}" "${tag}"
|
|
done
|
|
|
|
ARCH_BASE="${PRIMARY_TAGS[0]}"
|
|
for tag in "${EXTRA_MANIFEST_TAGS[@]}"; do
|
|
publish_manifest "${tag}" "${ARCH_BASE}"
|
|
done
|
|
|
|
echo ""
|
|
echo "Successfully published ${MODE} images"
|