英伟达实验室:用内核设计智能体优化Kimi线性注意力,提速达2.96倍

【英伟达实验室:用内核设计智能体优化Kimi线性注意力,提速达2.96倍】金色财经报道,9月27日,据NVIDIA Labs博客《KDA²》,其内核设计智能体(KDAgent)已能自主优化Moonshot AI的Kimi Delta Attention(KDAttn)算子。在英伟达B300、8,192 tokens下,相对官方FlashKDA基线:TIRx版几何平均加速2.96倍、CAKE-PTX版2.94倍、CuTe-DSL版2.85倍;同时新内核将8k tokens最终状态的相对误差由FlashKDA的3.45%降至0.22%(CuTe)与0.29%(TIRx),约为其十分之一。内核已开源。

上一篇:

下一篇:

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注