Skip to content

Update B200 Dsv4 configs#1655

Merged
Oseltamivir merged 8 commits into
mainfrom
wzhao/dsv4-b200-eplb
Jun 5, 2026
Merged

Update B200 Dsv4 configs#1655
Oseltamivir merged 8 commits into
mainfrom
wzhao/dsv4-b200-eplb

Conversation

@wzhao18
Copy link
Copy Markdown
Collaborator

@wzhao18 wzhao18 commented Jun 3, 2026

Note

Low Risk
Benchmark-only image and vLLM serve-flag changes; no production auth or application logic.

Overview
Updates DeepSeek-V4 FP4 B200 vLLM fixed-seq-len benchmarks to a pinned vLLM nightly image and turns on expert-parallel load balancing (EPLB) whenever DP attention (DEP) is active.

For DP_ATTENTION=true, dsv4_fp4_b200_vllm.sh now passes --enable-eplb with an NCCL communicator config alongside the existing deep_gemm_mega_moe MoE backend. perf-changelog.yaml documents this under dsv4-fp4-b200-vllm.

Reviewed by Cursor Bugbot for commit e1b0750. Bugbot is set up for automated code reviews on this repo. Configure here.

@wzhao18 wzhao18 requested a review from a team June 3, 2026 16:46
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 3, 2026

Thanks for the contribution! For vLLM & SGLang, please ensure that your recipes is similar to the official vLLM recipes and/or the SGLang cookbook

If it is not, please create a PR first before we can merge your single node PR into the master branch. Let's ensure that the documentation is first class such that the entire ML community can benefit from your hard work! Thank you

PR authors are responsible for ensuring that after merging, all GitHub Action jobs fully pass. A lot of the time, failures are just flakes and simply re-running the failed jobs will fix it. If re-running failed jobs is attempted, PR authors are responsible for ensuring it passes. See GitHub's docs on re-running failed jobs: https://docs.github.com/en/actions/how-tos/manage-workflow-runs/re-run-workflows-and-jobs#re-running-failed-jobs-in-a-workflow

As a rule of thumb, generally, PR authors should request a review & get a PR approval from the respective companies' CODEOWNERS before requesting a review from core maintainers.

If additional help is needed, PR authors can reach out to core maintainers over Slack.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 3, 2026

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 3, 2026

1 similar comment
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 3, 2026

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 3, 2026

wzhao18 added 2 commits June 3, 2026 17:26
Add eplb-config option to EPLB_ARGS for NCCL.
Removed unnecessary environment variable exports for NCCL and UCX.
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 3, 2026

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 4, 2026

1 similar comment
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 4, 2026

@wzhao18 wzhao18 changed the title [WIP] Update B200 Dsv4 configs Update B200 Dsv4 configs Jun 4, 2026
@wzhao18
Copy link
Copy Markdown
Collaborator Author

wzhao18 commented Jun 4, 2026

@kedarpotdar-nv @functionstackx @Oseltamivir Ready for review/merge. Thanks!

@Oseltamivir
Copy link
Copy Markdown
Collaborator

/reuse-sweep-run

Copy link
Copy Markdown
Collaborator

@Oseltamivir Oseltamivir left a comment

Choose a reason for hiding this comment

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

lgtm

@Oseltamivir Oseltamivir merged commit 53f61f8 into main Jun 5, 2026
18 of 19 checks passed
@Oseltamivir Oseltamivir deleted the wzhao/dsv4-b200-eplb branch June 5, 2026 18:14
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Jun 5, 2026

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

Development

Successfully merging this pull request may close these issues.

2 participants