diff --git a/.buildkite/release-pipeline.yaml b/.buildkite/release-pipeline.yaml index 8fc66793a..45b2996f7 100644 --- a/.buildkite/release-pipeline.yaml +++ b/.buildkite/release-pipeline.yaml @@ -611,7 +611,7 @@ steps: - "bash tools/vllm-rocm/generate-rocm-wheels-root-index.sh" env: S3_BUCKET: "vllm-wheels" - VARIANT: "rocm700" + VARIANT: "rocm721" # ROCm Job 6: Build ROCm Release Docker Image - label: ":docker: Build release image - x86_64 - ROCm" @@ -681,6 +681,7 @@ steps: - label: "Publish nightly ROCm image to DockerHub" depends_on: - build-rocm-release-image + if: build.env("NIGHTLY") == "1" agents: queue: small_cpu_queue_release commands: diff --git a/tools/vllm-rocm/generate-rocm-wheels-root-index.sh b/tools/vllm-rocm/generate-rocm-wheels-root-index.sh index 87b5c3228..650a71937 100755 --- a/tools/vllm-rocm/generate-rocm-wheels-root-index.sh +++ b/tools/vllm-rocm/generate-rocm-wheels-root-index.sh @@ -17,14 +17,14 @@ # # Environment variables: # S3_BUCKET - Bucket name (default: vllm-wheels) -# VARIANT - ROCm variant (default: rocm700) +# VARIANT - ROCm variant (default: rocm721) # DRY_RUN - Set to 1 for preview mode (same as --dry-run) set -euo pipefail # ======== Configuration ======== BUCKET="${S3_BUCKET:-vllm-wheels}" -VARIANT="${VARIANT:-rocm700}" +VARIANT="${VARIANT:-rocm721}" DRY_RUN="${DRY_RUN:-0}" FORCE_VERSION=""