From f4ca8b6783804cda69ec00b905036c8175c99eb2 Mon Sep 17 00:00:00 2001 From: liyong Date: Thu, 18 Mar 2021 16:08:24 +0800 Subject: [PATCH] fix gil release --- mindspore/ccsrc/backend/session/executor.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/session/executor.cc b/mindspore/ccsrc/backend/session/executor.cc index bd262f4758..26df6185d5 100644 --- a/mindspore/ccsrc/backend/session/executor.cc +++ b/mindspore/ccsrc/backend/session/executor.cc @@ -415,8 +415,10 @@ void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const Gr // Release GIL before calling into (potentially long-running) C++ code if (Py_IsInitialized()) { py::gil_scoped_release release; + session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); + } else { + session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); } - session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask); } } else { auto task = std::make_shared();