StitchCUDA: An Automated Multi-Agents End-to-End GPU Programing Framework with Rubric-based Agentic Reinforcement Learning
StitchCUDA: An Automated Multi-Agents End-to-End GPU Programing Framework with Rubric-based Agentic Reinforcement Learning
总结
问题
方法
摘要
本文提出了 StitchCUDA，一个专为端到端 GPU 程序生成设计的自动化多智能体框架。该框架通过 Planner、Coder 和 Verifier 三个专门智能体的协作，结合基于评估准则（Rubric-based）的智能强化学习，在 KernelBench 榜单上实现了近 100% 的成功率，性能最高提升至基线的 2.73 倍。
-