Skip to content

[CI] Solidify speculative decoding E2E coverage - #50713

Merged
mgoin merged 12 commits into
vllm-project:mainfrom
ROCm:akaratza_solidify_spec_dec
Aug 11, 2026
Merged

mgoin merged 12 commits into
vllm-project:mainfrom
ROCm:akaratza_solidify_spec_dec

Conversation

@AndreasKaratzas

Copy link
Copy Markdown
Member
  • Standardize the reorganized spec-decode tests on vllm_runner for consistent setup, defaults, and cleanup.
  • Add AMD CI mirrors for DFlash and DSpark, including NVFP4 targets through ROCm emulation.
  • Make prompt selection deterministic, remove unnecessary CI progress output, and select platform-appropriate backends.
  • Strengthen acceptance thresholds, expected-failure handling, and failure diagnostics.

This is a follow-up on #50330. It applies some NITS across the new spec-decode area while hardening deterministic setup and actionable failure reporting.

Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
@mergify

mergify Bot commented Aug 2, 2026

Copy link
Copy Markdown
Contributor

Documentation preview: https://vllm--50713.org.readthedocs.build/en/50713/

@mergify mergify Bot added documentation Improvements or additions to documentation ci/build labels Aug 2, 2026
AndreasKaratzas and others added 3 commits August 2, 2026 04:55
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@AndreasKaratzas AndreasKaratzas added the ready ONLY add when PR is ready to merge/full CI is needed label Aug 2, 2026
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Comment on lines +49 to +50
# GSM8K is text-only; avoid profiling unused multimodal towers.
limit_mm_per_prompt={"image": 0, "audio": 0, "video": 0},

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nit: we can just use language_model_only=True

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done :)

Comment on lines +35 to +47
@@ -43,10 +44,14 @@ def test_gemma4_dspark_correctness_and_acceptance_rate(
max_num_seqs=32,
gpu_memory_utilization=0.85,
enforce_eager=True,
enable_chunked_prefill=None,

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why do we need to set block_size and enable_chunked_prefill? These should be fine as their defaults

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actually VllmRunner defaults these to 16 and False, unlike LLM construction. Removing them would change behavior and disable Gemma4's supported chunked prefill.

@AndreasKaratzas

Copy link
Copy Markdown
Member Author

/ci run

@AndreasKaratzas AndreasKaratzas added the rocm Related to AMD ROCm label Aug 5, 2026
@github-project-automation github-project-automation Bot moved this to Todo in AMD Aug 5, 2026
@github-actions

github-actions Bot commented Aug 5, 2026

Copy link
Copy Markdown

✅ Triggered Buildkite CI #82443 for commit cb730347fdb3.

@AndreasKaratzas

Copy link
Copy Markdown
Member Author

/ci run

@github-actions

github-actions Bot commented Aug 5, 2026

Copy link
Copy Markdown

✅ Triggered Buildkite CI #82541 for commit 7a70dc2fcf8a.

@mergify

mergify Bot commented Aug 6, 2026

Copy link
Copy Markdown
Contributor

This pull request has merge conflicts that must be resolved before it can be
merged. Please rebase the PR, @AndreasKaratzas.

https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

@mergify mergify Bot added the needs-rebase label Aug 6, 2026
Retain the PR vllm_runner lifecycle and acceptance diagnostics while incorporating the async-scheduling fix from main.

Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
@AndreasKaratzas

Copy link
Copy Markdown
Member Author

/ci run

@github-actions

github-actions Bot commented Aug 7, 2026

Copy link
Copy Markdown

✅ Triggered Buildkite CI #82804 for commit d4b3244d6de6.

@mergify mergify Bot removed the needs-rebase label Aug 7, 2026
Retain the PR's vllm_runner lifecycle, ROCm-safe dSpark setup, and diagnostic assertions while incorporating the Qwen3.6 dSpark acceptance coverage from main.

Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
@AndreasKaratzas

Copy link
Copy Markdown
Member Author

/ci run

@github-actions

github-actions Bot commented Aug 7, 2026

Copy link
Copy Markdown

✅ Triggered Buildkite CI #82930 for commit d88663bc869e.

@mergify

mergify Bot commented Aug 8, 2026

Copy link
Copy Markdown
Contributor

This pull request has merge conflicts that must be resolved before it can be
merged. Please rebase the PR, @AndreasKaratzas.

https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

@mergify mergify Bot added the needs-rebase label Aug 8, 2026
Retain the PR's vllm_runner lifecycle and diagnostic assertions while incorporating hybrid Model Runner V2 MTP coverage from main.

Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
@AndreasKaratzas

Copy link
Copy Markdown
Member Author

/ci run

@github-actions

github-actions Bot commented Aug 8, 2026

Copy link
Copy Markdown

✅ Triggered Buildkite CI #83023 for commit 4281daf4958c.

@mergify mergify Bot removed the needs-rebase label Aug 8, 2026
@mgoin
mgoin merged commit 1a17273 into vllm-project:main Aug 11, 2026
105 checks passed
@github-project-automation github-project-automation Bot moved this from Todo to Done in AMD Aug 11, 2026
@AndreasKaratzas
AndreasKaratzas deleted the akaratza_solidify_spec_dec branch August 11, 2026 02:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci/build documentation Improvements or additions to documentation ready ONLY add when PR is ready to merge/full CI is needed rocm Related to AMD ROCm speculative-decoding

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

2 participants