Skip to content

[Perf] Add tuned triton moe config for Qwen3.5 H200, 9.9% E2E throughput improvement#37340

Merged
yewentao256 merged 3 commits intomainfrom
wentao-add-tuned-tritone-moe-config
Mar 18, 2026
Merged

[Perf] Add tuned triton moe config for Qwen3.5 H200, 9.9% E2E throughput improvement#37340
yewentao256 merged 3 commits intomainfrom
wentao-add-tuned-tritone-moe-config

Conversation

@yewentao256
Copy link
Copy Markdown
Member

@yewentao256 yewentao256 commented Mar 17, 2026

Purpose

  1. updating tuning script, adding support for Qwen3.5
  2. add a tuned config for it

Note: Computing resource is limited so just running for bs 128, which is the hot path in practice.

Test

export MODEL="Qwen/Qwen3.5-35B-A3B-FP8"
vllm serve $MODEL --port 9256 --enable-expert-parallel --enable-expert-parallel --profiler-config.profiler=torch --profiler-config.torch_profiler_dir=/home/yewentao256/profile_vllm

Acc

lm_eval --model local-completions --model_args "base_url=http://127.0.0.1:9256/v1/completions,model=$MODEL,num_concurrent=1024" --tasks gsm8k

|Tasks|Version|     Filter     |n-shot|  Metric   |   |Value |   |Stderr|
|-----|------:|----------------|-----:|-----------|---|-----:|---|-----:|
|gsm8k|      3|flexible-extract|     5|exact_match||0.7801|±  |0.0114|
|     |       |strict-match    |     5|exact_match||0.7650|±  |0.0117|

Perf

vllm bench serve --model $MODEL --dataset-name random --host 127.0.0.1 --port 9256 --random-input-len 2 --random-output-len 512 --request-rate inf --num-prompts 128 --num-warmups 16

# now
============ Serving Benchmark Result ============
Successful requests:                     128       
Failed requests:                         0         
Benchmark duration (s):                  9.44      
Total input tokens:                      256       
Total generated tokens:                  65536     
Request throughput (req/s):              13.56     
Output token throughput (tok/s):         6940.23   
Peak output token throughput (tok/s):    7168.00   
Peak concurrent requests:                128.00    
Total token throughput (tok/s):          6967.34   
---------------Time to First Token----------------
Mean TTFT (ms):                          169.15    
Median TTFT (ms):                        167.24    
P99 TTFT (ms):                           201.44    
-----Time per Output Token (excl. 1st token)------
Mean TPOT (ms):                          18.11     
Median TPOT (ms):                        18.12     
P99 TPOT (ms):                           18.12     
---------------Inter-token Latency----------------
Mean ITL (ms):                           18.11     
Median ITL (ms):                         18.06     
P99 ITL (ms):                            19.42     
==================================================

# main
============ Serving Benchmark Result ============
Successful requests:                     128       
Failed requests:                         0         
Benchmark duration (s):                  10.38     
Total input tokens:                      256       
Total generated tokens:                  65536     
Request throughput (req/s):              12.33     
Output token throughput (tok/s):         6313.91   
Peak output token throughput (tok/s):    6528.00   
Peak concurrent requests:                128.00    
Total token throughput (tok/s):          6338.57   
---------------Time to First Token----------------
Mean TTFT (ms):                          306.44    
Median TTFT (ms):                        305.06    
P99 TTFT (ms):                           317.39    
-----Time per Output Token (excl. 1st token)------
Mean TPOT (ms):                          19.69     
Median TPOT (ms):                        19.69     
P99 TPOT (ms):                           19.69     
---------------Inter-token Latency----------------
Mean ITL (ms):                           19.69     
Median ITL (ms):                         19.71     
P99 ITL (ms):                            21.24     
==================================================

Signed-off-by: yewentao256 <zhyanwentao@126.com>
@yewentao256 yewentao256 added the ready ONLY add when PR is ready to merge/full CI is needed label Mar 17, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

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

Code Review

This pull request introduces a tuned Triton MoE configuration for H200, which, according to the benchmarks, yields a significant E2E throughput improvement. The changes also update the tuning script to support the Qwen3.5 model and improve robustness by refactoring how model architecture and data types are determined. The code modifications are sound and directly support the PR's objectives. I found no issues with the implementation.

@mgoin mgoin changed the title [Perf] Add tuned triton moe config for H200, 9.9% E2E throughput improvement [Perf] Add tuned triton moe config for Qwen3.5 H200, 9.9% E2E throughput improvement Mar 17, 2026
Signed-off-by: yewentao256 <zhyanwentao@126.com>
@mergify mergify bot added performance Performance-related issues qwen Related to Qwen models labels Mar 17, 2026
@yewentao256 yewentao256 merged commit 0ef7f79 into main Mar 18, 2026
63 checks passed
@yewentao256 yewentao256 deleted the wentao-add-tuned-tritone-moe-config branch March 18, 2026 18:18
fxdawnn pushed a commit to fxdawnn/vllm that referenced this pull request Mar 19, 2026
…put improvement (vllm-project#37340)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
SouthWest7 pushed a commit to SouthWest7/vllm that referenced this pull request Mar 27, 2026
…put improvement (vllm-project#37340)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
khairulkabir1661 pushed a commit to khairulkabir1661/vllm that referenced this pull request Mar 27, 2026
…put improvement (vllm-project#37340)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
Monishver11 pushed a commit to Monishver11/vllm that referenced this pull request Mar 27, 2026
…put improvement (vllm-project#37340)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
Signed-off-by: Monishver Chandrasekaran <monishverchandrasekaran@gmail.com>
JiantaoXu pushed a commit to JiantaoXu/vllm that referenced this pull request Mar 28, 2026
…put improvement (vllm-project#37340)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
vrdn-23 pushed a commit to vrdn-23/vllm that referenced this pull request Mar 30, 2026
…put improvement (vllm-project#37340)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
Signed-off-by: Vinay Damodaran <vrdn@hey.com>
EricccYang pushed a commit to EricccYang/vllm that referenced this pull request Apr 1, 2026
…put improvement (vllm-project#37340)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
Signed-off-by: EricccYang <yangyang4991@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

performance Performance-related issues qwen Related to Qwen models ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants