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

Throw original exection in PyNative lazy build

This commit is contained in:
caifubi 2021-11-25 14:58:39 +08:00
parent 5048d3823b
commit bd6de7162b
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ccsrc/vm/backend.cc
View File

@ -27,6 +27,7 @@
#include "pipeline/jit/parse/data_converter.h"
#include "ir/anf.h"
#include "pybind_api/ir/base_ref_py.h"
#include "pybind_api/pybind_patch.h"
#include "utils/callbacks.h"
#include "utils/convert_utils.h"
#include "utils/log_adapter.h"
@ -1160,6 +1161,18 @@ void MindRTBackend::LazyExecuteTaskCallback() {
ms_context->set_param<bool>(MS_CTX_ENABLE_PYNATIVE_INFER, infer_flag);
MS_LOG(DEBUG) << "End";
} catch (const py::type_error &ex) {
op_lazy_builder.Reset();
throw py::type_error(ex);
} catch (const py::value_error &ex) {
op_lazy_builder.Reset();
throw py::value_error(ex);
} catch (const py::index_error &ex) {
op_lazy_builder.Reset();
throw py::index_error(ex);
} catch (const py::name_error &ex) {
op_lazy_builder.Reset();
throw py::name_error(ex);
} catch (const std::exception &ex) {
op_lazy_builder.Reset();
throw(std::runtime_error(ex.what()));