AKG内核代理:自动化AI工作负载的内核生成

Paper#AI Kernel Generation🔬 Research|分析: 2026年1月3日 16:06
发布: 2025年12月29日 12:42
1分で読める
ArXiv

分析

本文解决了AI系统开发中手动内核优化的关键瓶颈问题,特别是考虑到AI模型的日益复杂性和硬件平台的多样性。 提出的多代理系统AKG内核代理利用LLM代码生成来自动化跨多个DSL和硬件后端的内核生成、迁移和调优。 与基线实现相比,所展示的加速突出了这种方法的实际影响。
引用 / 来源
查看原文
"AKG kernel agent achieves an average speedup of 1.46x over PyTorch Eager baselines implementations."
A
ArXiv2025年12月29日 12:42
* 根据版权法第32条进行合法引用。