Dear TVM/Relay experts,
I am trying to understand the internals of relay.build_module.build() function, and it looks like a complicated puzzle (sorry for my lack of experience) to me.
By my debugging I was able to solve some of this puzzle, and I know that “relay.build_module.build()” calls ScheduleGetter(target).Create(source_func) function inside the compile_enginee.cc. However, I am lost little bit lost in my debugging. My goal is to add my own custom (probably optimized) operator or template implementations in relay/tvm build. For that, I’d like to understand the whole flow in detail.
- Could someone explain me what are the sequence of function calls from relay.build_module.build() all the way to code generation?
- Could someone explain me how to dump IR of certain operators?
Thanks in advance, John.