Browse Source

fix pynative gpu eval

tags/v1.0.0
panfengfeng 5 years ago
parent
commit
db558cd01b
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc

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

@@ -15,6 +15,7 @@
*/ */
#include "runtime/device/gpu/gpu_kernel_runtime.h" #include "runtime/device/gpu/gpu_kernel_runtime.h"
#include <algorithm> #include <algorithm>
#include "pybind11/pybind11.h"
#include "runtime/device/gpu/gpu_device_address.h" #include "runtime/device/gpu/gpu_device_address.h"
#include "runtime/device/gpu/cuda_driver.h" #include "runtime/device/gpu/cuda_driver.h"
#include "runtime/device/gpu/gpu_buffer_mgr.h" #include "runtime/device/gpu/gpu_buffer_mgr.h"
@@ -40,6 +41,7 @@
namespace mindspore { namespace mindspore {
namespace device { namespace device {
namespace gpu { namespace gpu {
namespace py = pybind11;
using mindspore::device::memswap::MemSwapInfoSet; using mindspore::device::memswap::MemSwapInfoSet;
using mindspore::device::memswap::MemSwapManager; using mindspore::device::memswap::MemSwapManager;
using mindspore::device::memswap::SwapKind; using mindspore::device::memswap::SwapKind;
@@ -468,6 +470,7 @@ bool GPUKernelRuntime::Run(session::KernelGraph *graph, bool is_task_sink, Debug


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


Loading…
Cancel
Save