sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6561 [Data Dump] gpu dump iteration bug 

Merge pull request !6561 from caifubi/dump
This commit is contained in:
mindspore-ci-bot 2020-09-21 08:58:25 +08:00 committed by Gitee
parent 406052eb58 97be0fbc54
commit 63d6e139ab
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/backend/session/gpu_session.cc
View File

@ -344,6 +344,8 @@ void GPUSession::Dump(const std::shared_ptr<KernelGraph> &kernel_graph) const {
if (debugger_->DebuggerBackendEnabled()) {
MS_EXCEPTION_IF_NULL(kernel_graph);
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
if (debugger_ && dump_enabled) {
Dump(kernel_graph);
} else {
DumpJsonParser::GetInstance().UpdateDumpIter();
}
if (debugger_) {
debugger_->PostExecute();

2
mindspore/ccsrc/backend/session/kernel_graph.cc
View File

@ -731,7 +731,7 @@ void GetAllFatherRealNode(const AnfNodePtr &anf_node, std::vector<AnfNodePtr> *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);