For a simple trick to bring what function is being called by the runtime, add following line to GraphRuntime::Run
void GraphRuntime::Run() {
// setup the array and requirements.
for (size_t i = 0; i < op_execs_.size(); ++i) {
+ fprintf(stderr, "fused_layer %d\n", (int)i);
if (op_execs_[i]) op_execs_[i]();
}
}
In addition, you can get a list of functions in GraphRuntime::SetupOpExecs by applying
std::tie(op_execs_[nid], op_args) =
CreateTVMOp(inode.param, args, inode.inputs.size());
+ std::cerr << nid << " " << inode.param.func_name << std::endl;
for (size_t i = 0; i < inode.inputs.size(); i++) {