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