Kimi K3 底层设施:FlashKDA 与分块并行计算优化

nrehiew_ · x · 2026-07-29

作者深入探讨了 Kimi K3 的基础设施细节,特别是 FlashKDA。由于 KDA 的头数较少且递归无法分割,传统的张量并行(TP)效率低下,因此团队在块之间实现了上下文并行。

为了解决 delta 更新规则依赖前一个状态的问题,他们将计算拆分为独立计算 delta 规则因子和本地状态,仅需一次 All Gather 即可应用 delta 规则,大幅提升了计算效率。

所属事件:Kimi K3 技术报告深度解读:架构与训练系统全面曝光(18 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →