Browse Source

Load plugin for IR Initialize.

tags/v1.1.0
unknown 3 years ago
parent
commit
bac4dd5102
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      ge/ir_build/ge_ir_build.cc

+ 4
- 0
ge/ir_build/ge_ir_build.cc View File

@@ -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;
} }


Loading…
Cancel
Save