GPU Performance Portability needs Autotuning
Not on the map. Why?
Abstract (arXiv)
As LLMs grow in complexity, achieving state-of-the-art performance requires tight co-design across algorithms, software, and hardware. Today's reliance on a single dominant platform limits portability, creates vendor lock-in, and raises barriers for new AI hardware. In this work, we make the case for combining just-in-time (JIT) compilation with comprehensive kernel parameter autotuning to enable portable LLM inference with state-of-the-art performance without code changes. Focusing on performance-critical LLM kernels, we demonstrate that this approach explores up to 15x more kernel parameter configurations, produces significantly more diverse code across multiple dimensions, and even outperforms vendor-optimized implementations by up to 230%, all while reducing kernel code size by 70x and eliminating manual code optimizations. Our results highlight autotuning as a promising path to unlocking model portability across GPU vendors.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 10 of its 25 references that are in the map. First posted on arXiv 30 Apr 2025.
Builds on
10 of 25 references.
References · 10
- 2017Attention Is All You Needcited by 60,361
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 519
- 2025FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Servingcited by 258
- 2020The Hardware Lotterycited by 160
- 2020Agile Autotuning of a Transprecision Tensor Accelerator Overlay for TVM Compiler Stackcited by 4
Built on by
Citing papers per year: 2025: 1.
1 paper.
- 2025The Anatomy of a Triton Attention Kernelcited by 3
- 2025The Anatomy of a Triton Attention Kernel7 Oct 2025