创建

登录 ReadmeX

或

了解我们如何处理你的信息: 隐私政策

ReadmeX

陪你读懂 AI 动态

今天,想先了解什么?

选一份动态总结,或直接聊聊你的问题。

社区和关注动态总结只属于你自己。

资讯

Attention与MoE分布式推理的取舍

AI 总结

报告围绕权重、KV缓存、激活、通信、容量和延迟预算,分析张量并行、数据并行、专家并行、上下文并行与流水线并行的取舍。作者认为,GQA可按KV头切分,张量并行度不应超过KV头数;MLA的共享潜变量缓存在张量并行下需要复制。文章还比较了MoE张量并行与专家并行在通信、显存、负载均衡和实现复杂度方面的差异。报告标题将其标为第2篇,正文则称其为“推理基础设施方法论”系列的第3篇,共8篇。其比较数据与结论属于作者分析,现有证据未提供独立佐证。

深度见解

报告提出了一套工程判断框架:先确定模型状态如何切分,再核算单机容量、通信、显存与延迟约束,从而为大规模模型推理选择并行策略提供参考。

围绕权重、KV、激活和通信预算,推演 Attention 与 MoE 的并行组合,并分析长上下文和流水线的额外约束。

本文为「推理 INFRA 方法论」系列第 3/8 篇。发布日期按 2026 年 1–11 月编排;内容统一整理于 2026-11-01,包含年内后续更新的模型与实验资料。

Attention/MoE组合方式

Attention-TP/DP和MLA的关系:

GQA:KV 沿 n_kv 个头切 → TP ≤ n_kv(=8 时 TP8 正好),TP 同时摊薄 KV 显存 ✅
MLA:KV 是全头共享的一个 latent → 无 head 维可切 → TP 下只能【复制】,但是tp能切weight降低计算量
     TP8 = 同一份 KV 存 8 遍 → 显存爆炸 ❌
⟹ MLA 模型 attention 必走 DP:每卡管自己的请求和 KV,不复制

Attention-DP/TP和MoE-TP/EP组合下通信机制区别:

#DeepSeek V3#DeepGEMM#SGLang#H20#GQA#MLA

原文约 20 分钟读完阅读原文
群聊

还没有评论,来说说你的看法。