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

reset exception listener

This commit is contained in:
kswang 2021-01-06 10:47:14 +08:00
parent 1d1f6841b9
commit 51d6dbdffe
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -292,6 +292,7 @@ void Executor::RunTask(const std::shared_ptr<Task> &task, bool sync) {
return finished;
});
}
ClearDoneTasks();
MsException::Instance().CheckException();
}

1
mindspore/core/ir/tensor.h
View File

@ -89,6 +89,7 @@ class WaitEvent : public ExceptionListener {
}
MsException::Instance().SetExceptionListener(const_cast<WaitEvent *>(this));
cond_var_.wait(lock, [this] { return !need_wait_; });
MsException::Instance().SetExceptionListener(nullptr);
MsException::Instance().CheckException();
}