Skip to content

Commit

Permalink
Update rpc_module.cc (#10881)
Browse files Browse the repository at this point in the history
Sometimes our locally built module is wrapped as a submodule of the top module. In such cases, we want to fetch the symbol from the imported modules, other than only searching in the top level.
  • Loading branch information
Maximilianxu authored Apr 5, 2022
1 parent f096ad8 commit 05caa7a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/runtime/rpc/rpc_module.cc
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ class RPCModuleNode final : public ModuleNode {
return WrapRemoteFunc(sess_->GetFunction(name));
} else {
InitRemoteFunc(&remote_mod_get_function_, "tvm.rpc.server.ModuleGetFunction");
return remote_mod_get_function_(GetRef<Module>(this), name, false);
return remote_mod_get_function_(GetRef<Module>(this), name, true);
}
}

Expand Down

0 comments on commit 05caa7a

Please sign in to comment.