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