Browse Source

!6561 [Data Dump] gpu dump iteration bug

Merge pull request !6561 from caifubi/dump
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
63d6e139ab
2 changed files with 5 additions and 1 deletions
  1. +4
    -0
      mindspore/ccsrc/backend/session/gpu_session.cc
  2. +1
    -1
      mindspore/ccsrc/backend/session/kernel_graph.cc

+ 4
- 0
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();


+ 1
- 1
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);


Loading…
Cancel
Save