mlsanityData to 14 Oct 2026

AgentKernelArena: Generalization-Aware Benchmarking of GPU Kernel Optimization Agents

arXiv 2605.16819 ↗First posted 16 May 2026cs.CL cs.AI cs.LGcited by 2

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

Abstract (arXiv)

GPU kernel optimization is increasingly critical for efficient deep learning systems, but writing high-performance kernels still requires substantial low-level expertise. Recent AI coding agents can iteratively read code, invoke compilers and profilers, and refine implementations, yet existing kernel benchmarks evaluate single LLM calls rather than full agent workflows, and none include both kernel-to-kernel optimization and unseen-configuration generalization testing. We present AgentKernelArena, an open-source benchmark for measuring AI coding agents on GPU kernel optimization. The benchmark contains 196 tasks spanning HIP-to-HIP optimization, Triton-to-Triton optimization, and PyTorch-to-HIP translation, and evaluates complete agent workflows in isolated workspaces using gated compilation, correctness, and performance checks, centralized scoring and an unseen-configuration generalization protocol that tests whether optimizations transfer to input configurations the agent never observed. Across production agents including Cursor Agent, Claude Code, and Codex Agent, we find near-perfect compilation and high correctness rates on most task categories, with the strongest configurations achieving mean speedups of up to 6.89x on PyTorch-to-HIP, 6.69x on HIP-to-HIP, and 2.13x on Triton-to-Triton tasks. Our unseen-configuration evaluation shows that HIP-to-HIP and Triton-to-Triton optimizations largely transfer to unseen input shapes, while PyTorch-to-HIP exhibits substantial correctness drops, indicating that agents generating kernels from scratch frequently hardcode shape-specific assumptions. AgentKernelArena is designed as a modular, extensible framework for rigorous evaluation of agentic GPU kernel optimization across agents, tasks, and hardware targets.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Software engineering coding agents, one of 16 directions in LLM agents and tool use. Builds on 14 of its 20 references that are in the map. First posted on arXiv 16 May 2026.

Builds on

14 of 20 references.

Closest ancestors · 2

Same direction, newest first

  1. 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimization16 Sep
  2. 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineering6 May

Other references · 12

  1. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  2. 2021Program Synthesis with Large Language Modelscited by 3,637
  3. 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,373
  4. 2023AgentBench: Evaluating LLMs as Agentscited by 1,267
  5. 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
  6. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
  7. 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMscited by 40
  8. 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 38
  9. 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarkscited by 32
  10. 2025QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generationcited by 19
  11. 2025TritonForge: Profiling-Guided Framework for Automated Triton Kernel Optimizationcited by 16
  12. 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generationcited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Agentic Kernel Optimization: Generating State-of-the-Art GPU Kernels Without Hand-Written CUDA
  2. 2025STARK: Strategic Team of Agents for Refining Kernels
  3. 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimization
  4. 2026KernelArc: A Multi-Agent Framework for GPU Kernel Optimization
  5. 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization
  6. 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTe
  7. 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization
  8. 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidance