|
|
@@ -32,6 +32,9 @@ |
|
|
|
#include "graph/common/ge_call_wrapper.h" |
|
|
|
#include "register/op_registry.h" |
|
|
|
#include "common/ge/tbe_plugin_manager.h" |
|
|
|
#ifndef ONLY_COMPILE_OPEN_SRC |
|
|
|
#include "toolchain/plog.h" |
|
|
|
#endif |
|
|
|
|
|
|
|
using domi::OpRegistry; |
|
|
|
using std::map; |
|
|
@@ -129,6 +132,11 @@ Status GEInitializeImpl(const std::map<string, string> &options) { |
|
|
|
|
|
|
|
// Initialize GE, prepare for execution, call GELib::Initialize |
|
|
|
Status GEInitialize(const std::map<string, string> &options) { |
|
|
|
#ifndef ONLY_COMPILE_OPEN_SRC |
|
|
|
if (DlogReportInitialize() != SUCCESS) { |
|
|
|
GELOGW("Dlog report device log initialize failed."); |
|
|
|
} |
|
|
|
#endif |
|
|
|
return GEInitializeImpl(options); |
|
|
|
} |
|
|
|
|
|
|
@@ -143,6 +151,11 @@ Status GEInitialize(const std::map<AscendString, AscendString> &options) { |
|
|
|
std::string val = option.second.GetString(); |
|
|
|
str_options[key] = val; |
|
|
|
} |
|
|
|
#ifndef ONLY_COMPILE_OPEN_SRC |
|
|
|
if (DlogReportInitialize() != SUCCESS) { |
|
|
|
GELOGW("Dlog report device log initialize failed."); |
|
|
|
} |
|
|
|
#endif |
|
|
|
return GEInitializeImpl(str_options); |
|
|
|
} |
|
|
|
|
|
|
@@ -187,6 +200,12 @@ Status GEFinalize() { |
|
|
|
// to avoid memory fragment, use malloc_trim to back free stack to system |
|
|
|
malloc_trim(0); |
|
|
|
|
|
|
|
#ifndef ONLY_COMPILE_OPEN_SRC |
|
|
|
if (DlogReportFinalize() != SUCCESS) { |
|
|
|
GELOGW("Dlog report device log finalize failed."); |
|
|
|
} |
|
|
|
#endif |
|
|
|
|
|
|
|
GELOGT(TRACE_STOP, "GEFinalize finished"); |
|
|
|
return ret; |
|
|
|
} |
|
|
|