Update B200 Dsv4 configs#1655
Conversation
|
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. |
|
see unofficial run visualizer at https://inferencex.semianalysis.com/inference?unofficialRun=26899483031 |
|
see unofficial run visualizer at https://inferencex.semianalysis.com/inference?unofficialRun=26900694153 |
1 similar comment
|
see unofficial run visualizer at https://inferencex.semianalysis.com/inference?unofficialRun=26900694153 |
Set environment variables for NIXL EPLB configuration.
|
see unofficial run visualizer at https://inferencex.semianalysis.com/inference?unofficialRun=26905612623 |
Add eplb-config option to EPLB_ARGS for NCCL.
Removed unnecessary environment variable exports for NCCL and UCX.
|
see unofficial run visualizer at https://inferencex.semianalysis.com/inference?unofficialRun=26905612623 |
|
see unofficial run visualizer at https://inferencex.semianalysis.com/inference?unofficialRun=26914041013 |
1 similar comment
|
see unofficial run visualizer at https://inferencex.semianalysis.com/inference?unofficialRun=26914041013 |
|
@kedarpotdar-nv @functionstackx @Oseltamivir Ready for review/merge. Thanks! |
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_nccland synchronous EPLB, alongside the existingdeep_gemm_mega_moeMoE backend indsv4_fp4_b200_vllm.sh.Pins the
dsv4-fp4-b200-vllmconfig innvidia-master.yamlto a specific vLLM nightly container image instead ofv0.22.0, and records the EPLB change inperf-changelog.yaml.Reviewed by Cursor Bugbot for commit 783dfe8. Bugbot is set up for automated code reviews on this repo. Configure here.