跳到正文
HuggingFace Daily Papers· Gabriele Oliaro, Jaeseong Lee, Zhaoyuan Su, Sreeram Vennam, Zhihao Jia, Samyam Rajbhandari·· 4 天前精选AI 评分65

提问而非装备:面向深度内核优化的问答式编码智能体

Ask, Don't Equip: Question-Driven Coding Agents for Deep Kernel Optimization

AI 导读

研究人员提出KernelSeek,一种基于提问的GPU内核优化智能体,无需特定领域知识或参考实现即可提升CUDA内核性能。在相同任务下,KernelSeek达到1.13x FlashAttention-3的性能,优于传统装备丰富型智能体的0.50x。该智能体通过在决策点提出问题引导优化,而非依赖更多工具或信息。

推荐理由

通过提问而非装备工具引导智能体优化GPU内核,KernelSeek在相同任务中达到1.13x FlashAttention-3的性能,证明精准提问比提供更多信息更有效。

来源:HuggingFace Daily Papers · arxiv.org