The Next Frontier for AI-Generated Kernels: Correctness
KernelBench was recently proposed as a GPU-programming challenge set for LLMs, measuring their ability to generate correct and efficient GPU kernels for a variety of problems. With the advances of LLMs, most frontier models can now pass most of KernelBench with ease, and the performance (and corresponding complexity) of the generated kernels is increasing. In contrast, the correctness aspect of these kernels has gotten relatively little attention. KernelBench uses a fixed test suite to accept a kernel as valid. Is this perhaps enough, given the usual regularity of GPU kernels? Do buggy kernels make it through this test suite?
We claim that testing is hopelessly inadequate for such tasks, finding serious problems in KernelBench’s test suite, with correct kernels being rejected and buggy kernels being accepted. While some of these issues are easily solvable, others are fundamental and unavoidable, particularly if one considers an adversarial LLM. We argue that AI-generated kernel solutions should come with formal proofs of correctness, and show that this is now practical. Using Kuiper, a recent verified GPU programming framework, we present verified implementations for all 100 KernelBench Level 1 tasks, written almost entirely by LLM coding agents
Mon 15 JunDisplayed time zone: Mountain Time (US & Canada) change
10:40 - 12:20 | |||
10:40 20mTalk | Agentic Code Reasoning PAgE Pre-print | ||
11:00 20mTalk | Towards Verified Code Reasoning by LLMs PAgE Meghana Aparna Sistla Google DeepMind, Gogul Balakrishnan Google, Patrick Rondon Google, José Pablo Cambronero Google, USA, Michele Tufano Google, Satish Chandra Meta Platforms, Inc. | ||
11:20 20mTalk | Testing, Credible Compilation, and Verification in the Axon Verified Compiler in Lean and Claude CodeRemote PAgE Martin C. Rinard Massachusetts Institute of Technology | ||
11:40 20mTalk | The Next Frontier for AI-Generated Kernels: Correctness PAgE DOI Pre-print | ||
12:00 20mTalk | Testing LLM-Generated Distributed Protocol Code PAgE | ||