字节清华发布 CUDA Agent：RL 生成内核速度大幅超越 torch.compile

pmttyji · reddit · 2026-08-18

ByteDance Seed 与清华大学 SIA 联合发布了 CUDA Agent，这是一个基于大规模智能体强化学习（RL）的系统，旨在生成高性能 CUDA 内核。针对 GPU 内核优化这一高门槛任务，该系统通过可扩展的数据合成管道、增强的 CUDA 开发环境以及稳定的 RL 算法技术，让模型通过实际运行代码并获得反馈来学习优化。在 KernelBench 测试中，CUDA Agent 在 Level-1 到 Level-3 三个难度上分别实现了比 torch.compile 快 100%、100% 和 92% 的速率，在最难的 Level-3 设置上比 Claude Opus 4.5 和 Gemini 3 Pro 等最强私有模型快约 40%。