mlsanityData to 14 Oct 2026

A Two-Stage GPU Kernel Tuner Combining Semantic Refactoring and Search-Based Optimization

arXiv 2601.12698 ↗First posted 19 Jan 2026cs.CLcited by 3

Not on the map. Why?

Abstract (arXiv)

GPU code optimization is a key performance bottleneck for HPC workloads as well as large-model training and inference. Although compiler optimizations and hand-written kernels can partially alleviate this issue, achieving near-hardware-limit performance still relies heavily on manual code refactoring and parameter tuning. Recent progress in LLM-agent-based kernel generation and optimization has been reported, yet many approaches primarily focus on direct code rewriting, where parameter choices are often implicit and hard to control, or require human intervention, leading to unstable performance gains. This paper introduces a template-based rewriting layer on top of an agent-driven iterative loop: kernels are semantically refactored into explicitly parameterizable templates, and template parameters are then optimized via search-based autotuning, yielding more stable and higher-quality speedups. Experiments on a set of real-world kernels demonstrate speedups exceeding 3x in the best case. We extract representative CUDA kernels from SGLang as evaluation targets; the proposed agentic tuner iteratively performs templating, testing, analysis, and planning, and leverages profiling feedback to execute constrained parameter search under hardware resource limits. Compared to agent-only direct rewriting, the template-plus-search design significantly reduces the randomness of iterative optimization, making the process more interpretable and enabling a more systematic approach toward high-performance configurations. The proposed method can be further extended to OpenCL, HIP, and other backends to deliver automated performance optimization for real production workloads.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 24 of its 44 references that are in the map. First posted on arXiv 19 Jan 2026.

Builds on

24 of 44 references.

References · 24

  1. 2023AutoGen: Enabling Next-Gen LLM Applications via Multi-Agent Conversationcited by 1,988
  2. 2023MetaGPT: Meta Programming for A Multi-Agent Collaborative Frameworkcited by 1,808
  3. 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,737
  4. 2022Competition-Level Code Generation with AlphaCodecited by 1,462
  5. 2023CAMEL: Communicative Agents for "Mind" Exploration of Large Language Model Societycited by 1,395
  6. 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
  7. 2023ChatDev: Communicative Agents for Software Developmentcited by 919
  8. 2016TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systemscited by 664
  9. 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
  10. 2018Beyond Data and Model Parallelism for Deep Neural Networkscited by 179
  11. 2018Glow: Graph Lowering Compiler Techniques for Neural Networkscited by 75
  12. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
12 more
  1. 2024Trace is the Next AutoDiff: Generative Optimization with Rich Feedback, Execution Traces, and LLMscited by 69
  2. 2020MLIR: A Compiler Infrastructure for the End of Moore's Lawcited by 65
  3. 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 51
  4. 2024Mirage: A Multi-Level Superoptimizer for Tensor Programscited by 41
  5. 2024ThunderKittens: Simple, Fast, and Adorable AI Kernelscited by 36
  6. 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarkscited by 32
  7. 2024LLM-Vectorizer: LLM-based Verified Loop Vectorizercited by 25
  8. 2024Improving Parallel Program Performance with LLM Optimizers via Agent-System Interfacescited by 11
  9. 2017DLVM: A modern compiler infrastructure for deep learning systemscited by 4
  10. 2025VecTrans: Enhancing Compiler Auto-Vectorization through LLM-Assisted Code Transformationscited by 4
  11. 2025Task-Based Programming for Adaptive Mesh Refinement in Compressible Flow Simulationscited by 2
  12. 2025Mapple: A Domain-Specific Language for Mapping Distributed Programscited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025STARK: Strategic Team of Agents for Refining Kernels
  2. 2026Agentic Kernel Optimization: Generating State-of-the-Art GPU Kernels Without Hand-Written CUDA
  3. 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidance
  4. 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimization
  5. 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimization
  6. 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTe
  7. 2025GPU Kernel Scientist: An LLM-Driven Framework for Iterative Kernel Optimization
  8. 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization