跳到正文
arXiv · ML· Qitong Sun, Jun Han, Tianlin Li, Zhe Tang, Sheng Chen, Fei Yang, Aishan Liu, Xianglong Liu, Yang Liu·· 9 小时前AI 评分42

KernelClaw: 基于双记忆机制的多智能体GPU内核优化框架

KernelClaw: A Multi-Agent Framework for GPU Kernel Optimization with Dual-Memory Mechanisms

AI 导读

KernelClaw 通过双级记忆架构协调多智能体,取代 LLM 隐式启发式以实现可解释的 GPU 内核优化。 该框架在 KernelBench 1-3 级达到 100% 成功率,平均加速比分别为 6.98x、7.89x 和 1.94x。 代码已开源,性能优于先前基线。

来源:arXiv · ML · arxiv.org