2022年7月21日
TVM：使用调度模板和AutoTVM优化算子
摘要： 本节学习如何使用TVM 张量表达式(TE)语言来编写调度模板，这些模板可以被autoTVM搜索到，以找到最佳调度。这个过程称为auto-Tuning,它有助于优化张量计算的自动化过程。 本节建立在如何使用TE编写矩阵乘法的基础上 auto-tuning的步骤如下： 第一搜索空间 第二运行一个搜索算法 阅读全文
posted @ 2022-07-21 23:54 牛犁heart 阅读(889) 评论(0) 推荐(0)