arXiv · cs.AI 人工智能· Songqiao Su, Xiaoya Li, Albert Wang, Guoyin Wang, Jiwei Li, Chris Shum·2026-11-13 13:00· 4 小时前精选AI 评分78

CUDA-L2：用大语言模型与强化学习自动优化 HGEMM 内核，性能超过 cuBLAS

CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learning

AI 导读

CUDA-L2 以 CUDA 执行速度为强化学习奖励，自动优化半精度通用矩阵乘（HGEMM）的 CUDA 内核，覆盖 1000 组配置。

推荐理由

论文用 LLM 加强化学习自动优化 HGEMM 内核，并在 1000 组配置上给出与 cuBLAS 系列库的逐项对比。