!6561 [Data Dump] gpu dump iteration bug

Merge pull request !6561 from caifubi/dump
pull/6561/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 63d6e139ab

@ -344,6 +344,8 @@ void GPUSession::Dump(const std::shared_ptr<KernelGraph> &kernel_graph) const {
if (debugger_->DebuggerBackendEnabled()) { if (debugger_->DebuggerBackendEnabled()) {
MS_EXCEPTION_IF_NULL(kernel_graph); MS_EXCEPTION_IF_NULL(kernel_graph);
E2eDumpUtil::DumpData(kernel_graph.get(), debugger_.get()); E2eDumpUtil::DumpData(kernel_graph.get(), debugger_.get());
} else {
DumpJsonParser::GetInstance().UpdateDumpIter();
} }
} }
@ -365,6 +367,8 @@ void GPUSession::PostIterationDbg(const std::shared_ptr<KernelGraph> &kernel_gra
// debug used for dump // debug used for dump
if (debugger_ && dump_enabled) { if (debugger_ && dump_enabled) {
Dump(kernel_graph); Dump(kernel_graph);
} else {
DumpJsonParser::GetInstance().UpdateDumpIter();
} }
if (debugger_) { if (debugger_) {
debugger_->PostExecute(); debugger_->PostExecute();

@ -731,7 +731,7 @@ void GetAllFatherRealNode(const AnfNodePtr &anf_node, std::vector<AnfNodePtr> *r
MS_EXCEPTION_IF_NULL(result); MS_EXCEPTION_IF_NULL(result);
MS_EXCEPTION_IF_NULL(visited); MS_EXCEPTION_IF_NULL(visited);
if (visited->find(anf_node) != visited->end()) { 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; return;
} }
visited->insert(anf_node); visited->insert(anf_node);

Loading…
Cancel
Save