2024年4月27日
TVM Pass优化 -- 算子融合(FuseOps)
摘要： 定义 算子融合 就是将多个计算单元合并到一个计算单元里完成计算，减少中间数据读写内存的操作，从而节省计算时间。 TVM中将算子融合分为四种： kElemWise：两个tensor之间按照元素逐个操作的算子，实际上所有的四则运算都是这种类型 kBroadcast：带有广播操作的算子 kInjectiv 阅读全文
posted @ 2024-04-27 21:46 牛犁heart 阅读(525) 评论(0) 推荐(0) 编辑