|
|
|
@ -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
|
|
|
|
|
{
|
|
|
|
|