Skip to content

Commit d2d9043

Browse files
perf(dsv41flash): add B200 vLLM AgentX DEP2 c24, DEP4 c96 and DEP2 c64 host-KV rows / 新增 B200 vLLM AgentX DEP2 c24、DEP4 c96 与 DEP2 c64 主机 KV 行
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
1 parent d502d40 commit d2d9043

3 files changed

Lines changed: 105 additions & 9 deletions

File tree

‎inferencex-e2e/benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml‎

Lines changed: 94 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ zip_override_dep2:
127127

128128
# DEP4: TP1 x DP4 + EP4 with MegaMoE behind a consistent-hash router. At
129129
# c128, new prefills are admitted every 48 engine steps, aligned across DP
130-
# ranks.
130+
# ranks; c64 keeps the vLLM default of 1.
131131
zip_override_dep4:
132132
frontend:
133133
type: vllm-router
@@ -147,12 +147,12 @@ zip_override_dep4:
147147
enable-expert-parallel: true
148148
kernel-config: '{"enable_flashinfer_autotune":true,"moe_backend":"deep_gemm_mega_moe"}'
149149
engram-config: '{"cpu_offload":true,"embedding_across_dp":true}'
150-
max-num-seqs: [128]
151-
prefill-schedule-interval: [48]
150+
max-num-seqs: [64, 128]
151+
prefill-schedule-interval: [1, 48]
152152
compilation-config: '{"mode":"VLLM_COMPILE","cudagraph_mode":"FULL_AND_PIECEWISE","cudagraph_capture_sizes":[6,12,18,24,30,36,48,60,72,96,120,144,192,240,288,384,480,576,768,1020,1536,2046,3072,4092,6144,8190],"decoder_replay_cudagraph_capture_sizes":[6,12,18,24,30,36,48,60,72,96,120,144,192,240,288,384,480,576,768,1020,1536,2048,2304,2560,2816,3072,3328,3584,3840,4096]}'
153153
benchmark:
154154
env:
155-
CONC: ['128']
155+
CONC: ['64', '128']
156156

157157
# TP2 c1: the TP4 c1 settings on two GPUs.
158158
override_tp2_c1:
@@ -205,3 +205,93 @@ override_dep2_c128_dram:
205205
env:
206206
CONC: '128'
207207
KV_OFFLOADING: dram
208+
209+
# DEP2 c24: the DEP2 c32 settings (piecewise graphs to 4096 tokens,
210+
# prefill-schedule-interval 64) at concurrency 24.
211+
override_dep2_c24:
212+
frontend:
213+
type: vllm-router
214+
args:
215+
policy: consistent_hash
216+
prometheus-host: 127.0.0.1
217+
prometheus-port: 18000
218+
request-timeout-secs: 14400
219+
disable-retries: true
220+
setup_script: vllm-router-0.1.14.sh
221+
roles:
222+
agg:
223+
gpus: 2
224+
args:
225+
tensor-parallel-size: 1
226+
data-parallel-size: 2
227+
enable-expert-parallel: true
228+
kernel-config: '{"enable_flashinfer_autotune":true,"moe_backend":"deep_gemm_mega_moe"}'
229+
engram-config: '{"cpu_offload":true,"embedding_across_dp":true}'
230+
max-num-seqs: 32
231+
compilation-config: '{"mode":"VLLM_COMPILE","cudagraph_mode":"FULL_AND_PIECEWISE","cudagraph_capture_sizes":[6,12,18,24,30,36,48,60,72,96,120,144,192,240,288,384,480,576,640,768,896,1024,1280,1536,1792,2048,2304,2560,2816,3072,3328,3584,3840,4096]}'
232+
max-cudagraph-capture-size: 4096
233+
max-num-batched-tokens: 4096
234+
prefill-schedule-interval: 64
235+
benchmark:
236+
env:
237+
CONC: '24'
238+
239+
# DEP4 c96: the DEP4 c128 settings (prefill-schedule-interval 48) at
240+
# concurrency 96.
241+
override_dep4_c96:
242+
frontend:
243+
type: vllm-router
244+
args:
245+
policy: consistent_hash
246+
prometheus-host: 127.0.0.1
247+
prometheus-port: 18000
248+
request-timeout-secs: 14400
249+
disable-retries: true
250+
setup_script: vllm-router-0.1.14.sh
251+
roles:
252+
agg:
253+
gpus: 4
254+
args:
255+
tensor-parallel-size: 1
256+
data-parallel-size: 4
257+
enable-expert-parallel: true
258+
kernel-config: '{"enable_flashinfer_autotune":true,"moe_backend":"deep_gemm_mega_moe"}'
259+
engram-config: '{"cpu_offload":true,"embedding_across_dp":true}'
260+
max-num-seqs: 96
261+
prefill-schedule-interval: 48
262+
compilation-config: '{"mode":"VLLM_COMPILE","cudagraph_mode":"FULL_AND_PIECEWISE","cudagraph_capture_sizes":[6,12,18,24,30,36,48,60,72,96,120,144,192,240,288,384,480,576,768,1020,1536,2046,3072,4092,6144,8190],"decoder_replay_cudagraph_capture_sizes":[6,12,18,24,30,36,48,60,72,96,120,144,192,240,288,384,480,576,768,1020,1536,2048,2304,2560,2816,3072,3328,3584,3840,4096]}'
263+
benchmark:
264+
env:
265+
CONC: '96'
266+
267+
# DEP2 c64 host KV: the DEP2 c128 host-KV settings at concurrency 64.
268+
override_dep2_c64_dram:
269+
frontend:
270+
type: vllm-router
271+
args:
272+
policy: consistent_hash
273+
prometheus-host: 127.0.0.1
274+
prometheus-port: 18000
275+
request-timeout-secs: 14400
276+
disable-retries: true
277+
setup_script: vllm-router-0.1.14.sh
278+
roles:
279+
agg:
280+
gpus: 2
281+
args:
282+
tensor-parallel-size: 1
283+
data-parallel-size: 2
284+
enable-expert-parallel: true
285+
kernel-config: '{"enable_flashinfer_autotune":true,"moe_backend":"deep_gemm_mega_moe"}'
286+
engram-config: '{"cpu_offload":true,"embedding_across_dp":true}'
287+
max-num-seqs: 48
288+
compilation-config: '{"mode":"VLLM_COMPILE","cudagraph_mode":"FULL_AND_PIECEWISE","cudagraph_capture_sizes":[6,12,18,24,30,36,48,60,72,96,120,144,192,240,288,384,480,576]}'
289+
max-cudagraph-capture-size: 576
290+
max-num-batched-tokens: 4096
291+
gpu-memory-utilization: 0.955
292+
kv-offloading-backend: native
293+
kv-offloading-size: 32
294+
benchmark:
295+
env:
296+
CONC: '64'
297+
KV_OFFLOADING: dram

‎inferencex-e2e/configs/nvidia-master.yaml‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8673,11 +8673,11 @@ dsv41flash-fp4-b200-vllm-agentic-dspark:
86738673
# TP2 runs the TP4 c1 settings on two GPUs.
86748674
- { tp: 2, kv-offloading: none, spec-decoding: mtp, conc-list: [1], srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml }
86758675
# DEP2 is native TP1 x DP2 + EP2 with MegaMoE behind a consistent-hash vLLM Router.
8676-
- { tp: 2, ep: 2, dp-attn: true, kv-offloading: none, spec-decoding: mtp, conc-list: [8, 16, 32], router: { name: vllm-router, version: "0.1.14" }, srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml }
8676+
- { tp: 2, ep: 2, dp-attn: true, kv-offloading: none, spec-decoding: mtp, conc-list: [8, 16, 24, 32], router: { name: vllm-router, version: "0.1.14" }, srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml }
86778677
# DEP2 c128 adds vLLM's native CPU KV offloading at 32 GiB per DP rank.
8678-
- { tp: 2, ep: 2, dp-attn: true, kv-offloading: dram, kv-offload-backend: { name: native }, spec-decoding: mtp, conc-list: [128], router: { name: vllm-router, version: "0.1.14" }, srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml }
8678+
- { tp: 2, ep: 2, dp-attn: true, kv-offloading: dram, kv-offload-backend: { name: native }, spec-decoding: mtp, conc-list: [64, 128], router: { name: vllm-router, version: "0.1.14" }, srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml }
86798679
# DEP4 is native TP1 x DP4 + EP4 with MegaMoE behind the same router.
8680-
- { tp: 4, ep: 4, dp-attn: true, kv-offloading: none, spec-decoding: mtp, conc-list: [128], router: { name: vllm-router, version: "0.1.14" }, srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml }
8680+
- { tp: 4, ep: 4, dp-attn: true, kv-offloading: none, spec-decoding: mtp, conc-list: [64, 96, 128], router: { name: vllm-router, version: "0.1.14" }, srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv41flash/vllm/b200-fp4-mtp/agentic.yaml }
86818681

86828682
# SGLang arm for DeepSeek-V4.1-Flash AgentX on B200, from the SGLang cookbook
86838683
# (https://lmsysorg.mintlify.app/cookbook/autoregressive/DeepSeek/DeepSeek-V4_1).

‎inferencex-e2e/perf-changelog.yaml‎

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9652,6 +9652,12 @@
96529652
scenario-type:
96539653
- agentic-coding
96549654
description:
9655-
- "Tune B200 vLLM AgentX DEP2/DEP4 with prefill-schedule-interval and extended piecewise CUDA graph capture."
9656-
- "Add TP2 c1 and DEP2 c128 host-KV (native CPU offloading) rows."
9655+
- "DEP4 c128: admit new prefills every 48 engine steps (prefill-schedule-interval 48). DEP4 c64 is unchanged."
9656+
- "DEP2 c32: extend piecewise CUDA graph capture to 4096 tokens (max-cudagraph-capture-size 4096) and set prefill-schedule-interval 64. DEP2 c8 and c16 are unchanged."
9657+
- "Add a TP2 c1 row that runs the TP4 c1 settings on two GPUs."
9658+
- "Add a DEP2 c24 row with the DEP2 c32 settings (piecewise graphs to 4096 tokens, prefill-schedule-interval 64) at concurrency 24."
9659+
- "Add a DEP4 c96 row with the DEP4 c128 settings (prefill-schedule-interval 48) and max-num-seqs 96."
9660+
- "Add a DEP2 c64 host-KV row with the DEP2 c128 host-KV settings at concurrency 64."
9661+
- "Add a DEP2 c128 row with vLLM native CPU KV offloading (kv-offloading: dram, backend native), 32 GiB per DP rank, gpu-memory-utilization 0.955 and max-num-seqs 48."
9662+
- "Image, DSpark settings and all other points are unchanged."
96579663
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/3837

0 commit comments

Comments
 (0)