@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / 8c4fc4202a
Fork 0

A high-throughput and memory-efficient inference and serving engine for LLMs

Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git

Synced Nov 3, 19:00 UTC

[CI] Inline build artifact annotations in release pipeline (#42357)

Signed-off-by: khluu <khluu000@gmail.com>

Kevin H. Luu committed May 12, 2026 at 15:57 UTC 8c4fc4202a6cd8bd8141e21be5be62eb7b896b24
3 files changed +31 -38
.buildkite/release-pipeline.yaml
+22 -11
@@ -28,6 +28,7 @@ steps:
28 - "mkdir artifacts"
29 - "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
30 - "bash .buildkite/scripts/upload-nightly-wheels.sh"
31 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "s3://vllm-wheels/$$BUILDKITE_COMMIT/$(cd artifacts/dist && echo *.whl)"'
32 env:
33 DOCKER_BUILDKIT: "1"
34
@@ -41,6 +42,7 @@ steps:
42 - "mkdir artifacts"
43 - "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
44 - "bash .buildkite/scripts/upload-nightly-wheels.sh"
45 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "s3://vllm-wheels/$$BUILDKITE_COMMIT/$(cd artifacts/dist && echo *.whl)"'
46 env:
47 DOCKER_BUILDKIT: "1"
48
@@ -54,6 +56,7 @@ steps:
56 - "mkdir artifacts"
57 - "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
58 - "bash .buildkite/scripts/upload-nightly-wheels.sh"
59 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "s3://vllm-wheels/$$BUILDKITE_COMMIT/$(cd artifacts/dist && echo *.whl)"'
60 env:
61 DOCKER_BUILDKIT: "1"
62
@@ -67,6 +70,7 @@ steps:
70 - "mkdir artifacts"
71 - "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
72 - "bash .buildkite/scripts/upload-nightly-wheels.sh"
73 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "s3://vllm-wheels/$$BUILDKITE_COMMIT/$(cd artifacts/dist && echo *.whl)"'
74 env:
75 DOCKER_BUILDKIT: "1"
76
@@ -80,6 +84,7 @@ steps:
84 - "mkdir artifacts"
85 - "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
86 - "bash .buildkite/scripts/upload-nightly-wheels.sh"
87 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "s3://vllm-wheels/$$BUILDKITE_COMMIT/$(cd artifacts/dist && echo *.whl)"'
88 env:
89 DOCKER_BUILDKIT: "1"
90
@@ -93,6 +98,7 @@ steps:
98 - "mkdir artifacts"
99 - "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
100 - "bash .buildkite/scripts/upload-nightly-wheels.sh"
101 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "s3://vllm-wheels/$$BUILDKITE_COMMIT/$(cd artifacts/dist && echo *.whl)"'
102 env:
103 DOCKER_BUILDKIT: "1"
104
@@ -138,6 +144,7 @@ steps:
144 # re-tag to default image tag and push, just in case arm64 build fails
145 - "docker tag public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m) public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT"
146 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT"
147 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)"'
148
149 - label: "Build release image - aarch64 - CUDA 13.0"
150 depends_on: ~
@@ -160,6 +167,7 @@ steps:
167 --progress plain \
168 -f docker/Dockerfile .
169 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)"
170 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)"'
171
172 - label: "Build release image - x86_64 - CUDA 12.9"
173 depends_on: ~
@@ -184,6 +192,7 @@ steps:
192 # re-tag to default image tag and push, just in case arm64 build fails
193 - "docker tag public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129"
194 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129"
195 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129"'
196
197 - label: "Build release image - aarch64 - CUDA 12.9"
198 depends_on: ~
@@ -205,6 +214,7 @@ steps:
214 --progress plain \
215 -f docker/Dockerfile .
216 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129"
217 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129"'
218
219 - label: "Build release image - x86_64 - CUDA 13.0 - Ubuntu 24.04"
220 depends_on: ~
@@ -231,6 +241,7 @@ steps:
241 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-ubuntu2404"
242 - "docker tag public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-ubuntu2404 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-ubuntu2404"
243 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-ubuntu2404"
244 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-ubuntu2404"'
245
246 - label: "Build release image - aarch64 - CUDA 13.0 - Ubuntu 24.04"
247 depends_on: ~
@@ -255,6 +266,7 @@ steps:
266 --progress plain \
267 -f docker/Dockerfile .
268 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-ubuntu2404"
269 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-ubuntu2404"'
270
271 - label: "Build release image - x86_64 - CUDA 12.9 - Ubuntu 24.04"
272 depends_on: ~
@@ -280,6 +292,7 @@ steps:
292 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129-ubuntu2404"
293 - "docker tag public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129-ubuntu2404 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129-ubuntu2404"
294 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129-ubuntu2404"
295 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129-ubuntu2404"'
296
297 - label: "Build release image - aarch64 - CUDA 12.9 - Ubuntu 24.04"
298 depends_on: ~
@@ -303,6 +316,7 @@ steps:
316 --progress plain \
317 -f docker/Dockerfile .
318 - "docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129-ubuntu2404"
319 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-$(uname -m)-cu129-ubuntu2404"'
320
321 - block: "Build release image for x86_64 CPU"
322 key: block-cpu-release-image-build
@@ -320,6 +334,7 @@ steps:
334 - "DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg GIT_REPO_CHECK=1 --build-arg VLLM_CPU_X86=true --tag public.ecr.aws/q9t5s3a7/vllm-cpu-release-repo:$(buildkite-agent meta-data get release-version) --tag public.ecr.aws/q9t5s3a7/vllm-cpu-release-repo:latest --progress plain --target vllm-openai -f docker/Dockerfile.cpu ."
335 - "docker push public.ecr.aws/q9t5s3a7/vllm-cpu-release-repo:latest"
336 - "docker push public.ecr.aws/q9t5s3a7/vllm-cpu-release-repo:$(buildkite-agent meta-data get release-version)"
337 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-cpu-release-repo:$(buildkite-agent meta-data get release-version)"'
338 env:
339 DOCKER_BUILDKIT: "1"
340
@@ -339,6 +354,7 @@ steps:
354 - "DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg GIT_REPO_CHECK=1 --tag public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:$(buildkite-agent meta-data get release-version) --tag public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:latest --progress plain --target vllm-openai -f docker/Dockerfile.cpu ."
355 - "docker push public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:latest"
356 - "docker push public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:$(buildkite-agent meta-data get release-version)"
357 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "$$BUILDKITE_LABEL" "public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:$(buildkite-agent meta-data get release-version)"'
358 env:
359 DOCKER_BUILDKIT: "1"
360
@@ -356,15 +372,7 @@ steps:
372 - "aws ecr-public get-login-password --region us-east-1 | docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7"
373 - "docker manifest create public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-x86_64 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-aarch64 --amend"
374 - "docker manifest push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT"
359 -
360 - - label: "Annotate release workflow - CUDA 13.0"
361 - depends_on:
362 - - create-multi-arch-manifest
363 - id: annotate-release-workflow
364 - agents:
365 - queue: small_cpu_queue_release
366 - commands:
367 - - "bash .buildkite/scripts/annotate-release.sh"
375 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "Manifest: CUDA 13.0" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT"'
376
377 - label: "Create multi-arch manifest - CUDA 12.9"
378 depends_on:
@@ -377,6 +385,7 @@ steps:
385 - "aws ecr-public get-login-password --region us-east-1 | docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7"
386 - "docker manifest create public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-x86_64-cu129 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-aarch64-cu129 --amend"
387 - "docker manifest push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129"
388 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "Manifest: CUDA 12.9" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129"'
389
390 - label: "Create multi-arch manifest - CUDA 13.0 - Ubuntu 24.04"
391 depends_on:
@@ -389,6 +398,7 @@ steps:
398 - "aws ecr-public get-login-password --region us-east-1 | docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7"
399 - "docker manifest create public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-ubuntu2404 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-x86_64-ubuntu2404 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-aarch64-ubuntu2404 --amend"
400 - "docker manifest push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-ubuntu2404"
401 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "Manifest: CUDA 13.0 Ubuntu 24.04" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-ubuntu2404"'
402
403 - label: "Create multi-arch manifest - CUDA 12.9 - Ubuntu 24.04"
404 depends_on:
@@ -401,6 +411,7 @@ steps:
411 - "aws ecr-public get-login-password --region us-east-1 | docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7"
412 - "docker manifest create public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129-ubuntu2404 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-x86_64-cu129-ubuntu2404 public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-aarch64-cu129-ubuntu2404 --amend"
413 - "docker manifest push public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129-ubuntu2404"
414 + - 'bash .buildkite/scripts/annotate-build-artifact.sh "Manifest: CUDA 12.9 Ubuntu 24.04" "public.ecr.aws/q9t5s3a7/vllm-release-repo:$BUILDKITE_COMMIT-cu129-ubuntu2404"'
415
416 - label: "Publish nightly multi-arch image to DockerHub"
417 depends_on:
@@ -551,7 +562,7 @@ steps:
562 echo ""
563 echo " Build complete - Image and wheels cached"
564 fi
554 -
565 +
566 artifact_paths:
567 - "artifacts/rocm-base-wheels/*.whl"
568 env:
@@ -767,7 +778,7 @@ steps:
778
779 # Push to ECR
780 docker push public.ecr.aws/q9t5s3a7/vllm-release-repo:$${BUILDKITE_COMMIT}-rocm
770 -
781 +
782 echo ""
783 echo " Successfully built and pushed ROCm release image"
784 echo " Image: public.ecr.aws/q9t5s3a7/vllm-release-repo:$${BUILDKITE_COMMIT}-rocm"
.buildkite/scripts/annotate-build-artifact.sh new
+9
@@ -0,0 +1,9 @@
1 +#!/bin/bash
2 +# SPDX-License-Identifier: Apache-2.0
3 +# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
4 +#
5 +# Append a build artifact line to the Buildkite annotation.
6 +# Usage: annotate-build-artifact.sh <label> <value>
7 +set -e
8 +echo "- **${1}**: \`${2}\`" | \
9 + buildkite-agent annotate --append --style 'info' --context 'release-artifacts'
.buildkite/scripts/annotate-release.sh deleted
-27
@@ -1,27 +0,0 @@
1 -#!/bin/bash
2 -
3 -set -ex
4 -
5 -# Get release version, default to 1.0.0.dev for nightly/per-commit builds
6 -RELEASE_VERSION=$(buildkite-agent meta-data get release-version 2>/dev/null | sed 's/^v//')
7 -if [ -z "${RELEASE_VERSION}" ]; then
8 - RELEASE_VERSION="1.0.0.dev"
9 -fi
10 -
11 -buildkite-agent annotate --style 'info' --context 'release-workflow' << EOF
12 -To download the wheel (by commit):
13 -\`\`\`
14 -aws s3 cp s3://vllm-wheels/${BUILDKITE_COMMIT}/vllm-${RELEASE_VERSION}-cp38-abi3-manylinux_2_35_x86_64.whl .
15 -aws s3 cp s3://vllm-wheels/${BUILDKITE_COMMIT}/vllm-${RELEASE_VERSION}-cp38-abi3-manylinux_2_35_aarch64.whl .
16 -
17 -(Optional) For CUDA 12.9:
18 -aws s3 cp s3://vllm-wheels/${BUILDKITE_COMMIT}/vllm-${RELEASE_VERSION}+cu129-cp38-abi3-manylinux_2_31_x86_64.whl .
19 -aws s3 cp s3://vllm-wheels/${BUILDKITE_COMMIT}/vllm-${RELEASE_VERSION}+cu129-cp38-abi3-manylinux_2_31_aarch64.whl .
20 -
21 -(Optional) For CPU:
22 -aws s3 cp s3://vllm-wheels/${BUILDKITE_COMMIT}/vllm-${RELEASE_VERSION}+cpu-cp38-abi3-manylinux_2_35_x86_64.whl .
23 -aws s3 cp s3://vllm-wheels/${BUILDKITE_COMMIT}/vllm-${RELEASE_VERSION}+cpu-cp38-abi3-manylinux_2_35_aarch64.whl .
24 -\`\`\`
25 -
26 -Docker images are published automatically by the "Publish release images to DockerHub" pipeline step.
27 -EOF