diff --git a/src/runtime/module.cc b/src/runtime/module.cc index cf7c5d921f2f..de372e5de053 100644 --- a/src/runtime/module.cc +++ b/src/runtime/module.cc @@ -198,6 +198,10 @@ TVM_REGISTER_GLOBAL("runtime.ModuleGetFormat").set_body_typed([](Module mod) { }); TVM_REGISTER_GLOBAL("runtime.ModuleLoadFromFile").set_body_typed(Module::LoadFromFile); +TVM_REGISTER_GLOBAL("runtime.ModuleGetFunction") + .set_body_typed([](Module mod, String name, bool query_imports) { + return mod->GetFunction(name, query_imports); + }); TVM_REGISTER_GLOBAL("runtime.ModuleSaveToFile") .set_body_typed([](Module mod, String name, tvm::String fmt) { mod->SaveToFile(name, fmt); });