p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5730 fix tensor release exception 

Merge pull request !5730 from kisnwang/async-run-graph
This commit is contained in:
mindspore-ci-bot 2020-09-04 11:34:47 +08:00 committed by Gitee
parent 9018737e99 6f527e0cd2
commit 7045d5941f
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mindspore/ccsrc/backend/session/executor.cc
View File

@ -141,17 +141,20 @@ void Executor::WorkerLoop() {
} catch (const std::exception &e) {
exception_ptr_ = std::current_exception();
}
if (task->type_ == kCompileNodes) {
auto task_type = task->type_;
task = nullptr;
if (task_type == kCompileNodes) {
compile_cond_var_.notify_all();
} else if (task->type_ == kCompileGraph) {
} else if (task_type == kCompileGraph) {
compile_cond_var_.notify_all();
} else if (task->type_ == kBuildGraph) {
} else if (task_type == kBuildGraph) {
build_cond_var_.notify_all();
} else if (task->type_ == kRunGraph) {
} else if (task_type == kRunGraph) {
run_cond_var_.notify_all();
} else if (task->type_ == kBuildOp) {
} else if (task_type == kBuildOp) {
build_op_cond_var_.notify_all();
} else if (task->type_ == kRunOp) {
} else if (task_type == kRunOp) {
run_op_cond_var_.notify_all();
}
}