Skip to content

P
Paddle

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

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

bring back share data

上级 0ed3f359
显示空白变更内容
内联 并排
Showing with 5 addition and 5 deletion
paddle/operators/parallel_do_op.cc
浏览文件 @ e443d034
...@@ -79,11 +79,11 @@ inline void CopyOrShare(const framework::Variable& src, ...@@ -79,11 +79,11 @@ inline void CopyOrShare(const framework::Variable& src,
auto *dst_sr = dst->GetMutable<SelectedRows>(); auto *dst_sr = dst->GetMutable<SelectedRows>();
dst_sr->set_rows(src_sr.rows()); dst_sr->set_rows(src_sr.rows());
dst_sr->set_height(src_sr.height()); dst_sr->set_height(src_sr.height());
if (src_sr.value().place() == dst_place) {
dst_sr->mutable_value()->ShareDataWith(src_sr.value());
} else {
Copy(src_sr.value(), dst_place, dst_sr->mutable_value()); Copy(src_sr.value(), dst_place, dst_sr->mutable_value());
// if (src_sr.value().place() == dst_place) { }
// dst_sr->mutable_value()->ShareDataWith(src_sr.value());
// } else {
// }
} else { } else {
PADDLE_THROW("Expect LoDTensor/SelectedRows, get %s", src.Type().name()); PADDLE_THROW("Expect LoDTensor/SelectedRows, get %s", src.Type().name());
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或