Skip to content

dsv4-fp4-b300-sglang-mtp: add piecewise cuda graph flags#1702

Open
yhyang201 wants to merge 6 commits into
mainfrom
dsv4-fp4-b300-sglang-mtp-piecewise-cuda-graph
Open

dsv4-fp4-b300-sglang-mtp: add piecewise cuda graph flags#1702
yhyang201 wants to merge 6 commits into
mainfrom
dsv4-fp4-b300-sglang-mtp-piecewise-cuda-graph

Conversation

@yhyang201

@yhyang201 yhyang201 commented Jun 10, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Add --enforce-piecewise-cuda-graph, --enable-mixed-chunk, --enable-breakable-cuda-graph to MTP sglang serve launch command

Stacked on #1700.


Note

Low Risk
Changes are confined to benchmark scripts, CI config, and perf changelog; they affect how B300 perf runs are launched, not production serving or security-sensitive code.

Overview
Updates the dsv4-fp4-b300-sglang-mtp benchmark to a newer SGLang nightly image and retunes the dsv4_fp4_b300_sglang_mtp.sh launch recipe for both DP-attention and TP-only paths.

Serve flags: Both profiles now pass --enable-deepseek-v4-fp4-indexer, --enforce-piecewise-cuda-graph, --enable-mixed-chunk, and --enable-breakable-cuda-graph. The DP-attn branch also gains --cuda-graph-max-bs 256 (unchanged in intent) alongside the new graph/chunk flags.

Environment: Replaces several SGLANG_OPT_* toggles with a smaller set (e.g. SGLANG_JIT_DEEPGEMM_FAST_WARMUP, radix/reasoning-related vars, and DP-specific mega-MoE FP4 settings). DP-attn EAGLE speculative decoding shifts from (1,1,2) to (3,1,4) to align with the TP-only profile.

CI/config: nvidia-master.yaml points dsv4-fp4-b300-sglang-mtp at the new image tag; perf-changelog.yaml records the cuda-graph flag additions for this config key.

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

@github-actions

Copy link
Copy Markdown
Contributor

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.

@yhyang201 yhyang201 force-pushed the dsv4-fp4-b300-sglang-mtp-piecewise-cuda-graph branch from e423398 to 3204bd5 Compare June 10, 2026 15:06
@yhyang201 yhyang201 changed the base branch from dsv4-fp4-b300-sglang-mtp-align-gb300 to main June 10, 2026 15:22
@yhyang201 yhyang201 requested a review from a team June 10, 2026 15:22
@github-actions

Copy link
Copy Markdown
Contributor

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