浏览代码

continue to send remain data when release gpu resource

tags/v1.0.0
anzhengqi 5 年前
父节点
当前提交
d2ba20bdbc
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. +5
    -0
      mindspore/ccsrc/runtime/device/gpu/gpu_buffer_mgr.cc

+ 5
- 0
mindspore/ccsrc/runtime/device/gpu/gpu_buffer_mgr.cc 查看文件

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


正在加载...
取消
保存