Col_lin
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix serving gil 

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
zhoufeng 2022-03-11 16:42:15 +08:00
parent 0b1ec574cf
commit dc2633df03
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/pybind_api/gil_scoped_long_running.h
View File

@ -28,7 +28,11 @@ namespace py = pybind11;
namespace mindspore {
class GilScopedLongRunningHook : public ScopedLongRunningHook {
public:
void Enter() override { release_ = std::make_unique<py::gil_scoped_release>(); }
void Enter() override {
if (PyGILState_Check() != 0) {
release_ = std::make_unique<py::gil_scoped_release>();
}
}
void Leave() override { release_ = nullptr; }
private: