PassNet: Scaling Large Language Models for Graph Compiler Pass Generation
Not on the map. Why?
Abstract (arXiv)
Modern tensor compilers such as TorchInductor deliver substantial speedups on mainstream models, yet face a systematic performance ceiling on long-tail workloads -- our profiling shows that 43% of real-world subgraphs experience end-to-end slowdowns under default compilation. While LLMs offer a path toward automated optimization, existing efforts focus on standalone kernel generation. We argue that pass generation -- where LLMs author structured graph transformations that integrate directly into compiler pipelines -- is the more appropriate abstraction. We propose PassNet, the first large-scale ecosystem for LLM-based compiler pass generation, comprising: (1) PassNet-Dataset, over 18K unique computational graphs from 100K real-world models; and (2) PassBench, 200 curated long-tail fusible tasks (comprising 2,060 subgraphs in total) evaluated under the Error-aware Speedup Score (ES_t) -- a metric unifying correctness, stability, and performance -- with layered integrity defenses against systematic LLM exploitation. Experiments reveal that PassBench is both highly discriminative and genuinely unsaturated: the best frontier model trails TorchInductor by 37% in aggregate, yet on individual subgraphs LLMs achieve up to 3x speedup over the same compiler -- indicating that the bottleneck is consistency, not capability. Fine-tuning a small model on merely ~4K PassNet trajectories yields a 2.67x improvement approaching frontier-model performance, demonstrating substantial headroom and validating PassNet as live training infrastructure for advancing LLM-driven compiler optimization. All data, benchmarks, and tooling are publicly available.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 22 of its 41 references that are in the map. First posted on arXiv 28 May 2026.
Builds on
22 of 41 references.
References · 22
- 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,373
- 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,737
- 2024OpenHands: An Open Platform for AI Software Developers as Generalist Agentscited by 1,002
- 2024A Survey on Large Language Models for Code Generationcited by 658
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
- 2025R2E-Gym: Procedural Environments and Hybrid Verifiers for Scaling Open-Weights SWE Agentscited by 158
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 78
- 2023A Survey of Large Language Models for Code: Evolution, Benchmarking, and Future Trendscited by 66
- 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 61
- 2026CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generationcited by 47
- 2020The Deep Learning Compiler: A Comprehensive Surveycited by 46
10 more
- 2021CompilerGym: Robust, Performant Compiler Optimization Environments for AI Researchcited by 41
- 2022Tensor Program Optimization with Probabilistic Programscited by 30
- 2022Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programscited by 28
- 2025STARK: Strategic Team of Agents for Refining Kernelscited by 25
- 2023TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphscited by 16
- 2022TLP: A Deep Learning-based Cost Model for Tensor Program Tuningcited by 14
- 2024MORepair: Teaching LLMs to Repair Code via Multi-Objective Fine-tuningcited by 13
- 2025Seeing is Fixing: Cross-Modal Reasoning with Multimodal LLMs for Visual Software Issue Fixingcited by 12
- 2023ComPile: A Large IR Dataset from Production Sourcescited by 10
- 2025Compiler-R1: Towards Agentic Compiler Auto-tuning with Reinforcement Learningcited by 7
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Step-TP: A Grounded, Step-Level Dataset with Chain-of-Thought Reasoning for LLM-Guided Tensor Program Optimization
- 2026Toward Compiler World Models: Learning Latent Dynamics for Efficient Tensor Program Search
- 2026TensorBench: Benchmarking Coding Agents on a Compiler-Based Tensor Framework
- 2024GraphArena: Evaluating and Exploring Large Language Models on Graph Computation
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?
- 2025Can Large Language Models Understand Intermediate Representations in Compilers?
- 2026TensorCast: The Missing Tensor Management Layer in Large Language Model Infrastructure
- 2026Prism: Symbolic Superoptimization of Tensor Programs