arXiv:2605.23215cs.LGcs.AI2026-05被引 2

为大模型生成的GPU内核设计了更贴近真实部署的基准测试。

FastKernels: Benchmarking GPU Kernel Generation in Production

论文配图:FastKernels: Benchmarking GPU Kernel Generation in Production
图 1 · 摘自论文原文
  • 构建了46个代表性架构的生产级内核基准,覆盖96.2% HuggingFace模型
  • 实测最强内核代理仅比现有方案快0.94倍,暴露了评测与实际部署脱节问题
  • 基准支持直接部署到生产代码库,适合作为内核优化研究的可靠测试平台

基于大语言模型的GPU内核生成代理发展迅速,但其进展受限于所依赖的基准测试。现有基准与真实生产推理框架严重脱节:仅在单张GPU上用合成输入评估内核,忽略编译栈影响,且奖励复制已有优化而非发现新方法。这导致奖励信号失真:代理学会生成在沙箱中表现良好但在实际系统中引发接口不兼容、编译冲突和隐性正确性下降的代码。本文提出FastKernels,一个围绕46个代表性架构(8类)构建的内核基准,其内核涵盖HuggingFace Transformers中96.2%(409/425)的架构。FastKernels同时作为最小化、生产级推理框架,主流LLM服务性能媲美vLLM和SGLang,对未充分支持架构的性能远超上游基准;每个任务接口与对应架构最先进库模块一致,支持优化内核直接部署至生产代码库。在该基准上评估最先进内核代理,最强者仅实现0.94×整体加速,弱者分别为0.78×和0.53×,证实评测与生产不匹配是领域关键瓶颈。项目代码已开源。

原文摘要 · Abstract (English)

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 https://github.com/Snowflake-AI-Research/fastkernels

GPU内核大模型推理基准测试生产部署

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。