Research & Papers

Houmao agents auto-generate GPU kernels, beat hand-written CUDA by 1101x

AI-coded kernels outperformed FlashInfer baselines—no human CUDA required.

Deep Dive

A new arXiv paper from Mao Luo and colleagues demonstrates that general-purpose code agents can autonomously generate state-of-the-art GPU kernels without any manually written CUDA. The team built Houmao, a multi-agent orchestration framework that coordinates heterogeneous coding agents to generate, debug, profile, and optimize kernels. Humans only defined the workflow, enforced correctness constraints, and supplied references—never reviewing or editing the actual kernel code. The experimentation used FlashInfer-Bench workloads—Fused MoE, DSA TopK Indexer, and DSA Sparse Attention—evaluated on NVIDIA B200 GPUs under a correctness-gated protocol.

Across roughly 1.9 billion agent tokens, the generated kernels achieved remarkable speedups over PyTorch baselines: 92.68x on Fused MoE, 1101.02x on DSA TopK Indexer, and 181.35x on DSA Sparse Attention. They also significantly outperformed the hand-optimized FlashInfer baselines. In the official MLSys 2026 FlashInfer AI Kernel Generation Contest, Houmao's generated Fused MoE kernel scored a 1.71x speedup over FlashInfer—beating the top result from the agent-assisted track (1.68x). This signals that disciplined, correctness-first AI workflows can match or exceed human expertise in low-level GPU programming.

Key Points
  • Houmao generated CUDA-free kernels using 1.9B agent tokens across a multi-agent workflow
  • Achieved 92.68x speedup on Fused MoE, 1101.02x on DSA TopK, 181.35x on Sparse Attention vs PyTorch
  • MLSys 2026 contest: 1.71x over FlashInfer baseline, beating the best human-assisted 1.68x result

Why It Matters

AI could replace specialized CUDA engineers, cutting GPU kernel dev time from weeks to hours.

📬 Get the top 10 AI stories daily