FastKernels:面向生产环境的GPU内核生成基准测试 / FastKernels: Benchmarking GPU Kernel Generation in Production
1️⃣ 一句话总结
本文提出了一个名为FastKernels的新型基准测试框架,它包含46种代表性架构和配套的生产级推理系统,旨在解决现有GPU内核生成基准与真实生产环境严重脱节的问题,实验表明即便最强的AI代理在内核生成中仅能达到与生产基线相当的性能,凸显了基准对齐对提升实际部署效率的关键作用。
LLM-based agents for GPU kernel generation are advancing rapidly, yet their progress is fundamentally constrained by the benchmarks they optimize against. Existing benchmarks are poorly aligned with production inference frameworks: they evaluate kernels on a single GPU with synthetic inputs, ignore the surrounding compilation stack, and reward replicating known optimizations rather than discovering new ones. The resulting reward signals are misleading: agents learn to generate kernels that score well in sandboxes but introduce interface incompatibilities, compilation-stack conflicts, and silent correctness degradation when integrated into real systems. We introduce FastKernels, a kernel benchmark built around a minimal set of 46 representative architectures spanning 8 categories, whose kernels collectively subsume those of 96.2% (409/425) of HuggingFace Transformers architectures. FastKernels doubles as a minimalistic, production-grade inference framework that runs at parity with hardened systems such as vLLM and SGLang on mainstream LLM serving and substantially exceeds upstream references on under-served architectures; each task's interface mirrors the corresponding module in the state-of-the-art library for its architecture family, enabling direct deployment of optimized kernels into production codebases. Evaluating state-of-the-art kernel agents on FastKernels, we find that even the strongest agent achieves only 0.94$\times$ aggregate speedup over production baselines, with weaker agents at $0.78\times$ and $0.53\times$ -- confirming that benchmark-production misalignment is a critical bottleneck for the field. We release FastKernels as a stepping stone toward kernel agents whose benchmark gains translate directly into production throughput improvements. Code is available at this https URL
FastKernels:面向生产环境的GPU内核生成基准测试 / FastKernels: Benchmarking GPU Kernel Generation in Production
本文提出了一个名为FastKernels的新型基准测试框架,它包含46种代表性架构和配套的生产级推理系统,旨在解决现有GPU内核生成基准与真实生产环境严重脱节的问题,实验表明即便最强的AI代理在内核生成中仅能达到与生产基线相当的性能,凸显了基准对齐对提升实际部署效率的关键作用。
源自 arXiv: 2605.23215