Tensorized

Abstract 在多种多样的设备上部署深度学习模型是一个重要的话题，专用硬件的蓬勃发展引入了一系列加速原语和多维张量计算方法。这些新的加速原语和不断出现的新的机器学习模型，带来了工程上的巨大挑战。本文提出了TensorIR，是为了优化这些有张量计算原语的张量化程序而设计的编译器抽象。TensorI ......