Replies: 1 comment
|
Yes — vLLM (V1 engine) supports data-parallel attention for DeepSeek and other MLA-based MoE models. Status
Why DP attention matters for DeepSeekMLA keeps the KV cache small, but TP replicates that KV across ranks. DP attention instead keeps one KV copy per rank for the attention layers and pairs it with expert parallelism (or TP) for the MoE expert layers. That cuts KV memory replication and lets you scale across nodes using sparse dispatch/combine (DeepEP over NVSHMEM/RDMA) rather than TP all-reduces. The vLLM docs explicitly recommend this topology for DeepSeek-class models: "data parallel for the attention layers and expert or tensor parallel (EP or TP) for the expert layers." How to enable it# DP=4, TP=2 on a single 8-GPU node
vllm serve deepseek-ai/DeepSeek-V3 --data-parallel-size 4 --tensor-parallel-size 2
# Same, but run the expert layers with expert parallelism instead of TP
vllm serve deepseek-ai/DeepSeek-V3 --data-parallel-size 4 --tensor-parallel-size 2 --enable-expert-parallelMulti-node and Ray are supported via When it pays offDP attention is a throughput play — it shines under high concurrency. Budget for the coordination overhead: MoE forward passes must stay synchronized across DP ranks (dummy forward passes run in ranks with no requests), and References: Data Parallel Deployment · Expert Parallel Deployment · tracking issue vllm-project/vllm#12871 |
Uh oh!
There was an error while loading. Please reload this page.
Does VLLM support DP attention for deepseek?
All reactions