From 97be0fbc54f649636cd7089563db0e24c869128c Mon Sep 17 00:00:00 2001 From: caifubi Date: Sat, 19 Sep 2020 15:34:08 +0800 Subject: [PATCH] fix gpu dump iteration bug --- mindspore/ccsrc/backend/session/gpu_session.cc | 4 ++++ mindspore/ccsrc/backend/session/kernel_graph.cc | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc index ce8a6eed0f..6c5ff97f1b 100644 --- a/mindspore/ccsrc/backend/session/gpu_session.cc +++ b/mindspore/ccsrc/backend/session/gpu_session.cc @@ -343,6 +343,8 @@ void GPUSession::Dump(const std::shared_ptr &kernel_graph) const { if (debugger_->DebuggerBackendEnabled()) { MS_EXCEPTION_IF_NULL(kernel_graph); E2eDumpUtil::DumpData(kernel_graph.get(), debugger_.get()); + } else { + DumpJsonParser::GetInstance().UpdateDumpIter(); } } @@ -364,6 +366,8 @@ void GPUSession::PostIterationDbg(const std::shared_ptr &kernel_gra // debug used for dump if (debugger_ && dump_enabled) { Dump(kernel_graph); + } else { + DumpJsonParser::GetInstance().UpdateDumpIter(); } if (debugger_) { debugger_->PostExecute(); diff --git a/mindspore/ccsrc/backend/session/kernel_graph.cc b/mindspore/ccsrc/backend/session/kernel_graph.cc index c0c0e0a6ff..31c3cab446 100644 --- a/mindspore/ccsrc/backend/session/kernel_graph.cc +++ b/mindspore/ccsrc/backend/session/kernel_graph.cc @@ -731,7 +731,7 @@ void GetAllFatherRealNode(const AnfNodePtr &anf_node, std::vector *r MS_EXCEPTION_IF_NULL(result); MS_EXCEPTION_IF_NULL(visited); if (visited->find(anf_node) != visited->end()) { - MS_LOG(WARNING) << "Node:" << anf_node->fullname_with_scope() << " has alreday been visited"; + MS_LOG(INFO) << "Node:" << anf_node->fullname_with_scope() << " has alreday been visited"; return; } visited->insert(anf_node);