mlsanityData to 13 Oct 2026

Learning When to Optimize: Verified Optimization Skills from Expert GPU-Kernel Lineages

arXiv 2605.28213 ↗First posted 27 May 2026cs.AIcited by 0

On the map: LLM agents and tool use → Software engineering coding agents

Abstract (arXiv)

LLM-based agents are increasingly used to generate GPU kernels, but they often struggle to determine when an optimization is sound because its required code state and dependencies are implicit in expert implementations. We introduce KLineage, which learns this missing "when" knowledge from expert kernels: instead of relying on forward rollouts, KLineage walks expert implementations backward through validation-gated simplifications and reverses each accepted step into a reusable optimization skill. Each skill records not only the optimization intent, but also when to apply the optimization technique, including where it applies in code, what conditions made it valid, what effect it has, and what failures its assumptions avoid. A downstream LLM materializes these skills on new code surfaces under the same compile/correctness/profile gate. This guidance on when to apply each optimization can help downstream models to generate higher-performance kernels. On five expert workloads across two NVIDIA architectures, these lineage-derived skills serve as an effective optimization curriculum, exceeding recent memory-based LLM-kernel baselines in both final kernel quality and optimization efficiency under the same fixed budget. We also demonstrate that the KLineage framework extends beyond NVIDIA GPUs to Ascend NPUs. Our code is publicly available at https://github.com/ict-agent/klineage.

Key facts. No citing papers in the map yet (data through 13 Oct 2026). Placed in Software engineering coding agents, one of 16 directions in LLM agents and tool use. Builds on 19 of its 28 references that are in the map. First posted on arXiv 27 May 2026.

Builds on

19 of 28 references.

Closest ancestors · 6

Same direction, newest first

  1. 2026FastKernels: Benchmarking GPU Kernel Generation in Production22 May
  2. 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTe1 Apr
  3. 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidance30 Mar
  4. 2026KernelBlaster: Continual Cross-Task CUDA Optimization via Memory-Augmented In-Context Reinforcement Learning15 Feb
  5. 2025AccelOpt: A Self-Improving LLM Agentic System for AI Accelerator Kernel Optimization19 Nov
  6. 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization9 Sep

Other references · 13

  1. 2025FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Servingcited by 253
  2. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 72
  3. 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 61
  4. 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 37
  5. 2024ThunderKittens: Simple, Fast, and Adorable AI Kernelscited by 35
  6. 2026K-Search: LLM Kernel Generation via Co-Evolving Intrinsic World Modelcited by 21
  7. 2025QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generationcited by 18
  8. 2025TritonForge: Profiling-Guided Framework for Automated Triton Kernel Optimizationcited by 16
  9. 2026KernelFoundry: Hardware-aware evolutionary GPU kernel optimizationcited by 13
  10. 2025GPU Kernel Optimization Beyond Full Builds: An LLM Framework with Minimal Executable Programscited by 3
  11. 2026A Two-Stage GPU Kernel Tuner Combining Semantic Refactoring and Search-Based Optimizationcited by 3
  12. 2026CuBridge: An LLM-Based Framework for Understanding and Reconstructing High-Performance Attention Kernelscited by 2
1 more
  1. 2026KEET: Explaining Performance of GPU Kernels Using LLM Agentscited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025GPU Kernel Scientist: An LLM-Driven Framework for Iterative Kernel Optimization
  2. 2026D2K-Bench: Can LLM Agents Turn Expert Designs into Efficient GPU Kernels?
  3. 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization
  4. 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimization
  5. 2025STARK: Strategic Team of Agents for Refining Kernels
  6. 2026Kernel Foundry: A Diagnosis-driven Evolutionary Kernel Optimizer with Multi-Experts
  7. 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimization
  8. 2026Prior Knowledge or Search? A Study of LLM Agents in Hardware-Aware Code Optimization