p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add longrunning for runop

This commit is contained in:
kswang 2020-12-01 10:09:48 +08:00
parent 4bd9460e45
commit 0ad5357ce7
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/backend/session/executor.cc
View File

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

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);