Repository navigation
test(vllm): cover Qwen3-TTS on the FlashInfer attention backend - #18257
Draft
JulienDarve wants to merge 1 commit into
Draft
JulienDarve wants to merge 1 commit into
JulienDarve wants to merge 1 commit into
Conversation
vLLM picks FlashInfer by default only on B200/GB200, so CI's L4 runners never exercised the FlashInfer-only warmup where Qwen3-TTS failed to start with vLLM 0.32.0 and vLLM-Omni 0.32.0rc1. Add a post-merge case that copies vLLM-Omni's Qwen3-TTS stage config with every stage set to FlashInfer and checks that the FlashInfer warmup ran. Signed-off-by: Julien Darve <jdarve@NVIDIA.com>
Contributor
✅ Dynamo PR CI passed — run 42102428871 (attempt 1) on
|
This branch has not been deployed
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.
Overview
Qwen3-TTS through vLLM-Omni failed to start on B200 and GB200 because a vLLM/vLLM-Omni version pairing broke only when vLLM uses the FlashInfer attention backend. CI never caught it: every CI GPU makes vLLM choose FlashAttention, so the existing
omni_audiocase passed on the broken versions. This PR adds a post-merge case that forces FlashInfer, so the next pairing that breaks it fails CI instead of release QA.Key Changes
0 lines of core logic; test-only.
tests/serve/test_vllm_omni.py: newomni_audio_flashinfercase (gpu_1,post_merge). It runs the same Qwen3-TTS launcher and request asomni_audio, but points the worker at a copy of vLLM-Omni's installedqwen3_tts.yamlwith every stage set toattention_config.backend: FLASHINFER(passed throughDYN_OMNI_STAGE_CONFIGS_PATH). The copy is written totmp_pathat test time, so Omni's own memory, connector and runner settings stay current.Warming up FlashInfer attentionlog line. That is the FlashInfer-only warmup where the bug occurred, and the check keeps the case from passing silently if the override stops taking effect.The Dynamo Omni worker forwards no attention-backend flag to vLLM-Omni, and vLLM 0.33 removed
VLLM_ATTENTION_BACKEND, so the stage config file is the only override available.Validation
Ran
test_omni_serve_deployment[omni_audio_flashinfer]from this branch on one NVIDIA RTX 6000 Ada GPU, inside the Dynamo 1.7.0 CI vLLM runtime image. The RTX 6000 Ada has compute capability 8.9, the same as CI's L4, where vLLM picks FlashAttention by default.maintodayWarming up FlashInfer attention., then raisesValueError: Missing Qwen3-TTS text conditioning(143 s)container/deps/vllm/install_vllm_omni.sh)1 passed in 141.57s)--collect-only)omni_audio; the post-merge and nightly vLLM 1-GPU filters select both cases;vllm and xpu_1selects onlyomni_audioThe case takes about the same time as
omni_audio, and it uses the same 1200 s timeout to cover model download and FlashInfer JIT compilation on a cold runner.Not yet run in CI: post-merge cases do not run on PRs, and on today's
mainpin this case fails by design.Issues and Ongoing Work
mainpin this case fails by design, which would turn post-merge red.textor precomputed text token ids. vllm-project/vllm-omni#9233).tests/serve/test_vllm_omni.pyor adds FlashInfer coverage for Omni.Pull Request Checklist
/pr-checklistskill.🤖 Generated with Claude Code