From bac4dd5102eec837ea82d1dc959bd5fb7b1fbf0e Mon Sep 17 00:00:00 2001 From: unknown Date: Wed, 18 Nov 2020 08:15:57 +0800 Subject: [PATCH] Load plugin for IR Initialize. --- ge/ir_build/ge_ir_build.cc | 4 ++++ 1 file changed, 4 insertions(+) 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; }