未验证 提交 8413e4c3 编写于 作者: D duanyanhui 提交者: GitHub

fix custom_device CopyToCpu to avoid crash (#53400)

上级 b6996598
...@@ -463,7 +463,7 @@ void Tensor::CopyToCpuImpl(T *data, ...@@ -463,7 +463,7 @@ void Tensor::CopyToCpuImpl(T *data,
t_data, t_data,
ele_num * sizeof(T), ele_num * sizeof(T),
dev_ctx->stream()); dev_ctx->stream());
// TODO(wangran16): sync_stream dev_ctx->GetStream()->Synchronize();
#else #else
PADDLE_THROW(paddle::platform::errors::InvalidArgument( PADDLE_THROW(paddle::platform::errors::InvalidArgument(
"The analysis predictor supports CPU, GPU, NPU and XPU now.")); "The analysis predictor supports CPU, GPU, NPU and XPU now."));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册