Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ base:
name: dsv4-fp4-b200-vllm-agentic
model:
path: hf:deepseek-ai/DeepSeek-V4-Pro-0813
container: vllm/vllm-openai:nightly-dev-x86_64-cu13.0.1-591bb95
container: vllm/vllm-openai:nightly-2eb491e18b1ce5b8a25e4c88f32dbb5d028d525c@sha256:514cd42702d29a7b8efe20b0a916523ac473ecebdec8d53b1a0d0f5439a48634
precision: fp4
resources:
gpu_type: b200
Expand Down Expand Up @@ -53,7 +53,6 @@ base:
VLLM_PREFIX_CACHE_RETENTION_INTERVAL: '32768'
VLLM_USE_V2_MODEL_RUNNER: '1'
VLLM_USE_RUST_FRONTEND: '1'
VLLM_DSV4_MEGA_FP8_COMBINE: '1'
VLLM_RPC_TIMEOUT: '600000'
TORCH_CUDA_ARCH_LIST: '10.0'
PYTHONNOUSERSITE: '1'
Expand Down Expand Up @@ -187,7 +186,7 @@ override_dep8_c32:
data-parallel-size: 8
enable-expert-parallel: true
enable-ep-weight-filter: true
moe-backend: deep_gemm_amxf4_mega_moe
moe-backend: deep_gemm_mega_moe
prefill-schedule-interval: 16
long-prefill-token-threshold: 512
kv-cache-memory-bytes: 37580963840
Expand Down Expand Up @@ -223,7 +222,7 @@ override_dep8_c64:
data-parallel-size: 8
enable-expert-parallel: true
enable-ep-weight-filter: true
moe-backend: deep_gemm_amxf4_mega_moe
moe-backend: deep_gemm_mega_moe
prefill-schedule-interval: 16
long-prefill-token-threshold: 512
kv-cache-memory-bytes: 37580963840
Expand Down Expand Up @@ -259,7 +258,7 @@ override_dep8_c96:
data-parallel-size: 8
enable-expert-parallel: true
enable-ep-weight-filter: true
moe-backend: deep_gemm_amxf4_mega_moe
moe-backend: deep_gemm_mega_moe
prefill-schedule-interval: 16
long-prefill-token-threshold: 512
kv-cache-memory-bytes: 37580963840
Expand Down Expand Up @@ -295,7 +294,7 @@ override_dep8_c128:
data-parallel-size: 8
enable-expert-parallel: true
enable-ep-weight-filter: true
moe-backend: deep_gemm_amxf4_mega_moe
moe-backend: deep_gemm_mega_moe
prefill-schedule-interval: 16
long-prefill-token-threshold: 512
kv-cache-memory-bytes: 37580963840
Expand Down Expand Up @@ -331,7 +330,7 @@ override_dep8_c160:
data-parallel-size: 8
enable-expert-parallel: true
enable-ep-weight-filter: true
moe-backend: deep_gemm_amxf4_mega_moe
moe-backend: deep_gemm_mega_moe
prefill-schedule-interval: 16
long-prefill-token-threshold: 512
kv-cache-memory-bytes: 36507222016
Expand Down Expand Up @@ -367,7 +366,7 @@ override_dep8_c192:
data-parallel-size: 8
enable-expert-parallel: true
enable-ep-weight-filter: true
moe-backend: deep_gemm_amxf4_mega_moe
moe-backend: deep_gemm_mega_moe
prefill-schedule-interval: 16
long-prefill-token-threshold: 512
kv-cache-memory-bytes: 36507222016
Expand Down
2 changes: 1 addition & 1 deletion inferencex-e2e/configs/nvidia-master.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -936,7 +936,7 @@ dsv4-fp4-b200-sglang-agentic-hicache-mtp:
- { tp: 8, ep: 8, dp-attn: true, kv-offloading: dram, kv-offload-backend: { name: hicache }, spec-decoding: draft_model, conc-list: [64, 96, 128, 160], router: { name: sglang-router, version: "0.3.2" }, srt-recipe: benchmarks/single_node/srt-slurm-recipes/dsv4/sglang/b200-fp4-mtp/agentic.yaml }

dsv4-fp4-b200-vllm-agentic-mtp:
image: vllm/vllm-openai:nightly-dev-x86_64-cu13.0.1-591bb95
image: vllm/vllm-openai:nightly-2eb491e18b1ce5b8a25e4c88f32dbb5d028d525c@sha256:514cd42702d29a7b8efe20b0a916523ac473ecebdec8d53b1a0d0f5439a48634
model: deepseek-ai/DeepSeek-V4-Pro-0813
model-prefix: dsv4
runner: cluster:b200-nscale
Expand Down
6 changes: 6 additions & 0 deletions inferencex-e2e/perf-changelog.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9538,3 +9538,9 @@
- "Run prefill context parallelism on the TP-only arms at conc 16, 32 and 48; conc 1 and 4 keep BCG."
- "Enable SGLANG_OPT_FP8_WO_A_FUSED_INVROPE."
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/3778

- config-keys:
- dsv4-fp4-b200-vllm-agentic-mtp
description:
- "Update the vLLM image to upstream nightly 2eb491e and switch DEP8 to the deep_gemm_mega_moe backend."
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/3917
Loading