AdaTune: Adaptive Tensor Program Compilation Made Efficient

Authors: Menghao Li, Minjia Zhang, Chi Wang, Mingqin Li

Abstract

Deep learning models are computationally intense, and implementations often have to be highly optimized by experts or hardware vendors to be usable in practice. The DL compiler, together with Learning to Compile have proven to be a powerful technique for optimizing tensor programs. However, a limitation of this approach is that it still suffers from unbearably long overall optimization time.

PDFOpen original