!5964 GeneratorDataset continue to send remain data when release gpu resource for GPU

Merge pull request !5964 from anzhengqi/I1TK0O-close-gpu2
This commit is contained in:
mindspore-ci-bot 2020-09-10 15:13:21 +08:00 committed by Gitee
commit ae7e8a7492
1 changed files with 5 additions and 0 deletions

View File

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