2023年12月21日
TensorIR: An Abstraction for Automatic Tensorized Program Optimization
摘要： Abstract 在多种多样的设备上部署深度学习模型是一个重要的话题，专用硬件的蓬勃发展引入了一系列加速原语和多维张量计算方法。这些新的加速原语和不断出现的新的机器学习模型，带来了工程上的巨大挑战。本文提出了TensorIR，是为了优化这些有张量计算原语的张量化程序而设计的编译器抽象。TensorI 阅读全文
posted @ 2023-12-21 18:53 Jareth 阅读(360) 评论(0) 推荐(0)