MaxKernel

MaxKernel 系统通过多智能体方法自动生成 TPU 内核

研究人员开发了 MaxKernel，这是一个多智能体系统，旨在自动创建高性能的定制内核，用于张量处理单元 (TPU)。该系统采用三种不同的范式：用于协作设计的“人在回路”智能体、用于指标驱动优化的“自主”智能体，以及用于更广泛探索的“基于图的自主搜索”。MaxKernel 利用专门的子智能体进行规划、实现、调试、测试和硬件分析，展示了其生成优化内核的能力，这些内核在基准测试套件和实际工作负载中可媲美专家调优的基线。