|
|
@@ -39,8 +39,6 @@ |
|
|
|
#include "graph/manager/graph_var_manager.h" |
|
|
|
#include "graph/load/new_model_manager/davinci_model.h" |
|
|
|
#include "opskernel_manager/ops_kernel_builder_manager.h" |
|
|
|
#include "graph/opsproto_manager.h" |
|
|
|
#include "ge_local_engine/engine/host_cpu_engine.h" |
|
|
|
|
|
|
|
using std::string; |
|
|
|
using std::vector; |
|
|
@@ -210,33 +208,6 @@ bool IsDynmaicDimsSizeMatchModel(const vector<uint64_t> cur_dynamic_dims, |
|
|
|
namespace ge { |
|
|
|
bool GeExecutor::isInit_ = false; |
|
|
|
|
|
|
|
static void InitOpsProtoManger() { |
|
|
|
string opsproto_path; |
|
|
|
const char *path_env = std::getenv("ASCEND_OPP_PATH"); |
|
|
|
if (path_env != nullptr) { |
|
|
|
string path = path_env; |
|
|
|
string file_path = RealPath(path.c_str()); |
|
|
|
if (file_path.empty()) { |
|
|
|
GELOGE(FAILED, "File path %s is invalid.", path.c_str()); |
|
|
|
return; |
|
|
|
} |
|
|
|
opsproto_path = (path + "/op_proto/custom/" + ":") + (path + "/op_proto/built-in/"); |
|
|
|
GELOGI("Get opsproto so path from env : %s", path.c_str()); |
|
|
|
} else { |
|
|
|
string path_base = PluginManager::GetPath(); |
|
|
|
GELOGI("path_base is %s", path_base.c_str()); |
|
|
|
path_base = path_base.substr(0, path_base.rfind('/')); |
|
|
|
path_base = path_base.substr(0, path_base.rfind('/') + 1); |
|
|
|
opsproto_path = (path_base + "ops/op_proto/custom/" + ":") + (path_base + "ops/op_proto/built-in/"); |
|
|
|
} |
|
|
|
|
|
|
|
GELOGI("Get opsproto path is %s", opsproto_path.c_str()); |
|
|
|
OpsProtoManager *manager = OpsProtoManager::Instance(); |
|
|
|
map<string, string> option_tmp; |
|
|
|
option_tmp.emplace(std::pair<string, string>(string("ge.opsProtoLibPath"), opsproto_path)); |
|
|
|
(void)manager->Initialize(option_tmp); |
|
|
|
} |
|
|
|
|
|
|
|
GeExecutor::GeExecutor() {} |
|
|
|
|
|
|
|
Status GeExecutor::Initialize() { |
|
|
@@ -246,16 +217,6 @@ Status GeExecutor::Initialize() { |
|
|
|
return ge::SUCCESS; |
|
|
|
} |
|
|
|
|
|
|
|
OpTilingManager::GetInstance().LoadSo(); |
|
|
|
|
|
|
|
Status initHostCpuEngineStatus = HostCpuEngine::GetInstance().Initialize(); |
|
|
|
if (initHostCpuEngineStatus != SUCCESS) { |
|
|
|
GELOGE(initHostCpuEngineStatus, "Failed to initialize HostCpuEngine"); |
|
|
|
return initHostCpuEngineStatus; |
|
|
|
} |
|
|
|
|
|
|
|
InitOpsProtoManger(); |
|
|
|
|
|
|
|
std::vector<rtMemType_t> mem_type(1, RT_MEMORY_HBM); |
|
|
|
mem_type.push_back(RT_MEMORY_P2P_DDR); |
|
|
|
auto ret = MemManager::Instance().Initialize(mem_type); |
|
|
|