Add wait

a822f8dd · Yu Yang · e9e27e0f · a822f8dd
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/pybind/tensor_py.h paddle/fluid/pybind/tensor_py.h +1 -0

未找到文件。
--- 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 <>