跳到正文
AGI Hunt· Snowflake·· 2 小时前AI 评分44

KernelSeek:靠“提问”而非堆工具的编码 AI 智能体,1.13× 跑赢 FlashAttention-3

KernelSeek:靠“提问”而非堆工具的编码 Agent,1.13× 跑赢 FlashAttention-3

AI 导读

论文《Ask, Don't Equip》提出 KernelSeek,以决策点提问驱动 LLM 编码 AI 智能体优化 GPU kernel。生成的 FlashAttention 前向 kernel 在 Hopper 与 Blackwell 全部 182 个场景中正确,并以 1.13× 跑赢 FlashAttention-3,对 FA-3/FA-4 的几何平均性能分别为 1.16× 和 1.46×。

来源:AGI Hunt · agihunt.info