Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2298
Star 20931
Fork 5422
P
Paddle
提交 a822f8dd 编写于 作者: Y Yu Yang
浏览文件
操作

Add wait

上级 e9e27e0f
隐藏空白更改
内联 并排
Showing with 1 addition and 0 deletion
paddle/fluid/pybind/tensor_py.h
浏览文件 @ a822f8dd
...@@ -190,6 +190,7 @@ void PyCUDATensorSetFromArray( ...@@ -190,6 +190,7 @@ void PyCUDATensorSetFromArray(
static_cast<const platform::CUDADeviceContext *>(pool.Get(place)); static_cast<const platform::CUDADeviceContext *>(pool.Get(place));
paddle::platform::GpuMemcpyAsync(dst, array.data(), sizeof(T) * array.size(), paddle::platform::GpuMemcpyAsync(dst, array.data(), sizeof(T) * array.size(),
cudaMemcpyHostToDevice, dev_ctx->stream()); cudaMemcpyHostToDevice, dev_ctx->stream());
dev_ctx->Wait();
} }
template <> template <>
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或