Browse Source

fix release tensor exception

tags/v1.0.0
kswang 5 years ago
parent
commit
6f527e0cd2
1 changed files with 9 additions and 6 deletions
  1. +9
    -6
      mindspore/ccsrc/backend/session/executor.cc

+ 9
- 6
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();
}
}


Loading…
Cancel
Save