GraphNet: A Large-Scale Computational Graph Dataset for Tensor Compiler Research
Not on the map. Why?
Abstract (arXiv)
We introduce GraphNet, a dataset of 2.7K real-world deep learning computational graphs with rich metadata, spanning six major task categories across multiple deep learning frameworks. To evaluate tensor compiler performance on these samples, we propose the benchmark metric Speedup Score S(t), which jointly considers runtime speedup and execution correctness under tunable tolerance levels, offering a reliable measure of general optimization capability. Furthermore, we extend S(t) to the Error-aware Speedup Score ES(t), which incorporates error information and helps compiler developers identify key performance bottlenecks. In this report, we benchmark the default tensor compilers, CINN for PaddlePaddle and TorchInductor for PyTorch, on computer vision (CV) and natural language processing (NLP) samples to demonstrate the practicality of GraphNet. The full construction pipeline with graph extraction and compiler evaluation tools is available at https://github.com/PaddlePaddle/GraphNet .
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 10 of its 22 references that are in the map. First posted on arXiv 28 Oct 2025.
Builds on
10 of 22 references.
References · 10
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2014cuDNN: Efficient Primitives for Deep Learningcited by 245
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
- 2018Glow: Graph Lowering Compiler Techniques for Neural Networkscited by 75
- 2020The Deep Learning Compiler: A Comprehensive Surveycited by 46
- 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
- 2025Leveraging Neural Graph Compilers in Machine Learning Research for Edge-Cloud Systemscited by 2
Built on by
Not cited yet.