relay.build調(diào)用關(guān)系
在用TVM優(yōu)化過程中，常見demo如下圖：

relay.build背后的調(diào)用機(jī)制梳理如下：
python中的build函數(shù)實(shí)現(xiàn)如下，此方法實(shí)際調(diào)用了BuildModule類的build方法，

BuildModule獲取的C++函數(shù)build方法，此處涉及混合編程：

_build_module._BuildModule() 來自于這句注冊(cè)：
tvm._ffi._init_api(“relay.build_module”, name)
self.mod[“build”]調(diào)用RelayBuildModule的GetFunction方法，其中name為“build”：

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

Optimize函數(shù)有很多優(yōu)化策略，

在此處執(zhí)行，實(shí)際執(zhí)行的是SequentialNode的operator方法：

Sequential::Sequential構(gòu)造函數(shù)如下，其中name為聲明帶默認(rèn)參數(shù)：

Sequential的operator->()方法調(diào)用的是SequentialNode的operator()方法：

codegen 代碼邏輯如下：

GetPackedFun函數(shù)調(diào)用的是提前注冊(cè)的函數(shù)CreateGraphCodegenMod()，
返回GraphRuntimeCodegenModule類型的對(duì)象

graph_codegen_ 指針指向的是GraphRuntimeCodegenModule類的對(duì)象，Init方法，
Codegen方法，GetIRModule都來自于注冊(cè)的GraphRuntimeCodegenModule 實(shí)例，

Codegen 調(diào)用GraphRuntimeCodegenModule的codegen_，是GraphRuntimeCodegen
的實(shí)例指針，調(diào)用的codegen代碼如下

參考鏈接：
https://www.pianshen.com/article/24811852648/

總結(jié)

以上是生活随笔為你收集整理的relay.build调用关系的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。