提交 d2ba20bd 编写于 作者: A anzhengqi

continue to send remain data when release gpu resource

上级 cbeb2dbf
...@@ -19,6 +19,10 @@ ...@@ -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 ...@@ -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
{ {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册