arXiv 2502.10517
KernelBench: Can LLMs Write Efficient GPU Kernels?
By Anne Ouyang, Simon Guo, et al.
Published 2025-02-14
Mindmap
Browse the paper's core ideas, clusters, and relationships in a structured outline.
Efficient GPU kernels are crucial for building performant machine learning architectures, but writing them is a time-consuming challenge that requires significant expertise; therefore, we explore using language models (LMs) to automate kernel generation. We introduce KernelBench, an open-source framework for evaluating LMs' ability to write fast and correct kernels on a suite of 250 carefully selected PyTorch ML wor…