标签：调用 relay GraphRuntimeCodegenModule codegen build 方法

relay.build调用关系
在用TVM优化过程中，常见demo如下图：

 

 relay.build背后的调用机制梳理如下：

 python中的build函数实现如下，此方法实际调用了BuildModule类的build方法，

 

BuildModule获取的C++函数build方法，此处涉及混合编程：

 

 _build_module._BuildModule() 来自于这句注册：

tvm._ffi._init_api("relay.build_module", __name__)
self.mod["build"]调用RelayBuildModule的GetFunction方法，其中name为“build”：

 

 RelayBuildModule类的build方法如下，主要包含两部分功能，Optimize和codegen：

 

 Optimize函数有很多优化策略，

 

 在此处执行，实际执行的是SequentialNode的operator方法：

 

 Sequential::Sequential构造函数如下，其中name为声明带默认参数：

 

 Sequential的operator->()方法调用的是SequentialNode的operator()方法：

 

 codegen 代码逻辑如下：

 

 GetPackedFun函数调用的是提前注册的函数CreateGraphCodegenMod()，

返回GraphRuntimeCodegenModule类型的对象

 

 graph_codegen_ 指针指向的是GraphRuntimeCodegenModule类的对象，Init方法，

Codegen方法，GetIRModule都来自于注册的GraphRuntimeCodegenModule 实例，

 

 Codegen 调用GraphRuntimeCodegenModule的codegen_，是GraphRuntimeCodegen

的实例指针，调用的codegen代码如下

 

 

参考链接：
https://www.pianshen.com/article/24811852648/

标签：调用,relay,GraphRuntimeCodegenModule,codegen,build,方法
来源： https://www.cnblogs.com/wujianming-110117/p/15785415.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。