提交 7045d594 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5730 fix tensor release exception

Merge pull request !5730 from kisnwang/async-run-graph
...@@ -141,17 +141,20 @@ void Executor::WorkerLoop() { ...@@ -141,17 +141,20 @@ void Executor::WorkerLoop() {
} catch (const std::exception &e) { } catch (const std::exception &e) {
exception_ptr_ = std::current_exception(); 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(); compile_cond_var_.notify_all();
} else if (task->type_ == kCompileGraph) { } else if (task_type == kCompileGraph) {
compile_cond_var_.notify_all(); compile_cond_var_.notify_all();
} else if (task->type_ == kBuildGraph) { } else if (task_type == kBuildGraph) {
build_cond_var_.notify_all(); build_cond_var_.notify_all();
} else if (task->type_ == kRunGraph) { } else if (task_type == kRunGraph) {
run_cond_var_.notify_all(); run_cond_var_.notify_all();
} else if (task->type_ == kBuildOp) { } else if (task_type == kBuildOp) {
build_op_cond_var_.notify_all(); build_op_cond_var_.notify_all();
} else if (task->type_ == kRunOp) { } else if (task_type == kRunOp) {
run_op_cond_var_.notify_all(); run_op_cond_var_.notify_all();
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册