mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!19314 Fix receive WaitCMD instead of terminate for multigraph on GPU issue 

Merge pull request !19314 from parastooashtari/master
This commit is contained in:
i-robot 2021-07-04 03:20:45 +00:00 committed by Gitee
parent 8008103050 a153f6b35f
commit 66c03ebc06
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/debug/debugger/debugger.cc
View File

@ -351,6 +351,9 @@ void Debugger::PreExecute(const KernelGraphPtr &graph_ptr, uint32_t graph_sum) {
} else if (graph_id == rungraph_id_list_.front() && device_target_ == kGPUDevice) {
// stop only when receive the first sub run graph for each step
// if we have stopped for the last kernel before, no need to stop again
if (pipeline::ExecutorPy::GetDebugTerminate()) {
return;
}
if (!(run_level_ == "node" && suspended_at_last_kernel_)) {
CommandLoop();
}

2
mindspore/ccsrc/debug/debugger/debugger_utils.cc
View File

@ -51,6 +51,7 @@ std::vector<int> CheckRealOutput(const std::string &node_name, const size_t &out
}
return real_outputs;
}
void LoadInputs(const CNodePtr &cnode, const KernelLaunchInfo *launch_info_, uint32_t exec_order_) {
// get inputs
auto kernel_inputs = launch_info_->inputs_;
@ -77,6 +78,7 @@ void LoadInputs(const CNodePtr &cnode, const KernelLaunchInfo *launch_info_, uin
#endif
}
}
void LoadOutputs(const CNodePtr &cnode, const KernelLaunchInfo *launch_info_, uint32_t exec_order_) {
// get outputs
auto kernel_outputs = launch_info_->outputs_;