Repository navigation
Fix FlashInfer A2A dispatcher during CUDA graph capture - #29461
Merged
Merged
Conversation
merrymercy
requested review from
BBuf,
Edwardf0t1,
Fridge003,
HaiShaw,
Ying1123,
ch-wan and
ispobock
as code owners
June 26, 2026 21:52
Contributor
|
Warning You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again! |
Contributor
Author
|
/tag-and-rerun-ci |
…2A dispatcher During CUDA graph capture, all EP ranks use the same capture batch size, so x.shape[0] is uniform and safe to use as runtime_max_tokens_per_rank. Gate the EP>1 branch on `not get_is_capture_mode()` so it only fires during live inference where EP ranks may have different token counts. Use get_is_capture_mode() rather than is_in_breakable_cuda_graph() to avoid regressing breakable graph replay, where EP ranks could select different graph sizes when require_mlp_tp_gather is False.
merrymercy
force-pushed
the
lmzheng/fix-flashinfer-a2a
branch
from
June 28, 2026 04:45
3a7559a to
5c84459
Compare
2 of 5 tasks
This was referenced Jun 28, 2026
2 of 5 tasks
4 tasks
Chronostasys
pushed a commit
to MindLab-Research/sglang
that referenced
this pull request
Aug 24, 2026
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
1 participant
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.
Summary
not is_graph_capture(combiningget_is_capture_mode()andis_in_breakable_cuda_graph()).Original commits
d3abe7396CI States
Latest PR Test (Base): ✅ Run #28311737180
Latest PR Test (Extra): ❌ Run #28311737102