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

- 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.
Kernel generation results reported on translation-style benchmarks overstate what models do on real framework code, and this gives a harder target measured on end-to-end performance.
“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
articleBacktrader-Bench: Benchmarking LLM Agents on Algorithmic Trading with Self-Generated MCQsRuoxi Zhao, Maziar Raissi
Checking sign-in…
Loading comments…