Skip to content

Fix FlashInfer NVLS routing buffer race - #7263

Open
wdykas wants to merge 1 commit into
NVIDIA:mainfrom
wdykas:fix-flashinfer-nvls-routing-race
Open

Fix FlashInfer NVLS routing buffer race#7263
wdykas wants to merge 1 commit into
NVIDIA:mainfrom
wdykas:fix-flashinfer-nvls-routing-race

Conversation

@wdykas

@wdykas wdykas commented Sep 11, 2026

Copy link
Copy Markdown
Contributor

What does this PR do?

Clear the FlashInfer NVLS routing buffer before the fused metadata synchronization fence. Clearing it afterwards can race with an early peer all-gather write and replace valid routing entries with -1.

This fix applies to the ordinary FlashInfer + NVLS inference path and is independent of batch-invariant mode.

Testing

  • Focused unit regression: 1 passed
  • Black, isort, and Ruff checks passed for both changed files

Signed-off-by: wdykas <wdykas@nvidia.com>
@copy-pr-bot

copy-pr-bot Bot commented Sep 11, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@wdykas
wdykas marked this pull request as ready for review September 11, 2026 23:04
@wdykas
wdykas requested review from a team as code owners September 11, 2026 23:04
@wdykas

wdykas commented Sep 11, 2026

Copy link
Copy Markdown
Contributor Author

/ok to test 2878c0d

@wdykas
wdykas marked this pull request as draft September 11, 2026 23:08
@wdykas
wdykas marked this pull request as ready for review September 11, 2026 23:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant