摘要: TVM Pass IR如何使用 随着Relay / tir中优化遍数的增加,执行并手动维护其依赖关系变得很棘手。引入了一个基础结构来管理优化过程,并应用于TVM堆栈中IR的不同层。 Relay / tir程序的优化可以以各种粒度应用,即分别使用tvm.relay.transform.Function 阅读全文
posted @ 2021-03-26 06:08 吴建明wujianming 阅读(567) 评论(0) 推荐(0) 编辑
摘要: 自定义pass编写 TVM是一个框架,抽象了机器学习加速器的异质性。有时,用户可能需要自定义一些分析和IR转​​换,使TVM适应自己的专用硬件。本文可帮助用户在TVM中编写自定义pass。 先决条件 在阅读本文之前,假设读者已经熟悉以下主题: 在TVM中编写算法并进行调度。否则,请参见示例教程,例如 阅读全文
posted @ 2021-03-26 05:47 吴建明wujianming 阅读(293) 评论(0) 推荐(0) 编辑