nncase: An End-to-End Compiler for Efficient LLM Deployment on Heterogeneous Storage Architectures
Not on the map. Why?
Abstract (arXiv)
The efficient deployment of large language models (LLMs) is hindered by memory architecture heterogeneity, where traditional compilers suffer from fragmented workflows and high adaptation costs. We present nncase, an open-source, end-to-end compilation framework designed to unify optimization across diverse targets. Central to nncase is an e-graph-based term rewriting engine that mitigates the phase ordering problem, enabling global exploration of computation and data movement strategies. The framework integrates three key modules: Auto Vectorize for adapting to heterogeneous computing units, Auto Distribution for searching parallel strategies with cost-aware communication optimization, and Auto Schedule for maximizing on-chip cache locality. Furthermore, a buffer-aware Codegen phase ensures efficient kernel instantiation. Evaluations show that nncase outperforms mainstream frameworks like MLC LLM and Intel IPEX on Qwen3 series models and achieves performance comparable to the hand-optimized llama.cpp on CPUs, demonstrating the viability of automated compilation for high-performance LLM deployment. The source code is available at https://github.com/kendryte/nncase.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 58 references that are in the map. First posted on arXiv 25 Dec 2025.
Builds on
11 of 58 references.
References · 11
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
- 2018Learning to Optimize Tensor Programscited by 140
- 2021GSPMD: General and Scalable Parallelization for ML Computation Graphscited by 111
- 2022Tensor Program Optimization with Probabilistic Programscited by 30
- 2021Equality Saturation for Tensor Graph Superoptimizationcited by 27
- 2021Pure Tensor Program Rewriting via Access Patterns (Representation Pearl)cited by 7
- 2021End-to-end Adaptive Distributed Training on PaddlePaddlecited by 5
- 2023Colossal-Auto: Unified Automation of Parallelization and Activation Checkpoint for Large-scale Modelscited by 5
- 2021Tuna: A Static Analysis Approach to Optimizing Deep Neural Networkscited by 3
- 2023AutoDDL: Automatic Distributed Deep Learning with Near-Optimal Bandwidth Costcited by 1
- 2024Optimizing Tensor Computation Graphs with Equality Saturation and Monte Carlo Tree Searchcited by 1
Built on by
Not cited yet.