update profiling training_Trace parser

4 years ago · 71f0dd4cab
parent d2ef3006a2
commit 71f0dd4cab
1 changed files with 3 additions and 0 deletions
--- a/ge/common/profiling/profiling_manager.cc
+++ b/ge/common/profiling/profiling_manager.cc
@ -143,6 +143,9 @@ ge::Status ProfilingManager::ParseOptions(const std::string &options) {
  }
  try {
    Json prof_options = Json::parse(options);
+    if (options.find(kTrainingTrace) == std::string::npos) {
+      return ge::SUCCESS;
+    }
    const std::string training_trace = prof_options[kTrainingTrace];
    if (training_trace.empty()) {
      GELOGI("Training trace will not take effect.");