Browse Source

avoid trace_provider being set repeatedly

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
feature/build-system-rewrite
zhoufeng 4 years ago
parent
commit
1689121532
1 changed files with 9 additions and 1 deletions
  1. +9
    -1
      mindspore/core/utils/log_adapter.cc

+ 9
- 1
mindspore/core/utils/log_adapter.cc View File

@@ -126,7 +126,15 @@ void LogWriter::set_exception_handler(const ExceptionHandler &exception_handler)
exception_handler_ = exception_handler;
}

void LogWriter::set_trace_provider(const TraceProvider &trace_provider) { trace_provider_ = trace_provider; }
void LogWriter::set_trace_provider(const TraceProvider &trace_provider) {
// todo: delete has_set flag
static bool has_set = false;
if (!has_set) {
MS_LOG(WARNING) << "trace provider has been set, skip.";
trace_provider_ = trace_provider;
has_set = true;
}
}

LogWriter::TraceProvider LogWriter::trace_provider() { return trace_provider_; }



Loading…
Cancel
Save