研究人员开发了 MaxKernel,这是一个多智能体系统,旨在自动创建高性能的定制内核,用于张量处理单元 (TPU)。该系统采用三种不同的范式:用于协作设计的“人在回路”智能体、用于指标驱动优化的“自主”智能体,以及用于更广泛探索的“基于图的自主搜索”。MaxKernel 利用专门的子智能体进行规划、实现、调试、测试和硬件分析,展示了其生成优化内核的能力,这些内核在基准测试套件和实际工作负载中可媲美专家调优的基线。 AI
影响 该系统可以显著加速优化 AI 硬件内核的开发,有可能降低定制加速器编程的门槛。
排序理由 研究论文,详细介绍了一个用于内核生成的新智能系统。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →