mlsanityData to 7 Oct 2026

Agentic Kernel Optimization: Generating State-of-the-Art GPU Kernels Without Hand-Written CUDA

arXiv 2608.14560 ↗First posted 25 May 2026cs.DC cs.SEcited by 0

On the map: LLM agents and tool use → Software engineering coding agents

also in 1 more
Abstract (arXiv)

We study whether general-purpose code agents can produce state-of-the-art GPU kernels without any manually written CUDA code. We investigate this question using representative workloads from FlashInfer-Bench, focusing on the Fused MoE, DSA TopK Indexer, and DSA Sparse Attention, and evaluate all generated kernels under the correctness-gated FlashInfer-Bench protocol on NVIDIA B200 GPUs. Starting from the PyTorch implementations, workload definitions, benchmark commands, and a compact set of CUDA optimization skills, we build a kernel optimization workflow in Houmao, a multi-agent orchestration framework for heterogeneous coding agents, to generate, debug, profile, and optimize the kernels. Humans remain strictly in an orchestration role: defining the workflow, enforcing correctness and anti-hacking constraints, supplying key references, and redirecting the search when progress stalls, without reviewing or editing the kernel code itself. Across roughly 1.9 billion agent tokens, the resulting kernels achieve speedups of 92.68x on Fused MoE, 1101.02x on DSA TopK Indexer, and 181.35x on DSA Sparse Attention relative to the PyTorch reference implementations, while also significantly outperforming the corresponding FlashInfer baselines. In the official evaluation of the MLSys 2026 FlashInfer AI Kernel Generation Contest, our generated Fused MoE kernel achieves a 1.71x speedup over the FlashInfer baseline, exceeding the top result of the Fused MoE agent-assisted track, which reports a 1.68x speedup. These results suggest that, under a disciplined correctness-first workflow, code agents can serve as effective autonomous optimizers for modern GPU kernel development.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Software engineering coding agents, one of 16 directions in LLM agents and tool use. Builds on 7 of its 15 references that are in the map. First posted on arXiv 25 May 2026.

Builds on

7 of 15 references.

Closest ancestors · 2

Same direction, newest first

  1. 2025AKG kernel Agent: A Multi-Agent Framework for Cross-Platform Kernel Synthesis29 Dec
  2. 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization9 Sep

Other references · 5

  1. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  2. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  3. 2022MegaBlocks: Efficient Sparse Training with Mixture-of-Expertscited by 204
  4. 2026Dr. Kernel: Reinforcement Learning Done Right for Triton Kernel Generationscited by 21
  5. 2025TritonForge: Profiling-Guided Framework for Automated Triton Kernel Optimizationcited by 15

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026AgentKernelArena: Generalization-Aware Benchmarking of GPU Kernel Optimization Agents
  2. 2026KernelArc: A Multi-Agent Framework for GPU Kernel Optimization
  3. 2025STARK: Strategic Team of Agents for Refining Kernels
  4. 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization
  5. 2026A Two-Stage GPU Kernel Tuner Combining Semantic Refactoring and Search-Based Optimization
  6. 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidance
  7. 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTe
  8. 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimization