修改Graph
- 使用改图接口修改Graph
如果用户想要直接优化图的结构,比如将某些特定子图替换成一个大算子,以减少计算步骤、外存访问、调度时间等,或者在某些算子中间添加一个算子,此时可以通过本节内容将图直接修改成期望的结构。 - 使用自定义Pass修改Graph
改图功能主要有两种调用方式,用户可以在构建完Graph后,直接调用改图接口修改Graph,也可以将改图函数封装为自定义Pass,并通过REGISTER_CUSTOM_PASS注册宏进行改图Pass注册,通过把改图函数编译成动态库插件方式,注册的Pass可以在指定的阶段被调用。