TVM学习(二):算符融合 judy / 周五, 18 十二月 2020 - 15:21 算符融合将多个计算单元揉进一个计算核中进行,减少了中间数据的搬移,节省了计算时间。TVM中将计算算符分成四种: 1 injective。一一映射函数,比如加法,点乘等。 2 reduction。输入到输出具有降维性质的,比如sum。 3 complex-out。这是计算比较复杂的,比如卷积运算等。 4 opaque。无法被融合的算符,比如sort。 阅读更多 关于 TVM学习(二):算符融合登录 发表评论