Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

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

Sync Copy

上级 d085f507
显示空白变更内容
内联 并排
Showing with 1 addition and 0 deletion
paddle/fluid/pybind/tensor_py.h
浏览文件 @ 0ca28b85
...@@ -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.
先完成此消息的编辑！
想要评论请 注册 或