AI优化算子提升推理效率,Kimi K3实现3倍速度突破

AI新闻阅读 4
AI优化算子提升推理效率,Kimi K3实现3倍速度突破

Inferact团队开源TPU Megakernels巨型算子,将Kimi K3的92层MoE架构整合至单一程序,实现单用户709 tokens/s的推理速度。相较英伟达GB200的452 tokens/s,该方案在关闭投机解码时仍保持1.4-2倍性能优势。

TPU v7显存带宽虽低于GB200(7380GB/s vs 8000GB/s),但通过算子融合技术实现性能突破。该方案编译时间缩短至90秒内,相较传统TPU编译流程效率提升显著。

浪潮信息在AICC2026发布元脑SD200 Ultra超节点服务器,采用128芯本土AI芯片,实现K3模型生成时延5.85毫秒。其「超级算子」技术让K3自身参与算子优化,突破传统架构限制。

算子融合技术通过减少数据搬运次数提升效率。传统框架中每个token生成需调用数千次算子,导致显存带宽利用率低下。Inferact将92层MoE整合为单一kernel,实现数据预加载和连续计算。

AI正在接管算子编写工作。DeepSeek工程师刘胜表示,AI助手已能自主优化CUDA代码,预计半年内AI编写算子将超越人类工程师。Cursor与英伟达合作优化235个CUDA算子,平均提速38%。

Inferact与浪潮信息采用不同融合策略:前者整合完整解码流程,后者聚焦KDA/Gated MLA/MoE模块。SD200 Ultra通过3D Hyper Mesh架构实现8TB统一显存,支持10万亿参数模型单机部署。

对称内存技术使SD200 Ultra实现0.69微秒跨卡通信时延,AllReduce效率提升3.5倍。通过流水线调度和异步搬运机制,实现计算与数据搬运同步进行,最大限度利用芯片资源。

系统级创新成为大模型优化关键。浪潮信息通过超节点架构整合计算、存储与通信,为前沿模型提供统一资源池。这种软硬协同方案为大模型推理效率提升开辟新路径。

AI优化算子提升推理效率,Kimi K3实现3倍速度突破 - SmartAITool Blog