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

Snowflake · hf · 2026-11-05

论文《Ask, Don't Equip》提出 KernelSeek,一个问题驱动的 LLM 编码 agent,用于 GPU kernel 深度优化。作者指出脚本式 agent 与开放式 agent 都缺“方向”——即在对的时刻提出关于剩余差距的假设。KernelSeek 由通用编码 agent、强制可验证目标的最小 harness、以及决策点上的提问(先来自人类,后由 agent 从题库自问)构成。同一任务、评测器、硬件与一周时间内,工具堆满的主流自主 agent 在 133 步后停滞于 0.50× FlashAttention-3,KernelSeek 达到 1.13×,并生成无依赖的 FlashAttention 前向 kernel(不依赖 CUTLASS、CuTe、cuBLAS、cuDNN),在 Hopper 与 Blackwell 上全部 182 个场景正确,对 FA-3/FA-4 几何平均分别达 1.16× 与 1.46×。

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →