mlsanityData to 7 Oct 2026

KernelFoundry: Hardware-aware evolutionary GPU kernel optimization

arXiv 2603.12440 ↗First posted 12 Mar 2026cs.DC cs.LGcited by 11

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

  1. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  2. 2025AlphaEvolve: A coding agent for scientific and algorithmic discoverycited by 763
  3. 2015Illuminating search spaces by mapping elitescited by 406
  4. 2014cuDNN: Efficient Primitives for Deep Learningcited by 240
  5. 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
  6. 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 199
  7. 2018Learning to Optimize Tensor Programscited by 138
  8. 2022Evolution through Large Modelscited by 131
  9. 2025Establishing Best Practices for Building Rigorous Agentic Benchmarkscited by 94
  10. 2019Covariance Matrix Adaptation for the Rapid Illumination of Behavior Spacecited by 78
  11. 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 73
  12. 2021Differentiable Quality Diversitycited by 71
24 more
  1. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 68
  2. 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 59
  3. 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimizationcited by 52
  4. 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 46
  5. 2018Ithemal: Accurate, Portable and Fast Basic Block Throughput Estimation using Deep Neural Networkscited by 43
  6. 2025CudaForge: An Agent Framework with Hardware Feedback for CUDA Kernel Optimizationcited by 42
  7. 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 35
  8. 2025CUDA-LLM: LLMs Can Write Efficient CUDA Kernelscited by 33
  9. 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarkscited by 31
  10. 2023Comparing Llama-2 and GPT-3 LLMs for HPC kernels generationcited by 30
  11. 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
  12. 2025STARK: Strategic Team of Agents for Refining Kernelscited by 23
  13. 2023Evaluation of OpenAI Codex for HPC Parallel Programming Models Kernel Generationcited by 22
  14. 2026PACEvolve: Enabling Progress-Aware Consistent Evolutioncited by 21
  15. 2024CodeRosetta: Pushing the Boundaries of Unsupervised Code Translation for Parallel Programmingcited by 19
  16. 2025QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generationcited by 16
  17. 2025TritonForge: Profiling-Guided Framework for Automated Triton Kernel Optimizationcited by 15
  18. 2025CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learningcited by 13
  19. 2025SwizzlePerf: Hardware-Aware LLMs for GPU Kernel Performance Optimizationcited by 11
  20. 2025PRAGMA: A Profiling-Reasoned Multi-Agent Framework for Automatic Kernel Optimizationcited by 11
  21. 2025cuPilot: A Strategy-Coordinated Multi-agent Framework for CUDA Kernel Evolutioncited by 7
  22. 2025AKG kernel Agent: A Multi-Agent Framework for Cross-Platform Kernel Synthesiscited by 7
  23. 2025NPUEval: Optimizing NPU Kernels with LLMs and Open Source Compilerscited by 6
  24. 2025PEAK: A Performance Engineering AI-Assistant for GPU Kernels Powered by Natural Language Transformationscited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 11.

11 papers.