为真实GPU核函数优化提供逼近硬件极限的基准测试
SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limits
- 构建235个生产级AI模型的CUDA内核问题,聚焦NVIDIA Blackwell架构
- 用理论硬件极限速度(SOL)替代软件基线,量化优化差距
- 内置防作弊机制,适合评估智能体优化器的真实性能
随着智能体系统在生成和优化GPU内核方面能力增强,现有基准测试因以超越软件基线为指标而制约了进展。本文提出SOL-ExecBench,包含从124个语言、扩散、视觉、音频、视频及混合架构的生产与新兴AI模型中提取的235个CUDA内核优化问题,目标适配NVIDIA Blackwell GPU。涵盖BF16、FP8、NVFP4下的前向与反向计算,部分内核最佳性能依赖Blackwell特有功能。不同于以往以软件实现为参照的基准,SOL-ExecBench通过SOLAR管道计算分析得出的硬件理论极限速度(SOL),设定固定优化目标。报告的SOL Score衡量候选内核在关闭释放定义基线与硬件极限之间的差距程度。为支持对智能体优化器的稳健评估,额外提供沙盒环境,含锁频GPU、清空L2缓存、隔离子进程执行及基于静态分析的常见奖励欺骗策略检测。SOL-ExecBench将内核基准测试从‘击败软件基线’转向‘逼近硬件极限’。
原文摘要 · Abstract (English)
As agentic AI systems become increasingly capable of generating and optimizing GPU kernels, progress is constrained by benchmarks that reward speedup over software baselines rather than proximity to hardware-efficient execution. We present SOL-ExecBench, a benchmark of 235 CUDA kernel optimization problems extracted from 124 production and emerging AI models spanning language, diffusion, vision, audio, video, and hybrid architectures, targeting NVIDIA Blackwell GPUs. The benchmark covers forward and backward workloads across BF16, FP8, and NVFP4, including kernels whose best performance is expected to rely on Blackwell-specific capabilities. Unlike prior benchmarks that evaluate kernels primarily relative to software implementations, SOL-ExecBench measures performance against analytically derived Speed-of-Light (SOL) bounds computed by SOLAR, our pipeline for deriving hardware-grounded SOL bounds, yielding a fixed target for hardware-efficient optimization. We report a SOL Score that quantifies how much of the gap between a release-defined scoring baseline and the hardware SOL bound a candidate kernel closes. To support robust evaluation of agentic optimizers, we additionally provide a sandboxed harness with GPU clock locking, L2 cache clearing, isolated subprocess execution, and static analysis based checks against common reward-hacking strategies. SOL-ExecBench reframes GPU kernel benchmarking from beating a mutable software baseline to closing the remaining gap to hardware Speed-of-Light.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。