掘金· 八月蝉·· 12 天前精选AI 评分78
推理 INFRA 方法论:TP、DP、EP、CP 与 PP 在分布式推理中的取舍
推理 INFRA 方法论(2):分布式推理:TP、DP、EP、CP 与 PP 的取舍
AI 导读
这篇教程按权重、KV、激活和通信预算推演 Attention 与 MoE 的并行组合,给出 TP、DP、EP、CP、PP 的选型判据。核心结论是 GQA 可沿头维切 KV 直接走 TP,MLA 的 KV 全头共享只能复制、必须走 DP;EP 的通信优势来自 EP 域大于 top-k,k=8 时 EP8 与 MoE-TP 持平、EP32 省 4 倍。
借鉴点
能拿走一套按模型结构选并行维度的判据,GQA 走 TP、MLA 必走 DP、卡多上 EP、长上下文用 CP/DCP,代价是结论绑定 DeepSeek V3 口径与 H20 实测,换模型或换卡要重算显存和通信预算。
来源:掘金 · juejin.cn