跳到正文
原文
arXiv 机器学习· Byonggon Chun·· 8 小时前AI 评分54

面向新型AI加速器的多智能体算子优化循环案例研究

A Case Study of a Multi-Agent Kernel-Optimization Loop That Learns a New AI Accelerator

AI 导读

论文提出一种面向新型AI加速器的多智能体算子优化闭环,能针对罕见硬件架构与缺乏文档的编程语言自主构建软硬件知识。研究以FuriosaAI RNGD NPU及其Rust DSL为例,在12天内对Gemma-4-12B的三个解码算子完成优化,获得相比官方基线9.293倍的几何平均加速比,达到权重搬运理论周期下限的1.25倍以内。

来源:arXiv 机器学习 · arxiv.org