【BYOC】TVM切分子图
# 前言
本篇通过举例说明 TVM 切分子图的方法。
TVM 切分子图,是将整个图中的部分算子拆分出来,包装为一个函数。TVM 把它称为复合函数。切分子图有很多用处,比如算子融合、跨平台优化、做 layergroup 等。TVM 为子图切分提供了好用的工具,本篇文章将结 《【BYOC】TVM 添加自定义编译器 ccompiler》,来介绍一下如何使用 TVM 的工具切分子图,令自定义编译器 ccompiler 支持的子图在 ccompiler 执行,不支持的子图在 CPU 运行的。
参考链接:《【BYOC】TVM 添加自定义编译器 ccompiler》
作为初学者,错误在所难免,还望不吝赐教。
more...






