CANN Bench: Benchmarking Agent Generated Kernels against Real NPU and Algorithmic Limits
Not on the map. Why?
Abstract (arXiv)
AI agents are now capable of writing, compiling, and iteratively optimizing low-level operator kernels on different hardware platforms. Existing benchmarks, however, focus almost exclusively on CUDA and Triton, leaving hardware ecosystems with less-exposed programming models without a common evaluation baseline. We present CANN Bench, an open benchmark for AI-generated operator code on Huawei's Ascend NPU. The current release covers 53 operators and 1060 test cases organized into four difficulty tiers -- from simple elementwise primitives to MoE dispatch and FlashAttention kernels -- spanning FP16, BF16, FP32, and INT8 precision formats. Evaluation adopts a three-dimensional weighted composite score that treats compilation, functional correctness, and performance as independent axes, providing a principled reward signal for kernel-generation agents. Performance is graded against an out-of-the-box PyTorch-on-Ascend baseline and an analytical per-case Hardware-Anchored Performance (HAP) limit on real NPU hardware, ensuring scores reflect genuine optimization headroom rather than measurement artifacts. The evaluation harness is designed to resist reward hacking from the ground up. CANN Bench is versioned within the official CANN repository and is designed for long-term community co-construction, providing the Ascend ecosystem with a quantitative, reproducible, and sustainably maintained yardstick for AI operator-authoring capability.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 20 of its 31 references that are in the map. First posted on arXiv 8 Jul 2026.
Builds on
20 of 31 references.
References · 20
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
- 2023Microscaling Data Formats for Deep Learningcited by 132
- 2026Conditional Memory via Scalable Lookup: A New Axis of Sparsity for Large Language Modelscited by 78
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 73
- 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 68
- 2026CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generationcited by 41
- 2025Training LLMs with MXFP4cited by 36
- 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 35
- 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarkscited by 31
- 2025GPU Kernel Scientist: An LLM-Driven Framework for Iterative Kernel Optimizationcited by 22
- 2026Dr. Kernel: Reinforcement Learning Done Right for Triton Kernel Generationscited by 21
- 2026AutoKernel: Autonomous GPU Kernel Optimization via Iterative Agent-Driven Searchcited by 14
8 more
- 2025EvoEngineer: Mastering Automated CUDA Kernel Code Evolution with Large Language Modelscited by 13
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limitscited by 13
- 2026AscendKernelGen: A Systematic Study of LLM-Based Kernel Generation for Neural Processing Unitscited by 11
- 2026AscendCraft: Automatic Ascend NPU Kernel Generation via DSL-Guided Transcompilationcited by 6
- 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTecited by 6
- 2026KernelCraft: Benchmarking for Agentic Close-to-Metal Kernel Generation on Emerging Hardwarecited by 4
- 2026Towards Cold-Start Drafting and Continual Refining: A Value-Driven Memory Approach with Application to NPU Kernel Synthesiscited by 3
- 2026AscendOptimizer: Episodic Agent for Ascend NPU Operator Optimizationcited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?
- 2026AgentKernelArena: Generalization-Aware Benchmarking of GPU Kernel Optimization Agents
- 2026AgenticCANN: Automated Ascend C Operator Generation via Knowledge-Augmented Agentic Evolution
- 2026Agentic Kernel Optimization: Generating State-of-the-Art GPU Kernels Without Hand-Written CUDA
- 2025AKG kernel Agent: A Multi-Agent Framework for Cross-Platform Kernel Synthesis
- 2025CudaForge: An Agent Framework with Hardware Feedback for CUDA Kernel Optimization
- 2025PRAGMA: A Profiling-Reasoned Multi-Agent Framework for Automatic Kernel Optimization
- 2025STARK: Strategic Team of Agents for Refining Kernels