Skip to content

Commit 7a428f8

Browse files
wzhao18claudeOseltamivirAnkur-singh
authored
Update vLLM MiniMax-M3 image tags / 更新 vLLM MiniMax-M3 镜像标签 (#2120)
* update * update changelog * Add VLLM_FLASHINFER_ALLREDUCE_BACKEND variable * fix router gemm * fixup * Update minimaxm3_fp4_b300.sh * update * Update Docker image version for vllm-openai * Extend minimax M3 fp4 update to B200/B300 + eagle configs - Bump all four fp4 vLLM images to nightly-4080263bb2c5d10deac17aaeb88e0823bc35bca9 - Remove VLLM_FLASHINFER_ALLREDUCE_BACKEND=trtllm from B200/B200-mtp/B300-mtp scripts - Add --attention_config.indexer_kv_dtype fp8 to B200 non-eagle config Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> * chore: refresh PR #2120 for sweep reuse [skip-sweep] --------- Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Co-authored-by: Bryan Shan <58582368+Oseltamivir@users.noreply.github.com> Co-authored-by: Ankur-singh <ankusingh@nvidia.com>
1 parent ea74d24 commit 7a428f8

6 files changed

Lines changed: 16 additions & 8 deletions

File tree

benchmarks/single_node/fixed_seq_len/minimaxm3_fp4_b200.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ SERVER_LOG=/workspace/server.log
3333

3434
export VLLM_ENGINE_READY_TIMEOUT_S=3600
3535
export VLLM_FLOAT32_MATMUL_PRECISION=high
36-
export VLLM_FLASHINFER_ALLREDUCE_BACKEND=trtllm
3736

3837
if [ "${DP_ATTENTION}" = "true" ]; then
3938
PARALLEL_ARGS="--tensor-parallel-size=1 --data-parallel-size=$TP --enable-expert-parallel"
@@ -52,6 +51,7 @@ start_gpu_monitor
5251
set -x
5352
vllm serve $MODEL --port $PORT \
5453
$PARALLEL_ARGS \
54+
--attention_config.indexer_kv_dtype fp8 \
5555
--gpu-memory-utilization 0.95 \
5656
--max-model-len $MAX_MODEL_LEN \
5757
--kv-cache-dtype fp8 \

benchmarks/single_node/fixed_seq_len/minimaxm3_fp4_b200_mtp.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@ SERVER_LOG=/workspace/server.log
4747

4848
export VLLM_ENGINE_READY_TIMEOUT_S=3600
4949
export VLLM_FLOAT32_MATMUL_PRECISION=high
50-
export VLLM_FLASHINFER_ALLREDUCE_BACKEND=trtllm
5150

5251
if [ "${DP_ATTENTION}" = "true" ]; then
5352
PARALLEL_ARGS="--tensor-parallel-size=1 --data-parallel-size=$TP --enable-expert-parallel"

benchmarks/single_node/fixed_seq_len/minimaxm3_fp4_b300.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@ SERVER_LOG=/workspace/server.log
3838

3939
export VLLM_ENGINE_READY_TIMEOUT_S=3600
4040
export VLLM_FLOAT32_MATMUL_PRECISION=high
41-
export VLLM_FLASHINFER_ALLREDUCE_BACKEND=trtllm
4241

4342
if [ "${DP_ATTENTION}" = "true" ]; then
4443
PARALLEL_ARGS="--tensor-parallel-size=1 --data-parallel-size=$TP --enable-expert-parallel"
@@ -57,6 +56,7 @@ start_gpu_monitor
5756
set -x
5857
vllm serve "$MODEL_PATH" --served-model-name "$MODEL" --host 0.0.0.0 --port $PORT \
5958
$PARALLEL_ARGS \
59+
--attention_config.indexer_kv_dtype fp8 \
6060
--gpu-memory-utilization 0.95 \
6161
--max-model-len $MAX_MODEL_LEN \
6262
--kv-cache-dtype fp8 \

benchmarks/single_node/fixed_seq_len/minimaxm3_fp4_b300_mtp.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,6 @@ SERVER_LOG=/workspace/server.log
5252

5353
export VLLM_ENGINE_READY_TIMEOUT_S=3600
5454
export VLLM_FLOAT32_MATMUL_PRECISION=high
55-
export VLLM_FLASHINFER_ALLREDUCE_BACKEND=trtllm
5655

5756
if [ "${DP_ATTENTION}" = "true" ]; then
5857
PARALLEL_ARGS="--tensor-parallel-size=1 --data-parallel-size=$TP --enable-expert-parallel"

configs/nvidia-master.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7527,7 +7527,7 @@ minimaxm3-fp8-b300-vllm:
75277527
# weights are pre-staged read-only at /scratch/models/MiniMax-M3-NVFP4 (added to
75287528
# the STAGED_MODELS allow-list in launch_b300-nv.sh).
75297529
minimaxm3-fp4-b300-vllm:
7530-
image: vllm/vllm-openai:nightly-93d8f834dd8acf33eb0e2a75b2711b628cb6e226
7530+
image: vllm/vllm-openai:nightly-4080263bb2c5d10deac17aaeb88e0823bc35bca9
75317531
model: nvidia/MiniMax-M3-NVFP4
75327532
model-prefix: minimaxm3
75337533
runner: b300
@@ -7553,7 +7553,7 @@ minimaxm3-fp4-b300-vllm:
75537553
# /scratch/models/MiniMax-M3-NVFP4 (added to the STAGED_MODELS allow-list in
75547554
# launch_b300-nv.sh); the EAGLE3 draft is downloaded to the writable models dir.
75557555
minimaxm3-fp4-b300-vllm-mtp:
7556-
image: vllm/vllm-openai:nightly-93d8f834dd8acf33eb0e2a75b2711b628cb6e226
7556+
image: vllm/vllm-openai:nightly-4080263bb2c5d10deac17aaeb88e0823bc35bca9
75577557
model: nvidia/MiniMax-M3-NVFP4
75587558
model-prefix: minimaxm3
75597559
runner: b300
@@ -7608,7 +7608,7 @@ minimaxm3-fp8-b200-vllm:
76087608
# (MSA sparse/index cache); weights are pre-staged at /scratch/fsw/models/MiniMax-M3-NVFP4
76097609
# (launch_b200-dgxc.sh resolves MODEL_PATH for minimaxm3-fp4).
76107610
minimaxm3-fp4-b200-vllm:
7611-
image: vllm/vllm-openai:nightly-93d8f834dd8acf33eb0e2a75b2711b628cb6e226
7611+
image: vllm/vllm-openai:nightly-4080263bb2c5d10deac17aaeb88e0823bc35bca9
76127612
model: nvidia/MiniMax-M3-NVFP4
76137613
model-prefix: minimaxm3
76147614
runner: b200-dgxc
@@ -7663,7 +7663,7 @@ minimaxm3-fp8-b200-vllm-mtp:
76637663
# resolves MODEL_PATH for minimaxm3-fp4); the EAGLE3 draft is fetched next to
76647664
# the target weights.
76657665
minimaxm3-fp4-b200-vllm-mtp:
7666-
image: vllm/vllm-openai:vllm-minimax-m3-perf-x86_64-13.0.1-8b00f41
7666+
image: vllm/vllm-openai:nightly-4080263bb2c5d10deac17aaeb88e0823bc35bca9
76677667
model: nvidia/MiniMax-M3-NVFP4
76687668
model-prefix: minimaxm3
76697669
runner: b200-dgxc

perf-changelog.yaml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5191,3 +5191,13 @@
51915191
- "Agentic experiment Variant K (probe, of the #2374 Variant G): REMOVES the kimi-k3-container-deps.sh in-container patch (setup_script, script file, and launcher copy) to verify whether the mamba_hybrid index_fill_ dtype patch is still required by the current vllm/vllm-openai:kimi-k3 image — expected to fail at the first decode step if it is"
51925192
- "Recipe: benchmarks/multi_node/srt-slurm-recipes/vllm/kimi-k3/agentic/agg-b200-tp8pp2-agentic.yaml on the cluster:b200-dgxc pool"
51935193
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/2391
5194+
5195+
- config-keys:
5196+
- minimaxm3-fp4-b200-vllm
5197+
- minimaxm3-fp4-b200-vllm-mtp
5198+
- minimaxm3-fp4-b300-vllm
5199+
- minimaxm3-fp4-b300-vllm-mtp
5200+
description:
5201+
- "Update vllm image, remove setting trtllm allreduce backend"
5202+
- "Use fp8 indexer cache dtype for the B200/B300 non-eagle configs"
5203+
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/2120

0 commit comments

Comments
 (0)