KernelFoundry: Hardware-aware evolutionary GPU kernel optimization
Not on the map. Why?
Abstract (arXiv)
GPU kernel optimization challenges LLMs beyond standard coding tasks, as it requires an understanding of hardware architecture, parallel computing optimization strategies, and profiling outputs. However, most existing approaches leveraging LLMs for kernel generation apply standard prompting and feedback loops, considering hardware only through profiling feedback. We introduce KernelFoundry, an evolutionary framework that efficiently explores the space of GPU kernels through (1) MAP-Elites quality diversity search with kernel-specific behavioral dimensions to sustain exploration; (2) meta-prompt evolution that co-evolves prompts with kernels to uncover task-specific optimization strategies, and (3) a template-based parameter optimization approach to tune kernels to inputs and hardware. We evaluate this framework on Kernel-Bench, robust-kbench and custom tasks, generating SYCL kernels as a cross-platform GPU programming paradigm, and CUDA kernels for comparison to prior work. Our approach consistently outperforms the baseline methods and achieves an average speedup of 2.3 on KernelBench for SYCL. Moreover, KernelFoundry is implemented as a distributed framework with remote access to diverse hardware, allowing quick benchmarking and featuring a flexible user input layer to support kernel generation for a wide range of real use cases beyond benchmarking.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 36 of its 60 references that are in the map. First posted on arXiv 12 Mar 2026.
Builds on
36 of 60 references.
References · 36
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2025AlphaEvolve: A coding agent for scientific and algorithmic discoverycited by 763
- 2015Illuminating search spaces by mapping elitescited by 406
- 2014cuDNN: Efficient Primitives for Deep Learningcited by 240
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 199
- 2018Learning to Optimize Tensor Programscited by 138
- 2022Evolution through Large Modelscited by 131
- 2025Establishing Best Practices for Building Rigorous Agentic Benchmarkscited by 94
- 2019Covariance Matrix Adaptation for the Rapid Illumination of Behavior Spacecited by 78
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 73
- 2021Differentiable Quality Diversitycited by 71
24 more
- 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 68
- 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 59
- 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimizationcited by 52
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 46
- 2018Ithemal: Accurate, Portable and Fast Basic Block Throughput Estimation using Deep Neural Networkscited by 43
- 2025CudaForge: An Agent Framework with Hardware Feedback for CUDA Kernel Optimizationcited by 42
- 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 35
- 2025CUDA-LLM: LLMs Can Write Efficient CUDA Kernelscited by 33
- 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarkscited by 31
- 2023Comparing Llama-2 and GPT-3 LLMs for HPC kernels generationcited by 30
- 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
- 2025STARK: Strategic Team of Agents for Refining Kernelscited by 23
- 2023Evaluation of OpenAI Codex for HPC Parallel Programming Models Kernel Generationcited by 22
- 2026PACEvolve: Enabling Progress-Aware Consistent Evolutioncited by 21
- 2024CodeRosetta: Pushing the Boundaries of Unsupervised Code Translation for Parallel Programmingcited by 19
- 2025QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generationcited by 16
- 2025TritonForge: Profiling-Guided Framework for Automated Triton Kernel Optimizationcited by 15
- 2025CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learningcited by 13
- 2025SwizzlePerf: Hardware-Aware LLMs for GPU Kernel Performance Optimizationcited by 11
- 2025PRAGMA: A Profiling-Reasoned Multi-Agent Framework for Automatic Kernel Optimizationcited by 11
- 2025cuPilot: A Strategy-Coordinated Multi-agent Framework for CUDA Kernel Evolutioncited by 7
- 2025AKG kernel Agent: A Multi-Agent Framework for Cross-Platform Kernel Synthesiscited by 7
- 2025NPUEval: Optimizing NPU Kernels with LLMs and Open Source Compilerscited by 6
- 2025PEAK: A Performance Engineering AI-Assistant for GPU Kernels Powered by Natural Language Transformationscited by 5
Built on by
Citing papers per year: 2026: 11.
11 papers.
- 2026AutoKernel: Autonomous GPU Kernel Optimization via Iterative Agent-Driven Searchcited by 14
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?cited by 6
- 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimizationcited by 4
- 2026What Do Evolutionary Coding Agents Evolve?cited by 3
- 2026Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPUcited by 1
- 2026Prism: Symbolic Superoptimization of Tensor Programscited by 0
- 2026Learning When to Optimize: Verified Optimization Skills from Expert GPU-Kernel Lineagescited by 0
- 2026Cross-Model Cross-Language AI Coding Agent Performance: Accuracy and Speed of Parallel CLRS Algorithmscited by 0
- 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimizationcited by 0
- 2026Don't Regenerate, Debug: A Domain-Specific Agent for Repairing Near-Miss Hardware Operatorscited by 0
- 2026Self-Evolving Coding Agents: From Digital Programs to Physical-World Intelligencecited by 0
- 2026Self-Evolving Coding Agents: From Digital Programs to Physical-World Intelligence28 Sep 2026
- 2026Don't Regenerate, Debug: A Domain-Specific Agent for Repairing Near-Miss Hardware Operators3 Aug 2026
- 2026Cross-Model Cross-Language AI Coding Agent Performance: Accuracy and Speed of Parallel CLRS Algorithms26 Jul 2026
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?26 Jun 2026
- 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimization24 Jun 2026
- 2026Learning When to Optimize: Verified Optimization Skills from Expert GPU-Kernel Lineages27 May 2026
- 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimization19 May 2026
- 2026What Do Evolutionary Coding Agents Evolve?19 May 2026
- 2026Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPU16 Apr 2026
- 2026Prism: Symbolic Superoptimization of Tensor Programs16 Apr 2026
- 2026AutoKernel: Autonomous GPU Kernel Optimization via Iterative Agent-Driven Search22 Mar 2026