DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels
Not on the map. Why?
Abstract (arXiv)
Diffusion large language models (dLLMs) have emerged as a compelling alternative to autoregressive (AR) LLMs, owing to their capacity for parallel token generation. This paradigm is particularly well-suited for code generation, where holistic structural planning and non-sequential refinement are critical. Despite this potential, tailoring dLLMs for CUDA kernel generation remains challenging, obstructed not only by the high specialization but also by the severe lack of high-quality training data. To address these challenges, we construct CuKe, an augmented supervised fine-tuning dataset optimized for high-performance CUDA kernels. On top of it, we propose a bi-phase curated reinforcement learning (BiC-RL) framework consisting of a CUDA kernel infilling stage and an end-to-end CUDA kernel generation stage. Leveraging this training framework, we introduce DICE, a series of diffusion large language models designed for CUDA kernel generation, spanning three parameter scales, 1.7B, 4B, and 8B. Extensive experiments on KernelBench demonstrate that DICE significantly outperforms both autoregressive and diffusion LLMs of comparable scale, establishing a new state-of-the-art for CUDA kernel generation.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 43 of its 49 references that are in the map. First posted on arXiv 12 Feb 2026.
Builds on
43 of 49 references.
References · 43
- 2023GPT-4 Technical Reportcited by 18,132
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2025Qwen3 Technical Reportcited by 8,195
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
- 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,442
- 2024DeepSeek-Coder: When the Large Language Model Meets Programming -- The Rise of Code Intelligencecited by 1,154
- 2024Qwen2.5-Coder Technical Reportcited by 1,151
- 2025Large Language Diffusion Modelscited by 872
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 385
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 273
31 more
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 225
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
- 2022SSD-LM: Semi-autoregressive Simplex-based Diffusion Language Model for Text Generation and Modular Controlcited by 191
- 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generationcited by 184
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 173
- 2025LLaDA2.0: Scaling Up Diffusion Language Models to 100Bcited by 154
- 2025SDAR: A Synergistic Diffusion-AutoRegression Paradigm for Scalable Sequence Generationcited by 118
- 2025Dream-Coder 7B: An Open Diffusion Language Model for Codecited by 81
- 2025Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Modelscited by 80
- 2025A Survey on Diffusion Language Modelscited by 75
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 73
- 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 68
- 2025Diffusion Beats Autoregressive in Data-Constrained Settingscited by 61
- 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 59
- 2025Seed-Coder: Let the Code Model Curate Data for Itselfcited by 46
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 46
- 2025Diffusion Language Models are Super Data Learnerscited by 45
- 2025Phi-4-Mini-Reasoning: Exploring the Limits of Small Reasoning Language Models in Mathcited by 44
- 2025CudaForge: An Agent Framework with Hardware Feedback for CUDA Kernel Optimizationcited by 42
- 2025Teaching Language Models to Critique via Reinforcement Learningcited by 38
- 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMscited by 38
- 2025WeDLM: Reconciling Diffusion Language Models with Standard Causal Attention for Fast Inferencecited by 38
- 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 35
- 2023CodeFusion: A Pre-trained Diffusion Model for Code Generationcited by 31
- 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
- 2025Training Optimal Large Diffusion Language Modelscited by 21
- 2026FlashInfer-Bench: Building the Virtuous Cycle for AI-driven LLM Systemscited by 20
- 2025Unifying Autoregressive and Diffusion-Based Sequence Generationcited by 17
- 2025Exploring the Potential of Diffusion Large Language Models in Code Generationcited by 16
- 2025ConCuR: Conciseness Makes State-of-the-Art Kernel Generationcited by 13
- 2026Stable-DiffCoder: Pushing the Frontier of Code Diffusion Large Language Modelcited by 12
Built on by
Citing papers per year: 2026: 1.
1 paper.