Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPU
Not on the map. Why?
Abstract (arXiv)
Porting deep learning algorithms to new hardware accelerators requires developers to repeatedly apply the same low-level optimizations -- quantization, memory access coalescing, tile size tuning, and architecture-specific workarounds -- to every Triton kernel in their code-base. This manual, repetitive effort is a major bottleneck: each kernel demands the same cycle of trial-and-error profiling against hardware constraints that vary across devices, yet the underlying optimization patterns remain largely consistent. We present Xe-Forge, a multi-stage LLM-powered pipeline that automates this process for Intel GPU. Given a functionally correct Triton kernel, the system applies up to nine optimization stages -- from algorithmic restructuring and operator fusion through block pointer modernization, GPU-specific tuning, and open-ended discovery -- each driven by a Chain-of-Verification-and-Refinement (CoVeR) agent that generates candidates, validates them on real hardware, and iterates on failures. A curated knowledge base encodes Intel GPU constraints (power-of-two warp counts, GRF modes, SLM sizing) that are absent from LLM training data, keeping the model within architecturally valid bounds. We evaluate Xe-Forge on 97 Level-2 KernelBench kernels and Flash Attention on the Intel Arc Pro B70, achieving a 1.17x geometric mean speedup over PyTorch eager with 67% of kernels improving, nine kernels exceeding 5x (up to 82x), and 2--13.3x speedups on Flash Attention across all tested configurations without regression -- demonstrating that structured domain knowledge with hardware-in-the-loop verification can systematically eliminate the repetitive porting effort that currently gates algorithm deployment on new accelerators.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 19 of its 42 references that are in the map. First posted on arXiv 16 Apr 2026.
Builds on
19 of 42 references.
References · 19
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 78
- 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 51
- 2026CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generationcited by 47
- 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMscited by 40
- 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarkscited by 32
- 2025STARK: Strategic Team of Agents for Refining Kernelscited by 25
- 2026AVO: Agentic Variation Operators for Autonomous Evolutionary Searchcited by 24
7 more
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limitscited by 18
- 2026AutoKernel: Autonomous GPU Kernel Optimization via Iterative Agent-Driven Searchcited by 18
- 2025TritonForge: Profiling-Guided Framework for Automated Triton Kernel Optimizationcited by 16
- 2026KernelFoundry: Hardware-aware evolutionary GPU kernel optimizationcited by 13
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimizationcited by 11
- 2026VibeTensor: System Software for Deep Learning, Fully Generated by AI Agentscited by 6
- 2025ML-Triton, A Multi-Level Compilation and Language Extension to Triton GPU Programmingcited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025GPU Kernel Scientist: An LLM-Driven Framework for Iterative Kernel Optimization
- 2026KernelBenchX: A Comprehensive Benchmark for Evaluating LLM-Generated GPU Kernels
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization
- 2026Are LLM-Generated GPU Kernels Production-Ready? A Trace-Driven Benchmark and Optimization Agent
- 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimization
- 2026AMDKernelVault: Large-Scale Datasets and Agentic Training for AMD GPU Kernel Optimization
- 2026FastKernels: Benchmarking GPU Kernel Generation in Production
- 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimization