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 5, 22:15 UTC
[ROCm] [Bugfix] [Release] Fix nightly rocm release pipeline (#38263)
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
TJian committed
Mar 27, 2026 at 02:47 UTC
bc9c6fbbe655072a0ff924a44b1069bd11ba0772
1 file changed
+2
-33
.buildkite/release-pipeline.yaml
+2
-33
@@ -402,38 +402,7 @@ steps:
402
# Save ECR tag for downstream jobs
403
buildkite-agent meta-data set "rocm-base-image-tag" "$${ECR_CACHE_TAG}"
404
405
- # Scenario 2: Image cached but wheels missing
406
- elif [ "$${IMAGE_EXISTS}" = "true" ] && [ "$${WHEELS_EXIST}" = "false" ]; then
407
- echo ""
408
- echo "PARTIAL CACHE - Image exists, extracting wheels..."
409
- echo ""
410
-
411
- docker pull "$${ECR_CACHE_TAG}"
412
-
413
- # Rebuild wheel extraction stage
414
- DOCKER_BUILDKIT=1 docker buildx build \
415
- --file docker/Dockerfile.rocm_base \
416
- --tag rocm-base-debs:$${BUILDKITE_BUILD_NUMBER} \
417
- --build-arg BASE_IMAGE="$${ECR_IMAGE_TAG}" \
418
- --target debs_wheel_release \
419
- --build-arg USE_SCCACHE=1 \
420
- --build-arg SCCACHE_BUCKET_NAME=vllm-build-sccache \
421
- --build-arg SCCACHE_REGION_NAME=us-west-2 \
422
- --build-arg SCCACHE_S3_NO_CREDENTIALS=0 \
423
- --load \
424
- .
425
-
426
- # Extract and upload wheels
427
- mkdir -p artifacts/rocm-base-wheels
428
- cid=$(docker create rocm-base-debs:$${BUILDKITE_BUILD_NUMBER})
429
- docker cp $${cid}:/app/debs/. artifacts/rocm-base-wheels/
430
- docker rm $${cid}
431
-
432
- .buildkite/scripts/cache-rocm-base-wheels.sh upload
433
-
434
- buildkite-agent meta-data set "rocm-base-image-tag" "$${ECR_CACHE_TAG}"
435
-
436
- # Scenario 3: Full rebuild needed
405
+ # Scenario 2: Full rebuild needed
406
else
407
echo ""
408
echo " CACHE MISS - Building from scratch..."
@@ -453,7 +422,7 @@ steps:
422
# Build wheel extraction stage
423
DOCKER_BUILDKIT=1 docker buildx build \
424
--file docker/Dockerfile.rocm_base \
456
- --tag rocm-base-debs:${BUILDKITE_BUILD_NUMBER} \
425
+ --tag rocm-base-debs:$${BUILDKITE_BUILD_NUMBER} \
426
--target debs_wheel_release \
427
--build-arg USE_SCCACHE=1 \
428
--build-arg SCCACHE_BUCKET_NAME=vllm-build-sccache \