!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:
commit
ae7e8a7492
|
@ -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
|
||||
{
|
||||
|
|
Loading…
Reference in New Issue