Megakernel内核落地,英伟达拉高Kimi K3推理吞吐

AI资讯15分钟前发布 文明旁观者
1 00

英伟达联合 vLLM 完成 Kimi K3 模型的推理层专项调优,在 GB300 NVL72 硬件环境中,单用户推理速度达到 423 tok/s,相比未优化版本提升 1.68 倍。

本次调优由 CentML 团队工程师主导,基于 CuTe DSL 编写 Megakernel 解码内核,整套内核代码规模达到 1.2 万行。整套方案对模型注意力模块、MoE 专家路由与跨卡通信链路做算子融合,削减计算间隙带来的资源损耗,适配 Kimi K3 超大参数量架构特性。

Kimi K3 为月之暗面推出的 2.8 万亿参数 MoE 模型,激活参数规模 104B,原生支持百万 Token 上下文窗口与多模态输入。大规模 MoE 架构在多卡集群部署时,跨节点数据交换会产生大量开销,直接拉高推理延迟。

这套优化方案面向 vLLM 推理引擎,限定在指定硬件拓扑与负载场景生效。同硬件条件下,长文本、智能 Agent 类任务的交互响应出现明显变化。英伟达同步推进 vLLM 对多款主流大模型的内核适配工作,覆盖 DeepSeek 等其他模型系列。

本次提速只针对推理运行时,模型本身权重与基础能力不会改动。生产环境部署需要配套对应版本 vLLM 与硬件集群,普通终端用户无法直接感知该变化。

Megakernel内核落地,英伟达拉高Kimi K3推理吞吐
© 版权声明
为这篇文章评分
10.0/ 10
3 人评价
点击⭐️进行评分

相关文章

暂无评论

none
暂无评论...