| @@ -19,6 +19,10 @@ | |||||
| #include <utility> | #include <utility> | ||||
| #include "utils/log_adapter.h" | #include "utils/log_adapter.h" | ||||
| #include "utils/ms_utils.h" | #include "utils/ms_utils.h" | ||||
| #include "pybind11/pybind11.h" | |||||
| #include "pybind11/stl.h" | |||||
| namespace py = pybind11; | |||||
| namespace mindspore { | namespace mindspore { | ||||
| namespace device { | namespace device { | ||||
| @@ -167,6 +171,7 @@ inline bool GpuBufferMgr::isCreated(unsigned int device_id, const std::string &c | |||||
| } | } | ||||
| bool GpuBufferMgr::CloseNotify() { | bool GpuBufferMgr::CloseNotify() { | ||||
| py::gil_scoped_release release; | |||||
| bool result = true; | bool result = true; | ||||
| // lock scope | // lock scope | ||||
| { | { | ||||