RealisticTritonBench: A Benchmark for Triton-Kernel Generation in Real-World AI Frameworks
- Source
- arxiv.org
- Author
- Jinjun Huang, Zhongzhen Wen, Tongtong Xu, Meng Yan, Xin Xia, Zhongxin Liu
- Date

Context
RealisticTritonBench tests AI-generated GPU code inside the larger software that must use it. A GPU kernel is a program that performs computations on a graphics processor; testing it alone can miss whether it works properly within an application. The benchmark's authors construct tasks from real code changes in AI frameworks and evaluate generated implementations after integrating them into their original frameworks.
This changes what a successful result must demonstrate: the implementation has to meet an engineering requirement and pass end-to-end tests, rather than merely perform well in an isolated kernel test. The authors report that the models they evaluated still struggle with these tasks.
- benchmark — A standard public test set for comparing AI models — the shared scoreboards behind every "model X beats model Y" claim.
- LLM — A large language model — the neural network behind tools like Claude and ChatGPT, trained on huge amounts of text to predict what comes next.
- eval — A repeatable test for AI quality — a set of tasks plus scoring — used the way software teams use test suites, because model output is too variable to judge by eyeballing.
“Combining usability, portability, and near-handwritten CUDA performance, Triton is widely adopted for implementing GPU kernels.”
Jinjun Huang et al.
“they rely on manually written evaluation scripts for individual kernels, which may contain flaws that models can exploit to bypass correctness checks and obtain inflated scores”
Jinjun Huang et al.
“To address these limitations, we introduce RealisticTritonBench, the first benchmark to derive Triton kernel generation tasks from real-world pull requests in popular AI frameworks, enabling realistic, production-like evaluation.”
Jinjun Huang et al.
“We evaluate leading LLMs on RealisticTritonBench and find that they still struggle with real-world Triton kernel generation tasks.”
Jinjun Huang et al.
articleDataKernelBench: Can LLMs Optimize Database Queries on GPUs?Gokul Karthik Kumar, Yotam Perlitz, Corey Lammie, Andrea Giovannini, Katja Hose
articleCUDA-Harness: Harnessing Agentic CUDA Kernel Generation and Optimization from Natural LanguageQi Fan, An Zou, Yehan Ma
articleLong-Horizon-Terminal-Bench 2.0: Automating Diverse Task Construction and Measuring Sustained Agent ProgressZongxia Li, Zhongzhi Li, Yucheng Shi, Ruhan Wang, Junyao Yang, Chengsong Huang, Zhichao Liu, Xiyang Wu, Anhao Li, Yue Yu, Ergan Shang, Fuxiao Liu, Tong Zheng, Ninghao Liu, Lichao Sun, Haitao Mi, Leowei Liang
Checking sign-in…
Loading comments…