Paper#AI Kernel Generation🔬 Research分析: 2026年1月3日 16:06

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

发布:2025年12月29日 12:42
1分で読める
ArXiv

分析

本文解决了AI系统开发中手动内核优化的关键瓶颈问题,特别是考虑到AI模型的日益复杂性和硬件平台的多样性。 提出的多代理系统AKG内核代理利用LLM代码生成来自动化跨多个DSL和硬件后端的内核生成、迁移和调优。 与基线实现相比,所展示的加速突出了这种方法的实际影响。

引用

AKG内核代理在PyTorch Eager基线实现上实现了平均1.46倍的加速。