AI as a Compiler: Compiling Triton kernels without the Triton compiler
Not on the map. Why?
Abstract (arXiv)
Compiler backends are expensive to build and maintain as programming models, workloads, and accelerators evolve. We investigate whether large language models can replace the conventional optimizing and lowering pipeline, a process that we call AI lowering. We study AI lowering from Triton to NVIDIA PTX: an LLM agent translates Triton kernels directly into PTX. We build an environment that evaluates candidate PTX, and an agentic harness in which an LLM translates Triton kernels into PTX. Across twelve common kernels on Ada, Hopper, and Blackwell GPUs and ten kernels from recent ML papers, AI lowering achieves 0.83x-3.34x the performance of autotuned Triton. The largest gains come from transformations that Triton's lowering pipeline does not perform, such as decoding packed binary weights directly into Tensor Core operands (3.34x on BitDelta), assigning each thread a complete softmax row in tensor memory (1.37x on FlashAttention), and reusing overlapping convolution windows (up to 2.23x). These results rely on a robust evaluation harness with comprehensive verification support. We build on Volta, an existing PTX verifier, and substantially extend it to support modern GPU architectures by introducing support for Blackwell's tcgen05 Tensor Core interface. This requires modeling three architectural features: managed tensor memory, descriptor-based operand layouts, and asynchronous execution coordinated through commits, waits, memory barriers, and proxy fences. We discuss the challenges involved in formalizing them, as well as the current limitations. Our results suggest an emerging future in which AI compilers replace custom-written intermediate representations and checkers, reducing the time and engineering effort required to bring up software for new general-purpose and custom chips.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 24 of its 39 references that are in the map. First posted on arXiv 29 Sep 2026.
Builds on
24 of 39 references.
References · 24
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 991
- 2023Symbolic Discovery of Optimization Algorithmscited by 434
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
- 2024SageAttention: Accurate 8-Bit Attention for Plug-and-play Inference Accelerationcited by 160
- 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
- 2025Dion: Distributed Orthonormalized Updatescited by 68
- 2020MLIR: A Compiler Infrastructure for the End of Moore's Lawcited by 65
- 2025Forgetting Transformer: Softmax Attention with a Forget Gatecited by 53
- 2025TileLang: A Composable Tiled Programming Model for AI Systemscited by 43
12 more
- 2021CompilerGym: Robust, Performant Compiler Optimization Environments for AI Researchcited by 41
- 2024Mirage: A Multi-Level Superoptimizer for Tensor Programscited by 41
- 2024BitDelta: Your Fine-Tune May Only Be Worth One Bitcited by 39
- 2026Dr. Kernel: Reinforcement Learning Done Right for Triton Kernel Generationscited by 22
- 2025ParallelKittens: Systematic and Practical Simplification of Multi-GPU AI Kernelscited by 12
- 2026FlashSinkhorn: IO-Aware Entropic Optimal Transport on GPUcited by 11
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?cited by 8
- 2025Equivalence Checking of ML GPU Kernelscited by 7
- 2025SuperCoder: Assembly Program Superoptimization with Large Language Modelscited by 6
- 2025Tawa: Automatic Warp Specialization for Modern GPUs with Asynchronous Referencescited by 6
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolutioncited by 5
- 2022Learning to Combine Instructions in LLVM Compilercited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarks
- 2025The Anatomy of a Triton Attention Kernel
- 2026KernelBenchX: A Comprehensive Benchmark for Evaluating LLM-Generated GPU Kernels
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?
- 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMs
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization
- 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidance