字节跳动Seed与清华大学AIR联合发布了CUDA Agent,这是一套面向CUDA内核生成的智能体强化学习系统。其核心目标是解决一个具体而棘手的问题:前沿大模型虽能生成语法正确的CUDA代码,但性能往往不及编译器优化后的版本。该系统通过大规模强化学习训练语言模型,使其生成的内核在速度上超越传统编译器。在KernelBench基准测试中,基础模型Seed1.6的通过率为74.0%,而CUDA Agent在此基础上进一步提升了内核执行效率。这一成果的意义在于,它展示了强化学习在底层系统优化中的潜力,为自动生成高性能计算代码提供了新路径,有望降低对人工调优的依赖,并推动AI在编译器与硬件协同设计领域的应用。
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅