arXiv Artificial Intelligence· Qitong Sun, Jun Han, Tianlin Li, Zhe Tang, Sheng Chen, Fei Yang, Aishan Liu, Xianglong Liu, Yang Liu·· 3 小时前AI 评分38
KernelClaw:面向 GPU Kernel 优化的双记忆多智能体框架
KernelClaw: A Multi-Agent Framework for GPU Kernel Optimization with Dual-Memory Mechanisms
AI 导读
KernelClaw 是一个带双层记忆架构的多智能体框架,用长期记忆复用专家优化知识、短期记忆避免重复回溯,替代 LLM 隐式启发式来优化 GPU kernel。在 KernelBench Levels 1-3 上成功率达 100%,相对 Torch Eager 的平均加速分别为 6.98x、7.89x 和 1.94x,优于此前基线。代码已公开。
来源:arXiv Artificial Intelligence · arxiv.org