pith. sign in

arxiv: 2605.30359 · v1 · pith:T4RS5ZXHnew · submitted 2026-05-08 · 💻 cs.NE · cs.DC· cs.LG· cs.PF· cs.SE· cs.SY· eess.SY

Kernel Foundry: A Diagnosis-driven Evolutionary Kernel Optimizer with Multi-Experts

classification 💻 cs.NE cs.DCcs.LGcs.PFcs.SEcs.SYeess.SY
keywords kernelcorrectnessevolutionarykernelsoptimizationdiagnosis-drivenfoundrymethod
0
0 comments X
read the original abstract

Generating high-performance GPU kernels remains challenging due to the need for both correctness and hardware-aware optimization. While large language models (LLMs) show promise in code generation, they often fail to produce kernels that are both correct and efficient. We propose Kernel Foundry, a diagnosis-driven evolutionary framework for automatic GPU kernel optimization. Our method combines expert-guided, retrieval-augmented initialization with a multi-island evolutionary search, where candidate kernels are iteratively refined using structured diagnostic feedback. A centralized experience library accumulates reusable optimization knowledge to guide subsequent evolution, while explicit mechanisms prevent cheating behaviors that bypass kernel-level computation. Experiments on KernelBench show that our method consistently improves both correctness and performance over strong baselines, achieving up to 100% correctness on Level~2.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.