mlsanityData to 7 Oct 2026

GPU Kernel Optimization Beyond Full Builds: An LLM Framework with Minimal Executable Programs

arXiv 2512.22147 ↗First posted 15 Dec 2025cs.DC cs.AI cs.LGcited by 3

Not on the map. Why?

Abstract (arXiv)

In high-performance computing, hotspot GPU kernels are primary bottlenecks, and expert manual tuning is costly and hard to port. Large language model methods often assume kernels can be compiled and executed cheaply, which fails in large applications where full builds and runs are expensive. We present an end-to-end LLM framework with performance feedback that optimizes kernels without building the full application. From independently extracted hotspot kernels, it automatically completes code into a Minimal Executable Program (MEP), then performs multi-round iterative optimization and evaluation outside the full application. The framework integrates Automatic Error Repair and Performance Pattern Inheritance to fix faults, preserve correctness, reuse effective tiling/memory/synchronization strategies, and reduce search cost. Optimized variants are reintegrated into the original application for validation. We evaluate on NVIDIA GPUs and the Haiguang Deep Computing Unit (DCU) platform (AMD-licensed architecture) using PolyBench, the AMD APP SDK, and hotspot kernels from large-scale supercomputing applications. The method achieves average speedups of 5.05x (PolyBench on NVIDIA), 7.77x (PolyBench on DCU), 1.77x (AMD APP SDK), and 1.25x on three hotspot kernels, surpassing direct LLM optimization. The approach requires no full-source dependencies, offers cross-platform portability, and enables practical, low-cost GPU kernel optimization.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 16 of its 28 references that are in the map. First posted on arXiv 15 Dec 2025.

Builds on

16 of 28 references.

References · 16

  1. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  2. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  3. 2022Competition-Level Code Generation with AlphaCodecited by 1,427
  4. 2016TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systemscited by 660
  5. 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
  6. 2014cuDNN: Efficient Primitives for Deep Learningcited by 240
  7. 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
  8. 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 199
  9. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 68
  10. 2020MLIR: A Compiler Infrastructure for the End of Moore's Lawcited by 64
  11. 2024Mirage: A Multi-Level Superoptimizer for Tensor Programscited by 39
  12. 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 35
4 more
  1. 2024ThunderKittens: Simple, Fast, and Adorable AI Kernelscited by 34
  2. 2025CUDA-LLM: LLMs Can Write Efficient CUDA Kernelscited by 33
  3. 2024LLM-Vectorizer: LLM-based Verified Loop Vectorizercited by 25
  4. 2025GPU Kernel Scientist: An LLM-Driven Framework for Iterative Kernel Optimizationcited by 22

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.