Dec, 2022

AGO: 解除图优化限制,提升移动 AI 推理性能

TL;DR该研究提出了 AGO,一种具有任意结构的图形优化框架,通过消除约束来提高深度模型的推理性能,并且能够有效地提高性能,包括设计一个图划分方案以实现任意子图结构,提出了强化运算符融合的机制以创造新的优化机会,以及提出了新颖的分治调整机制进行快速高效的性能调优。大量实验结果表明,相比于现有深度编译器,该系统能够提高可达 3.3 倍的推理性能。