perf: enable HybridEP in H100/B200 performance recipes - #3438
Open
seonjinn wants to merge 18 commits into
Open
Conversation
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Contributor
Author
|
/ok to test 67a4b3d |
Contributor
Author
|
/ok to test 67a4b3d |
Signed-off-by: seonjinn <sna@nvidia.com>
…6-performance-recipes-20260730
Contributor
Author
|
/ok to test 0480dc6 |
…ormance-recipes-20260730 Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
Signed-off-by: seonjinn <sna@nvidia.com>
…ormance-recipes-20260730 Signed-off-by: seonjinn <sna@nvidia.com> # Conflicts: # uv.lock
Contributor
Author
|
Contributor self-review status: this recipe PR remains blocked on a clean HybridEP runtime dependency. The current |
…20260817 Signed-off-by: seonjinn <sna@nvidia.com> # Conflicts: # uv.lock
Contributor
Author
|
/ok to test 1bef174 |
Signed-off-by: seonjinn <sna@nvidia.com>
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.
Dependencies
Depends on #3436 and #2964. Merge both prerequisites before this PR.
Nemotron3 Super additionally requires a Megatron-Bridge/Megatron-LM revision containing NVIDIA/Megatron-LM#5542 and NVIDIA/Megatron-LM#6114. This PR does not vendor a custom Megatron-LM fork.
Summary
logprob_chunk_size: 1024anddefer_fp32_logits: true.H100 memory fit and performance
On the current vLLM 0.25.1 dependency stack, the previous unchunked AllToAll Qwen3-30B-A3B 4n8g recipe reproduced a policy-training OOM at step 2. Chunking at 1024 still OOMed with AllToAll at step 12, while HybridEP completed 20/20 steps. The selected HybridEP + chunk-1024 setting also outperformed the memory-safe AllToAll + chunk-512 control.
The timings below are means over 17 matched steady-state samples from 20-step H100 80 GB runs.
Relative to the memory-safe AllToAll / 512 control, the selected setting reduces E2E step time by 1.59%, increases E2E throughput by 1.56%, reduces policy time by 3.12%, and reduces logprob time by 9.64%. The matched chunk-512 A/B showed no directional regression in reward, generation-KL, JS-divergence, probability-ratio, or masked-sequence metrics.
This establishes the recipe-level memory-fit regression on the vLLM 0.25.1 stack; it does not isolate vLLM as the sole causal component.
Testing
git diff --check