Skip to content

Update B200 Dsv4 configs#1655

Open
wzhao18 wants to merge 7 commits into
mainfrom
wzhao/dsv4-b200-eplb
Open

Update B200 Dsv4 configs#1655
wzhao18 wants to merge 7 commits into
mainfrom
wzhao/dsv4-b200-eplb

Conversation

@wzhao18
Copy link
Copy Markdown
Collaborator

@wzhao18 wzhao18 commented Jun 3, 2026

Note

Medium Risk
Changes the vLLM image pin and MoE/EPLB flags for DP-attention sweep points, which can shift measured throughput/latency and comparability to prior results.

Overview
Updates DeepSeek-V4 FP4 B200 vLLM benchmark serving so DP-attention search points turn on EPLB (expert parallel load balancing) with torch_nccl and synchronous EPLB, alongside the existing deep_gemm_mega_moe MoE backend in dsv4_fp4_b200_vllm.sh.

Pins the dsv4-fp4-b200-vllm config in nvidia-master.yaml to a specific vLLM nightly container image instead of v0.22.0, and records the EPLB change in perf-changelog.yaml.

Reviewed by Cursor Bugbot for commit 783dfe8. 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!

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

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant