diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc index ccf3b24e..0043dc8e 100644 --- a/ge/ir_build/ge_ir_build.cc +++ b/ge/ir_build/ge_ir_build.cc @@ -17,6 +17,7 @@ #include #include "common/auth/file_saver.h" +#include "common/ge/tbe_plugin_manager.h" #include "external/register/register_types.h" #include "framework/common/debug/ge_log.h" #include "framework/common/ge_inner_error_codes.h" @@ -162,6 +163,8 @@ graphStatus aclgrphBuildInitialize(std::map global_opt GELOGE(ret, "GE initialize failed!"); return GRAPH_FAILED; } + // for functional subgraph assign _parent_index. + TBEPluginManager::Instance().InitPreparation(global_options); } GELOGW("gelib has been initialized!"); return GRAPH_SUCCESS; @@ -169,6 +172,7 @@ graphStatus aclgrphBuildInitialize(std::map global_opt void aclgrphBuildFinalize() { if (ge::GELib::GetInstance() != nullptr && ge::GELib::GetInstance()->InitFlag()) { + (void)TBEPluginManager::Instance().Finalize(); (void)ge::GELib::GetInstance()->Finalize(); return; }