!1199 fix profiling core dump

Merge pull request !1199 from caifubi/fix-profiling-core-dump
5 years ago · 590e3f5e62
parent 0040764dec 8a845e2f07
commit 590e3f5e62
1 changed files with 1 additions and 0 deletions
--- a/mindspore/ccsrc/device/ascend/profiling/profiling_utils.cc
+++ b/mindspore/ccsrc/device/ascend/profiling/profiling_utils.cc
@ -324,6 +324,7 @@ void ProfilingUtils::ReportProfilingData(const std::vector<uint32_t> &task_ids,

  GraphDescReporter graph_reporter(context->device_id(), "vm.graph_desc_info_" + std::to_string(graph->graph_id()),
                                   ret->second);
+  graph_profiling_cnode_.erase(ret);
  graph_reporter.ReportData();
 }
 }  // namespace ascend