DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels
DICE:扩散大语言模型在生成CUDA内核方面表现出色
机构 * Westlake University(西湖大学) ; Hong Kong University of Science and Technology(香港科技大学) ; Rochester Institute of Technology(罗切斯特理工学院)
专题命中 代码生成 :code generation(abstract);分类 cs.CL、cs.LG
AI总结 提出CuKe数据集和BiC-RL训练框架,构建DICE系列扩散大语言模型(1.7B/4B/8B),在KernelBench上显著优于同类自回归和扩散模型,实现CUDA内核生成新SOTA。
Comments v2: Expanded with dLLM vs. autoregressive LLM comparisons, ablation studies, and qualitative case studies