vLLM/Recipes
DeepSeek

deepseek-ai/DeepSeek-V4-Pro

DeepSeek V4 flagship MoE (1.6T total / 49B active) with hybrid CSA+HCA attention, manifold-constrained hyper-connections, Muon-trained on 32T+ tokens, and three-tier reasoning.

Frontier 1.6T/49B reasoning MoE with native FP4+FP8 weights, MTP and DSpark speculative decoding, and 1M-token context

moe1600B / 49B1,048,576 ctxvLLM 0.20.0+text
Guide

Overview

DeepSeek-V4-Pro is the flagship of the V4 preview family: a 1.6T-total / 49B-active Mixture-of-Experts model. It pairs a hybrid attention stack — Compressed Sparse Attention (CSA) + Heavily Compressed Attention (HCA) — with Manifold-Constrained Hyper-Connections (mHC) to reach 27% of V3.2's per-token inference FLOPs and 10% of V3.2's KV cache at 1M context. Pre-trained on 32T+ tokens with the Muon optimizer for faster convergence; post-training is a two-stage pipeline (domain-specific expert cultivation + unified consolidation via on-policy distillation).

Checkpoint is FP4+FP8 mixed: MoE expert weights are stored in FP4 while the remaining (attention / norm / router) params stay in FP8.

An NVFP4 variant (nvidia/DeepSeek-V4-Pro-NVFP4) is also available — NVIDIA modelopt re-quantizes the MoE experts to standard NVFP4 while attention, shared experts, router head, and MTP stay FP8. Pick it from the Variant row; it runs on Blackwell GPUs with the FP4 indexer cache.

Checkpoints

Four checkpoints are on the Variant row. They differ in which weights you serve and which draft module ships with them — the speculative method itself is picked on the Spec Decoding row.

VariantRepoDraftNotes
FP8 (0813) (default)deepseek-ai/DeepSeek-V4-Pro-0813DSparkOfficial release, preview structure + DSpark
FP8 (Preview)deepseek-ai/DeepSeek-V4-ProMTPPreview FP4+FP8 mixed weights
NVFP4nvidia/DeepSeek-V4-Pro-NVFP4MTPmodelopt re-quant, Blackwell
DSparkdeepseek-ai/DeepSeek-V4-Pro-DSparkDSparkPreview weights + fused DSpark module

0813 is the official DeepSeek-V4-Pro release and the default here, superseding the preview with substantially stronger agentic capability — 87.9 on Terminal Bench 2.1, 62.7 on DeepSWE and 42.7 on HLE (60.0 with tools), versus 72.1, 12.8 and 37.7 for the preview. It is built on the preview model structure with a DSpark speculative decoding module attached. The preview weights remain available as the FP8 (Preview) variant for reproducing earlier results.

DSpark is not new weights — it is the preview checkpoint with a speculative decoding module attached (see DeepSpec). Both fused checkpoints add a dspark_* block to config.json; the preview and NVFP4 checkpoints have no such block, which is why the DSpark method is offered only on the two variants that carry the draft. Selecting either one auto-enables Spec Decoding with method: dspark, emitting:

--speculative-config '{"method":"dspark","num_speculative_tokens":7,"draft_sample_method":"probabilistic"}'

The fused checkpoints are ~893 GB on disk versus ~865 GB for the preview — the draft module is the difference. Both require vLLM 0.25.0, above this recipe's 0.20.0 baseline; the Install block bumps automatically when either is selected.

Reasoning modes

The chat template exposes three reasoning-effort modes:

  • Non-think — fast, intuitive responses.
  • Think High — explicit chain-of-thought for complex problem-solving and planning.
  • Think Max — maximum reasoning effort; requires --max-model-len >= 393216 (384K tokens) to avoid truncation.

Recommended sampling: temperature = 1.0, top_p = 1.0.

On the 0813 variant the effort levels are named low / high / max, and DeepSeek recommends top_p = 0.95 for agentic scenarios (1.0 otherwise) with temperature = 1.0. Allow up to 384K output tokens at the high and max levels.

Note that the 0813 release ships no Jinja chat template — the repo provides an encoding/ folder with encode_messages / parse_message_from_completion_text helpers instead. Serving through vLLM with --tokenizer-mode deepseek_v4 (the Tool Calling pill) applies the built-in DeepSeek-V4 encoding, so the OpenAI- compatible endpoint works without the helper scripts.

OpenAI Client Example

For DeepSeek-V4, keep reasoning controls in chat_template_kwargs, as it exposes a custom Think Max mode via "reasoning_effort": "max".

from openai import OpenAI

client = OpenAI(base_url="http://localhost:8000/v1", api_key="EMPTY")
model = "deepseek-ai/DeepSeek-V4-Pro-0813"
messages = [{"role": "user", "content": "What is 17*19? Return only the final integer."}]

# Non-think
resp = client.chat.completions.create(
    model=model,
    messages=messages,
)

# Think High
resp = client.chat.completions.create(
    model=model,
    messages=messages,
    extra_body={
        "chat_template_kwargs": {
            "thinking": True,
            "reasoning_effort": "high",
        },
    },
)

# Think Max
resp = client.chat.completions.create(
    model=model,
    messages=messages,
    extra_body={
        "chat_template_kwargs": {
            "thinking": True,
            "reasoning_effort": "max",
        },
    },
)
  • B300 (8× GPU): single-node DP + EP with --data-parallel-size 8.
  • H200 (8× GPU): DP + EP with --data-parallel-size 8. Context is capped at 800K tokens (--max-model-len 800000) to leave KV headroom with dense params replicated across ranks — applies to both single-node and multi-node H200.
  • MI355X (8× GPU): validated with ROCm + AITER (VLLM_ROCM_USE_AITER=1), --gpu-memory-utilization 0.9, --max-num-seqs 128, --max-num-batched-tokens 8192, and --distributed-executor-backend mp.
  • GB200 NVL4 (4× GPU per tray): the ~960 GB mixed-precision checkpoint does not fit on one tray; run multi-node DP + EP across 2 trays (8 GPUs total) with --data-parallel-size 8. Pick the "Multi-Node" tab and set nodes to 2.

MI355X (8×288GB)

export VLLM_ROCM_USE_AITER=1

vllm serve deepseek-ai/DeepSeek-V4-Pro \
  --host localhost \
  --port 8001 \
  --dtype auto \
  --kv-cache-dtype fp8 \
  --tensor-parallel-size 8 \
  --max-num-seqs 512 \
  --max-num-batched-tokens 8192 \
  --distributed-executor-backend mp \
  --trust-remote-code \
  --gpu-memory-utilization 0.9 \
  --tokenizer-mode deepseek_v4 \
  --reasoning-parser deepseek_v4 \
  --tool-call-parser deepseek_v4 \
  --enable-auto-tool-choice \
  --compilation-config '{"mode": 3, "cudagraph_mode": "FULL_DECODE_ONLY"}'

MI355X is validated on GSM8K dataset:

Launch command
MODEL=deepseek-ai/DeepSeek-V4-Pro
lm_eval --model local-completions \
  --model_args model=$MODEL,base_url=http://0.0.0.0:8001/v1/completions,num_concurrent=128,max_retries=10,max_gen_toks=2048,timeout=60000 \
  --batch_size auto \
  --tasks gsm8k \
  --num_fewshot 8 \
  --output_path . 2>&1 | tee -a eval.log
Reported result
local-completions ({'model': 'deepseek-ai/DeepSeek-V4-Pro', 'base_url': 'http://0.0.0.0:8001/v1/completions', 'num_concurrent': 128, 'max_retries': 10, 'max_gen_toks': 2048, 'timeout': 60000}), gen_kwargs: ({}), limit: None, num_fewshot: 8, batch_size: auto
|Tasks|Version|     Filter     |n-shot|  Metric   |   |Value |   |Stderr|
|-----|------:|----------------|-----:|-----------|---|-----:|---|-----:|
|gsm8k|      3|flexible-extract|     8|exact_match|↑  |0.9538|±  |0.0058|
|     |       |strict-match    |     8|exact_match|↑  |0.9545|±  |0.0057|

KV Cache Offloading

Agentic and multi-turn workloads reuse long prefixes whose KV state can exceed on-GPU capacity. The KV Offload row attaches a host-DRAM KV tier to any serving strategy — pick one of three connectors:

  • Simple — SimpleCPUOffloadConnector: spills KV blocks to a per-rank region of CPU DRAM on each node. The simplest way to extend effective KV capacity for a single instance.
  • Mooncake — MooncakeStoreConnector: pools CPU DRAM into a distributed shared KV store — either embedded (each rank's vLLM worker runs a Mooncake client that donates a DRAM segment) or standalone (a per-node mooncake_store_service owns the node's DRAM and contributes it to the pool, decoupling cache lifetime from the engine). A cluster-wide mooncake_master coordinates the pool; across 2+ instances, GPU workers share it for cross-instance prefix reuse. Install from the extra-install block above; see the Mooncake docs.
  • LMCache — LMCacheMPConnector: a node-local KV pool served by a companion lmcache server process, launched before vllm serve. Single-node strategies only. Install from the extra-install block above; see the LMCache docs.

Agentic serving on GB200/GB300 (DSpark)

For agentic serving, use the following common settings with DeepSeek-V4-Pro-0813 and its native DSpark draft. These are separate from the command-builder defaults.

Common settings

Set VLLM_USE_V2_MODEL_RUNNER=1 and use these engine arguments:

--tokenizer-mode deepseek_v4 --trust-remote-code
--kv-cache-dtype fp8 --block-size 256
--no-disable-hybrid-kv-cache-manager --disable-flashinfer-autotune
--attention-config '{"backend":"FLASHINFER_MLA_SPARSE_DSV4","use_prefill_query_quantization":true,"indexer_kv_dtype":"mxfp4"}'
--speculative-config '{"method":"dspark","num_speculative_tokens":6,"draft_sample_method":"probabilistic","rejection_sample_method":"block"}'

The example uses K=6, where K is num_speculative_tokens. Tune K and the prefill/decode (PD) ratio for each target concurrency to balance throughput, per-user latency and time to first token. Use the same K on both PD roles.

For aggregate TP serving, use --disable-custom-all-reduce, VLLM_ALLREDUCE_USE_FLASHINFER=1, VLLM_FLASHINFER_ALLREDUCE_BACKEND=auto; do not select an EP MoE backend. For DEP (TP1 with data and expert parallelism), both pools instead use --enable-expert-parallel --enable-ep-weight-filter --enable-cumem-allocator, --moe-backend deep_gemm_amxf4_mega_moe and VLLM_DSV4_MEGA_FP8_COMBINE=1. This MoE configuration supports the native mixed FP4/FP8 checkpoint, unlike the FP8-only deep_gemm_mega_moe setting above.

For aggregate and decode, set cudagraph_mode=FULL_DECODE_ONLY and mode=0 in --compilation-config, with explicit capture sizes n*(K+1) for n=1..S. S is max-num-seqs per DP rank, or per engine for pure TP. Set the maximum capture size to S*(K+1); aggregate serving can also include n*K sizes. Recompute the capture sizes when K or S changes. Size max-num-batched-tokens to cover verification tokens plus any prefill work scheduled on that engine.

Scheduling, NUMA binding, and streaming

DEP decode scheduling: the decode example below uses:

--prefill-schedule-interval 16 --long-prefill-token-threshold 512

The interval controls how often DP ranks admit new prefill requests. Increasing it can balance work across ranks but delays admission. The threshold limits long-prefill chunk size when requests share the token budget. Compare these overrides with leaving both flags unset, measuring throughput, time to first token, and inter-token latency at your target load. Retune when changing the workload, pool sizes, hardware, or engine version.

The prefill example uses --long-prefill-token-threshold 1024 and leaves --prefill-schedule-interval unset.

DEP NUMA binding: on every prefill and decode node, enable binding and provide one NUMA node ID per visible GPU. For the four-GPU layout where GPUs 0/1 are local to NUMA node 0 and GPUs 2/3 to NUMA node 1, use:

--numa-bind --numa-bind-nodes 0 0 1 1

Match the list to the local GPU order and topology on each node, including GB300 deployments. See the vLLM NUMA binding guide for automatic detection and explicit mappings.

Aggregate streaming: add --stream-interval 10 to the aggregate TP8 worker at concurrency 1, 4, or 8. This batches streamed output into larger updates to reduce host overhead; use a smaller interval when smoother token delivery matters more. The DEP configurations leave this override unset.

Running with Dynamo

The example below runs a DEP8 prefill pool and a DEP8 decode pool, each across two four-GPU GB200 nodes, plus a host for shared services and the frontend. Run the Python commands in vllm/vllm-openai:nightly-dev-arm64-cu13.0.1-20ebb28 with the model, recipe files, GPUs, and RDMA devices accessible; use host networking so the advertised addresses are reachable across nodes. Install the matching Dynamo packages in each frontend/worker container:

python3 -m pip install --extra-index-url https://pypi.nvidia.com \
  ai-dynamo==1.3.0.dev20260720 ai-dynamo-runtime==1.3.0.dev20260720

Save the following as dynamo-env.sh and source it in each shell. Replace the example addresses with routable addresses on your cluster. MODEL_PATH must contain the pinned 0813 checkpoint on every worker.

export INFRA_IP=10.0.0.10
export ETCD_ENDPOINTS="http://${INFRA_IP}:2379"
export NATS_SERVER="nats://${INFRA_IP}:4222"
export DYN_NAMESPACE=dspark
export MODEL_PATH=/models/DeepSeek-V4-Pro-0813
export SERVED_MODEL=deepseek-ai/DeepSeek-V4-Pro-0813
export PREFILL_HEAD_IP=10.0.0.11
export DECODE_HEAD_IP=10.0.0.21

Shared services. Provide etcd, nats-server, and mooncake_master on the service host's PATH. See the official Dynamo service setup and Mooncake installation for installation options. Start these services once on the private cluster network, before workers:

source ./dynamo-env.sh
etcd --data-dir ./etcd-data --listen-client-urls http://0.0.0.0:2379 \
  --advertise-client-urls "$ETCD_ENDPOINTS" >etcd.log 2>&1 &
cat >nats.conf <<'EOF'
port: 4222
max_payload: 33554432
jetstream { store_dir: "./nats-data" }
EOF
nats-server -c ./nats.conf >nats.log 2>&1 &
mooncake_master --port=8700 --enable_http_metadata_server=true \
  --http_metadata_server_port=8701 --eviction_high_watermark_ratio=0.9 \
  --default_kv_lease_ttl=10000 --rpc_thread_num=16 \
  --enable_metric_reporting=true --metrics_port=8702 >mooncake.log 2>&1 &

Wait for the services to listen before continuing. All hosts must reach the shared endpoints; workers must also reach each other's DP rendezvous and NIXL addresses. Existing managed services can supply the same endpoints.

Frontend. In a separate shell, start the HTTP endpoint for the disaggregated deployment:

source ./dynamo-env.sh
export DYN_ROUTER_ACTIVE_REQUEST_EXPIRY_SECS=3600
export DYN_TCP_CHANNEL_BUFFER=128
export DYN_TCP_REQUEST_TIMEOUT=60
python3 -m dynamo.frontend --http-port 8000 --router-mode random \
  --router-session-affinity-ttl-secs 900
  • Aggregate: --router-mode kv, DYN_ROUTER_TEMPERATURE=0 and DYN_TOKENIZER=fastokens.
  • Disaggregated: --router-mode random with session affinity, or kv with the same tokenizer settings. Send a stable conversation ID as x-dynamo-session-id and set --router-session-affinity-ttl-secs to cover the conversation lifetime.
  • Both DEP roles use MultiConnector with NixlConnector for PD transfer and MooncakeStoreConnector for shared prefix storage. Enable store lookup only on prefill, and set VLLM_CONNECTOR_PREFETCH_DEPTH=0 there. Role-specific --kv-transfer-config values:

Prefill — save as prefill-kv.json in the worker's working directory:

{"kv_connector":"MultiConnector","kv_role":"kv_both","kv_connector_extra_config":{"connectors":[{"kv_connector":"NixlConnector","kv_role":"kv_both","kv_load_failure_policy":"fail","kv_buffer_device":"cuda","kv_connector_extra_config":{"enforce_handshake_compat":false,"enable_cross_layers_blocks":false}},{"kv_connector":"MooncakeStoreConnector","kv_role":"kv_both","kv_connector_extra_config":{"load_async":true,"lookup_async":true,"enable_cross_layers_blocks":false,"enable_offload":false}}]}}

Decode — save as decode-kv.json in the worker's working directory:

{"kv_connector":"MultiConnector","kv_role":"kv_consumer","kv_connector_extra_config":{"connectors":[{"kv_connector":"NixlConnector","kv_role":"kv_consumer","kv_load_failure_policy":"fail","kv_buffer_device":"cuda","kv_connector_extra_config":{"enforce_handshake_compat":false,"enable_cross_layers_blocks":false}},{"kv_connector":"MooncakeStoreConnector","kv_role":"kv_consumer","kv_connector_extra_config":{"load_async":true,"lookup_async":false,"enable_lookup":false,"enable_cross_layers_blocks":false,"enable_offload":false}}]}}

Worker environment. Run the following in Bash on each worker node. Set NODE_IP to that node's routable address and NODE_RANK to 0 or 1 within its own pool. For example, prefill nodes 10.0.0.11 / 10.0.0.12 use ranks 0 / 1; decode nodes 10.0.0.21 / 10.0.0.22 also use ranks 0 / 1. All four GPUs are visible to one launch process per node. Adjust the NUMA and NIC mappings to match the hardware. The embedded Mooncake store uses 140 GB per local rank in this example; ensure enough host memory or resize it for your deployment.

source ./dynamo-env.sh
export NODE_IP=10.0.0.11  # Change on each worker node.
NODE_RANK=0              # Use 1 on the second node of either pool.
export CUDA_VISIBLE_DEVICES=0,1,2,3
export VLLM_HOST_IP="$NODE_IP"
export VLLM_NIXL_SIDE_CHANNEL_HOST="$NODE_IP"
export VLLM_NIXL_SIDE_CHANNEL_PORT=20096
export DYN_SYSTEM_PORT=8081
export VLLM_USE_V2_MODEL_RUNNER=1 VLLM_DSV4_MEGA_FP8_COMBINE=1
export PYTHONHASHSEED=0
export MOONCAKE_MASTER="${INFRA_IP}:8700"
export MOONCAKE_TE_META_DATA_SERVER="http://${INFRA_IP}:8701/metadata"
export MOONCAKE_LOCAL_HOSTNAME="$NODE_IP"
export MOONCAKE_CONFIG_PATH="$PWD/mooncake.json"
cat >"$MOONCAKE_CONFIG_PATH" <<EOF
{"metadata_server":"P2PHANDSHAKE",
 "master_server_address":"${MOONCAKE_MASTER}","global_segment_size":"140GB",
 "local_buffer_size":"4GB","protocol":"rdma",
 "device_name":"mlx5_0,mlx5_1,mlx5_2,mlx5_3","mode":"embedded","enable_offload":false}
EOF

COMMON_ARGS=(
  --model "$MODEL_PATH" --served-model-name "$SERVED_MODEL"
  --tensor-parallel-size 1 --pipeline-parallel-size 1
  --data-parallel-size 8 --data-parallel-size-local 4
  --data-parallel-start-rank "$((4 * NODE_RANK))"
  --data-parallel-rpc-port 13345 --data-parallel-hybrid-lb
  --enable-expert-parallel --enable-ep-weight-filter --enable-cumem-allocator
  --moe-backend deep_gemm_amxf4_mega_moe
  --numa-bind --numa-bind-nodes 0 0 1 1
  --tokenizer-mode deepseek_v4 --trust-remote-code
  --max-model-len 1048576 --kv-cache-dtype fp8 --block-size 256
  --no-disable-hybrid-kv-cache-manager --disable-flashinfer-autotune
  --attention-config '{"backend":"FLASHINFER_MLA_SPARSE_DSV4","use_prefill_query_quantization":true,"indexer_kv_dtype":"mxfp4"}'
  --speculative-config '{"method":"dspark","num_speculative_tokens":5,"draft_sample_method":"probabilistic","rejection_sample_method":"block"}'
)

Prefill workers. Run once on each prefill node after the worker environment block, keeping PREFILL_HEAD_IP identical across the pool:

export VLLM_CONNECTOR_PREFETCH_DEPTH=0
python3 -m dynamo.vllm "${COMMON_ARGS[@]}" --disaggregation-mode prefill \
  --data-parallel-address "$PREFILL_HEAD_IP" \
  --kv-transfer-config "$(cat prefill-kv.json)" \
  --max-num-seqs 64 --max-num-batched-tokens 8192 \
  --long-prefill-token-threshold 1024 --gpu-memory-utilization 0.85 \
  --max-cudagraph-capture-size 384

Decode workers. Run once on each decode node after the worker environment block. This example uses K=5 and S=8, so graph capture sizes are 6 through 48:

python3 -m dynamo.vllm "${COMMON_ARGS[@]}" --disaggregation-mode decode \
  --data-parallel-address "$DECODE_HEAD_IP" \
  --kv-transfer-config "$(cat decode-kv.json)" \
  --max-num-seqs 8 --max-num-batched-tokens 8240 \
  --prefill-schedule-interval 16 --long-prefill-token-threshold 512 \
  --gpu-memory-utilization 0.9 --kv-cache-memory-bytes 37580963840 \
  --max-cudagraph-capture-size 48 \
  --compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY","mode":0,"cudagraph_capture_sizes":[6,12,18,24,30,36,42,48]}'

For DEP12/DEP16, set --data-parallel-size to 12/16 in the corresponding pool and run on three/four four-GPU nodes. Each pool keeps its own head address; start ranks are 4 * NODE_RANK (0, 4, 8, 12). Tune scheduling for the target workload and recompute graph sizes when changing K or S. The same port numbers can be reused on separate hosts; co-located workers need distinct port allocations. The official Dynamo vLLM examples provide the aggregate and KV-routing launch patterns; apply this section's model and aggregate settings when using those patterns.

Once all workers register, send requests to the frontend, retaining the same session ID for turns in one conversation:

curl "http://${INFRA_IP}:8000/v1/chat/completions" \
  -H 'Content-Type: application/json' -H 'x-dynamo-session-id: example-session' \
  -d '{"model":"deepseek-ai/DeepSeek-V4-Pro-0813","messages":[{"role":"user","content":"Hello"}],"max_tokens":32}'

Agentic benchmark reproduction (InferenceX MI355X sweep)

The command above is the default serving configuration. The SemiAnalysis InferenceX MI355X benchmark lane for this checkpoint is a separate high-concurrency benchmark configuration — not the recipe default. To reproduce that sweep, start from the default command above and apply these deltas:

Source configuration:

image: vllm/vllm-openai-rocm:nightly-eed1f3d0c6043bd494424a22443ee198dd56f657

Applied to both arms:

  • Enable VLLM_ROCM_QUICK_REDUCE_QUANTIZATION=INT4 (previously commented out).
  • Disable VLLM_ROCM_USE_AITER_FUSION_SHARED_EXPERTS=0 for this nightly. DSv4-Pro is a mixed checkpoint (MXFP4 routed experts, FP8 shared expert), and this image fails while loading incompatible scales and shapes when the fused shared-expert path is enabled.
  • Raise --gpu-memory-utilization from 0.8 to 0.86.
  • Pin --max-num-batched-tokens 8192 instead of the nightly default of 16384. On the TP8 initialization check this raised GPU KV-cache capacity from 4,730,981 to 8,524,228 tokens and reduced peak activation memory from 11.44 GiB to 8.9 GiB.

Applied to the DP-attention arm only:

  • Cap --max-num-seqs at CONC rather than 2*CONC. The limit is per scheduler and DP-attention runs one scheduler per rank. The pure TP8 arm keeps the existing 2*CONC headroom for AgentX subagent fan-out.
  • Add --prefill-schedule-interval 8 and --long-prefill-token-threshold 16384.

With --max-num-batched-tokens 8192 now pinned for both arms, --long-prefill-token-threshold 16384 sits above the token budget and therefore never binds. It is inert rather than harmful, and the DP-attention point was measured with it present.

--compilation-config '{"mode":3,"cudagraph_mode":"FULL_AND_PIECEWISE"}' is unchanged; it was already in the checked-in recipe.

Export the AITER and INT4 quantized all-reduce environment variables before launch:

export VLLM_ROCM_USE_AITER=1
export VLLM_ROCM_QUICK_REDUCE_QUANTIZATION=INT4
export VLLM_ROCM_USE_AITER_MOE=1
export VLLM_ROCM_USE_AITER_FUSION_SHARED_EXPERTS=0
export VLLM_ENGINE_READY_TIMEOUT_S=10800

vllm serve "$MODEL_PATH" --served-model-name "$MODEL" \
  --host 0.0.0.0 \
  --port "$VLLM_BACKEND_PORT" \
  --trust-remote-code \
  --async-scheduling \
  --distributed-executor-backend mp \
  --kv-cache-dtype fp8 \
  --max-num-batched-tokens 8192 \
  --gpu-memory-utilization 0.86 \
  --moe-backend aiter \
  --compilation-config '{"mode":3,"cudagraph_mode":"FULL_AND_PIECEWISE"}' \
  --speculative-config '{"method": "mtp", "num_speculative_tokens": 3, "rejection_sample_method": "synthetic", "synthetic_acceptance_length": 2.49}' \
  --tokenizer-mode deepseek_v4 \
  --tool-call-parser deepseek_v4 \
  --reasoning-parser deepseek_v4 \
  --enable-auto-tool-choice \
  --enable-prefix-caching \
  --prefix-cache-retention-interval 32768 \
  --no-disable-hybrid-kv-cache-manager \
  --max-num-seqs "8"

--prefix-cache-retention-interval replaces the VLLM_PREFIX_CACHE_RETENTION_INTERVAL env var, removed in vLLM 0.30.0 (vllm-project/vllm#55353).

Fixed-sequence benchmark reproduction (InferenceX MI355X 8K/1K)

This is a second, separate InferenceX lane — not a variation of the Agentic sweep above. It is a fixed-sequence measurement at input length 8192 and output length 1024 on one 8× MI355X node, run as two arms: STP (no speculative decoding) and real MTP with two draft tokens. Neither arm is the general recipe default, and neither is a recommendation for other AMD workloads.

The checkpoint is deepseek-ai/DeepSeek-V4-Pro — the FP8 (Preview) pill in the Variant row, whose mixed FP4+FP8 weights InferenceX labels fp4. It is not the 0813 default variant and not the DSpark checkpoint.

Source configuration

ItemValue
InferenceX PR#2792
Config keysdsv4-fp4-mi355x-vllm (STP), dsv4-fp4-mi355x-vllm-mtp (MTP)
Hardwareone node, 8× MI355X
Imagevllm/vllm-openai-rocm:nightly-7c5dc571cbd1064ecc8a9b1045637ff647aa22cb
Image digestsha256:f0bdaf5217a09949842b45c1ea1f12260d3205ec81f143b320dfc2eb3ec95e55
ParallelismTP 8, DP 1, EP 1 — no --enable-expert-parallel
WorkloadISL 8192, OSL 1024; concurrency 4, 8, 16, 32, 64, 128, 256, 512
Requests per point10 × concurrency

The image is pinned to an immutable nightly tag rather than a floating :nightly, because the measurements depend on the ROCm DeepSeek-V4 kernel state in that specific build. The Install block's AMD default stays on :nightly deliberately — pin the tag above only when reproducing these numbers.

STP launch (no speculative decoding)

export VLLM_ROCM_USE_AITER=1
export VLLM_ROCM_USE_AITER_MOE=1
# Fuses the FP8 shared expert into the MXFP4 routed-expert AITER kernel.
# See the shared-expert note below for the build and eligibility requirements.
export VLLM_ROCM_USE_AITER_FUSION_SHARED_EXPERTS=1
export VLLM_ROCM_QUICK_REDUCE_QUANTIZATION=INT4

vllm serve deepseek-ai/DeepSeek-V4-Pro \
  --port "${PORT:-8000}" \
  --tensor-parallel-size 8 \
  --data-parallel-size 1 \
  --async-scheduling \
  --no-enable-prefix-caching \
  --distributed-executor-backend mp \
  --gpu-memory-utilization 0.8 \
  --kv-cache-dtype fp8 \
  --trust-remote-code \
  --moe-backend aiter \
  --tokenizer-mode deepseek_v4 \
  --reasoning-parser deepseek_v4 \
  --compilation-config '{"mode":3,"cudagraph_mode":"FULL_AND_PIECEWISE"}'

MTP arm (real speculation, 2 draft tokens)

The MTP arm is the same launch plus one flag. Everything else — environment, parallelism, memory, cache, executor, MoE backend, parsers, compilation — is identical:

  --speculative-config '{"method":"mtp","num_speculative_tokens":2}'

This is real MTP: draft tokens are verified against the target model. ROCm support for the DeepSeek-V4 MTP head landed in vllm-project/vllm#43385. The same two-token config is available from the command builder — enable Spec decoding and pick the MTP mode with the FP8 (Preview) variant selected.

Left at engine defaults

This lane deliberately passes no --max-model-len, --max-num-seqs, --max-num-batched-tokens, or --block-size, and no tool-calling flags. That differs from this recipe's general AMD configuration, which does set several of them — one reason this reproduction is documented separately rather than folded into the default command.

Benchmarking the MTP arm needs chat formatting

MTP-style speculation is trained against chat-formatted input. Benchmarking it with raw completion prompts silently depresses the acceptance rate, so measure it against the chat endpoint, which applies the DeepSeek-V4 chat template that --tokenizer-mode deepseek_v4 installs:

CONC=32

vllm bench serve \
  --model deepseek-ai/DeepSeek-V4-Pro \
  --host localhost \
  --port "${PORT:-8000}" \
  --backend openai-chat \
  --endpoint /v1/chat/completions \
  --dataset-name random \
  --random-input-len 8192 \
  --random-output-len 1024 \
  --num-prompts "$((CONC * 10))" \
  --max-concurrency "$CONC" \
  --ignore-eos \
  --trust-remote-code

InferenceX drives its own benchmark client with a private DeepSeek-V4 encoder rather than the tokenizer's built-in template, so the numbers on its dashboard are not byte-identical to what the command above produces. Use the chat endpoint for any MTP acceptance measurement regardless.

Shared-expert fusion requires a recent build

VLLM_ROCM_USE_AITER_FUSION_SHARED_EXPERTS=1 is a request, not a guarantee: vLLM gates the fused shared-expert path on its own eligibility check. vllm-project/vllm#53161, merged as de69e821, added the heterogeneous path that fuses the native-FP8 shared expert into the MXFP4 routed-expert kernel. Current :nightly images include it, so the flag takes effect. On any build older than de69e821 the check rejects this mixed FP4+FP8 checkpoint outright and the fusion self-disables at startup — including the immutable tag pinned in the reproduction table above, which predates the fix.

On a build that includes it, the fusion additionally requires gfx950 (MI355X), tensor parallelism 8, data parallelism 1, no expert parallelism and no EPLB, --moe-backend aiter, BF16 model dtype, and the DeepSeek-V4 block-128 FP8 shared-expert quantization config. Both arms here satisfy this. Data-parallel attention does not, and keeps the unfused path.

Eligibility is also re-checked per invocation against AITER's tuned-config coverage for the number of MoE input rows. Decode-sized batches are covered and take the fused kernel; large prefill chunks fall back to the unfused path, so an 8K-input workload runs fused during decode and unfused during prefill.

Failures are silent and logged at debug level, so check the server log before assuming the fused path executed rather than inferring it from the flag. The line to look for is Using AITER DeepSeek V4 heterogeneous MoE for CSV-covered MoE input rows.

How this differs from the Agentic sweep above

SettingFixed-sequence 8K/1KAgentic sweep
WorkloadISL 8192 / OSL 1024, conc 4–512agentic-coding traces
--gpu-memory-utilization0.80.86
Prefix caching--no-enable-prefix-caching--enable-prefix-caching
Speculationnone (STP) or real MTP, 2 tokenssynthetic MTP, 3 tokens
--max-num-seqsnot passed8
--max-num-batched-tokensnot passed8192

The Agentic lane's synthetic acceptance length is a benchmark-comparability device. This lane never uses it: the STP arm has no draft at all, and the MTP arm verifies against the real target.

Provenance

Validation for this configuration comes from InferenceX, not from this repository — no MI355X benchmark was rerun here.