[CI][ROCm] Avoid redundant image pulls during smoke validation - #54408
Merged
Merged
Conversation
Run ROCm image smoke checks inside the shared BuildKit graph and export a build-scoped marker. Preserve the Docker fallback for explicit image validation while avoiding the cold cross-daemon pull on ephemeral workers. Co-authored-by: OpenAI Codex <codex@openai.com> Signed-off-by: Andreas Karatzas <akaratza@amd.com>
AndreasKaratzas
requested review from
Harry-Chen,
dllehr-amd,
khluu and
tjtanaa
as code owners
August 30, 2026 21:52
There was a problem hiding this comment.
Claude Code Review
This repository is configured for manual code reviews. Comment @claude review for a one-time review, or @claude review always to subscribe this PR to a review on every future push.
Tip: disable this comment in your organization's Code Review settings.
Member
Author
|
/ci run |
|
✅ Triggered Buildkite CI #86257 for commit |
shen-shanshan
approved these changes
Sep 1, 2026
am-cohere
pushed a commit
to am-cohere/vllm
that referenced
this pull request
Sep 1, 2026
…project#54408) Signed-off-by: Andreas Karatzas <akaratza@amd.com> Co-authored-by: OpenAI Codex <codex@openai.com>
mylibrar
pushed a commit
to tanyuqian/vllm
that referenced
this pull request
Sep 3, 2026
…project#54408) Signed-off-by: Andreas Karatzas <akaratza@amd.com> Co-authored-by: OpenAI Codex <codex@openai.com>
sheralskumar
pushed a commit
to sheralskumar/vllm
that referenced
this pull request
Sep 8, 2026
…project#54408) Signed-off-by: Andreas Karatzas <akaratza@amd.com> Co-authored-by: OpenAI Codex <codex@openai.com>
D-G-Dimitrov
pushed a commit
to D-G-Dimitrov/vllm
that referenced
this pull request
Sep 8, 2026
…project#54408) Signed-off-by: Andreas Karatzas <akaratza@amd.com> Co-authored-by: OpenAI Codex <codex@openai.com> (cherry picked from commit 907b1a7)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
The linked AMD CI build 12440 image-build retry took 13m54s, including 339 seconds to cold-pull roughly 13 GB into host Docker even though the actual smoke checks took only 4.6 seconds. Older runs on persistent agents reused 33 of 39 layers and completed the same pull in 17–23 seconds, while the one-job Kubernetes workers observed from August 25 started with 0 of 41 host-Docker layers. The pull-based smoke path originated in #46904 and was retained while equivalent-image reuse expanded in #48646, so those PRs are relevant design origins but did not cause the abrupt August regression by themselves. Within the last two weeks, ROCm base refreshes in #53182 and #53712, plus version-sensitive Rust metadata from #52593, amplified cold-cache work; however, #53712 landed after the first slow Kubernetes run and the measured Rust work was only tens of seconds, so these were contributors rather than the sole cause. The timing transition instead aligns with BuildKit and host Docker using separate stores on fresh per-job workers, which forces the complete image graph to be downloaded a second time after BuildKit has built and pushed it. This change runs the same structural and import checks in a descendant BuildKit target and exports a build-scoped marker, eliminating the redundant transfer while retaining a Docker fallback for standalone or explicitly overridden image validation.
test_smokeDockerfile descendant that runs the existing checks without network access and exports only a tiny marker through a scratch stage, leaving the pushedtestimage unchanged.smoke-test-rocm-cito the existing Bake group with the same ROCm cache inputs and a local marker output scoped to the current Buildkite build.AI assistance
AI assistance from OpenAI Codex was used for Buildkite log analysis, implementation, testing, review, and PR drafting. The human submitter must review every changed line, rerun or confirm the relevant checks, and be able to explain and defend the change end-to-end before marking the PR ready for review.