未验证 提交 0d3de8d0 编写于 作者: L Leo Chen 提交者: GitHub

[NPU] remove duplicated stream sync in fetch op (#33819)

上级 43c38c67
...@@ -44,11 +44,6 @@ static void DataCopy(const framework::LoDTensor &src_item, ...@@ -44,11 +44,6 @@ static void DataCopy(const framework::LoDTensor &src_item,
TensorCopySync(src_item, platform::CPUPlace(), dst_item); TensorCopySync(src_item, platform::CPUPlace(), dst_item);
} }
#else #else
#ifdef PADDLE_WITH_ASCEND_CL
if (platform::is_npu_place(src_item.place())) {
platform::DeviceContextPool::Instance().Get(src_item.place())->Wait();
}
#endif
TensorCopySync(src_item, platform::CPUPlace(), dst_item); TensorCopySync(src_item, platform::CPUPlace(), dst_item);
#endif #endif
} else { } else {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册