arXiv cs.LG· Byonggon Chun·· 4 天前AI 评分44
多智能体 Kernel 优化循环适配新型 AI 加速器案例研究
A Case Study of a Multi-Agent Kernel-Optimization Loop That Learns a New AI Accelerator
AI 导读
研究人员构建了一种多智能体优化循环，并在 FuriosaAI RNGD NPU 上历时 12 天优化了 Gemma-4-12B 的三个解码算子。该循环通过自主建立 Rust DSL 文档知识、运行微基准测试与编译器合法性检查，结合设计树搜索探索算子实现。最终内核相比官方 baseline 取得 9.293x 几何平均加速比，相关提示词与测试工具均已开源。
来源：arXiv cs.LG · arxiv.org