Browse Source

!9256 remove gil release in gpu runtime

From: @kisnwang
Reviewed-by: @jjfeing,@chujinjin
Signed-off-by: @chujinjin
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
125940314f
2 changed files with 1 additions and 3 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/session/executor.cc
  2. +0
    -3
      mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc

+ 1
- 0
mindspore/ccsrc/backend/session/executor.cc View File

@@ -349,6 +349,7 @@ void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const Gr
tensor->Wait();
}
}
mindspore::ScopedLongRunning long_running;
SyncRunTask(task);
*outputs = task->outputs_;
}


+ 0
- 3
mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc View File

@@ -16,7 +16,6 @@
#include "runtime/device/gpu/gpu_kernel_runtime.h"
#include <algorithm>
#include <map>
#include "pybind11/pybind11.h"
#include "runtime/device/gpu/gpu_device_address.h"
#include "runtime/device/gpu/cuda_driver.h"
#include "runtime/device/gpu/gpu_buffer_mgr.h"
@@ -45,7 +44,6 @@
namespace mindspore {
namespace device {
namespace gpu {
namespace py = pybind11;
using mindspore::device::memswap::MemSwapInfoSet;
using mindspore::device::memswap::MemSwapManager;
using mindspore::device::memswap::SwapKind;
@@ -362,7 +360,6 @@ bool GPUKernelRuntime::Run(session::KernelGraph *graph, bool is_task_sink) {

ret = RunOneStep(graph);
} else {
py::gil_scoped_release gil_release;
ret = LaunchKernel(graph);
}
(void)gettimeofday(&end_time, nullptr);


Loading…
Cancel
Save