Add wait

7 years ago · a822f8ddbb
parent e9e27e0f32
commit a822f8ddbb
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/pybind/tensor_py.h
+++ b/paddle/fluid/pybind/tensor_py.h
@ -190,6 +190,7 @@ void PyCUDATensorSetFromArray(
      static_cast<const platform::CUDADeviceContext *>(pool.Get(place));
  paddle::platform::GpuMemcpyAsync(dst, array.data(), sizeof(T) * array.size(),
                                   cudaMemcpyHostToDevice, dev_ctx->stream());
+  dev_ctx->Wait();
 }

 template <>