|
|
@ -298,19 +298,6 @@ void MsContext::GetGeOptions(std::map<std::string, std::string> *ge_options) con
|
|
|
|
(*ge_options)["ge.exec.profilingOptions"] = profiling_options_;
|
|
|
|
(*ge_options)["ge.exec.profilingOptions"] = profiling_options_;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// only not supported in ge
|
|
|
|
|
|
|
|
auto tbe_plugin_path = common::GetEnv("ME_TBE_PLUGIN_PATH");
|
|
|
|
|
|
|
|
if (!tbe_plugin_path.empty()) {
|
|
|
|
|
|
|
|
char real_path[PATH_MAX] = {0};
|
|
|
|
|
|
|
|
if (nullptr == realpath(tbe_plugin_path.c_str(), real_path)) {
|
|
|
|
|
|
|
|
MS_LOG(ERROR) << "Ms tbe plugin Path error, " << tbe_plugin_path;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
tbe_plugin_path = real_path;
|
|
|
|
|
|
|
|
(*ge_options)["ge.TBE_plugin_path"] = tbe_plugin_path;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
MS_LOG(ERROR) << "Set TBE plugin path failed!";
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
(*ge_options)["rank_table_file"] = "";
|
|
|
|
(*ge_options)["rank_table_file"] = "";
|
|
|
|
auto env_ddk_version = common::GetEnv("DDK_VERSION");
|
|
|
|
auto env_ddk_version = common::GetEnv("DDK_VERSION");
|
|
|
|
if (!env_ddk_version.empty()) {
|
|
|
|
if (!env_ddk_version.empty()) {
|
|
|
@ -354,18 +341,6 @@ void MsContext::GetGeOptions(std::map<std::string, std::string> *ge_options) con
|
|
|
|
MS_LOG(INFO) << "Use AICPU, make sure aicpu lib is set in OPTION_EXEC_EXTERN_PLUGIN_PATH.";
|
|
|
|
MS_LOG(INFO) << "Use AICPU, make sure aicpu lib is set in OPTION_EXEC_EXTERN_PLUGIN_PATH.";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// all libs are set in same env variable "OPTION_EXEC_EXTERN_PLUGIN_PATH", such as FE, HCCL, AICPU, etc
|
|
|
|
|
|
|
|
auto load_path = common::GetEnv("OPTION_EXEC_EXTERN_PLUGIN_PATH");
|
|
|
|
|
|
|
|
if (!load_path.empty()) {
|
|
|
|
|
|
|
|
char real_path[PATH_MAX] = {0};
|
|
|
|
|
|
|
|
if (realpath(load_path.c_str(), real_path)) {
|
|
|
|
|
|
|
|
load_path = real_path;
|
|
|
|
|
|
|
|
(*ge_options)["ge.soLoadPath"] = load_path;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
MS_LOG(ERROR) << "Set lib load path failed!";
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
auto proto_lib_path = common::GetEnv("OPTION_PROTO_LIB_PATH");
|
|
|
|
auto proto_lib_path = common::GetEnv("OPTION_PROTO_LIB_PATH");
|
|
|
|
if (!proto_lib_path.empty()) {
|
|
|
|
if (!proto_lib_path.empty()) {
|
|
|
|
char real_path[PATH_MAX] = {0};
|
|
|
|
char real_path[PATH_MAX] = {0};
|
|
|
@ -374,7 +349,7 @@ void MsContext::GetGeOptions(std::map<std::string, std::string> *ge_options) con
|
|
|
|
(*ge_options)["ge.opsProtoLibPath"] = proto_lib_path;
|
|
|
|
(*ge_options)["ge.opsProtoLibPath"] = proto_lib_path;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
MS_LOG(ERROR) << "Set proto lib path failed!";
|
|
|
|
MS_LOG(WARNING) << "Set proto lib path failed!";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Enable auto mixed precision according to the context options
|
|
|
|
// Enable auto mixed precision according to the context options
|
|
|
|