Browse Source

!759 device os log missing

From: @taoxiangdong
Reviewed-by: @wqtshg,@youui,@ji_chen
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
cfbdb031c8
2 changed files with 20 additions and 0 deletions
  1. +19
    -0
      ge/client/ge_api.cc
  2. +1
    -0
      tests/ut/ge/CMakeLists.txt

+ 19
- 0
ge/client/ge_api.cc View File

@@ -32,6 +32,9 @@
#include "graph/common/ge_call_wrapper.h" #include "graph/common/ge_call_wrapper.h"
#include "register/op_registry.h" #include "register/op_registry.h"
#include "common/ge/tbe_plugin_manager.h" #include "common/ge/tbe_plugin_manager.h"
#ifndef ONLY_COMPILE_OPEN_SRC
#include "toolchain/plog.h"
#endif


using domi::OpRegistry; using domi::OpRegistry;
using std::map; using std::map;
@@ -129,6 +132,11 @@ Status GEInitializeImpl(const std::map<string, string> &options) {


// Initialize GE, prepare for execution, call GELib::Initialize // Initialize GE, prepare for execution, call GELib::Initialize
Status GEInitialize(const std::map<string, string> &options) { 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); return GEInitializeImpl(options);
} }


@@ -143,6 +151,11 @@ Status GEInitialize(const std::map<AscendString, AscendString> &options) {
std::string val = option.second.GetString(); std::string val = option.second.GetString();
str_options[key] = val; str_options[key] = val;
} }
#ifndef ONLY_COMPILE_OPEN_SRC
if (DlogReportInitialize() != SUCCESS) {
GELOGW("Dlog report device log initialize failed.");
}
#endif
return GEInitializeImpl(str_options); return GEInitializeImpl(str_options);
} }


@@ -187,6 +200,12 @@ Status GEFinalize() {
// to avoid memory fragment, use malloc_trim to back free stack to system // to avoid memory fragment, use malloc_trim to back free stack to system
malloc_trim(0); malloc_trim(0);


#ifndef ONLY_COMPILE_OPEN_SRC
if (DlogReportFinalize() != SUCCESS) {
GELOGW("Dlog report device log finalize failed.");
}
#endif

GELOGT(TRACE_STOP, "GEFinalize finished"); GELOGT(TRACE_STOP, "GEFinalize finished");
return ret; return ret;
} }


+ 1
- 0
tests/ut/ge/CMakeLists.txt View File

@@ -806,6 +806,7 @@ add_library(ge_build_common STATIC ${GRAPH_BUILD_COMMON_SRC_FILES} ${PROTO_SRCS}


target_compile_definitions(ge_build_common PRIVATE target_compile_definitions(ge_build_common PRIVATE
google=ascend_private google=ascend_private
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
) )


target_link_libraries(ge_build_common PRIVATE target_link_libraries(ge_build_common PRIVATE


Loading…
Cancel
Save