提交 e443d034 编写于 作者: Y Yang Yang

bring back share data

上级 0ed3f359
...@@ -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());
Copy(src_sr.value(), dst_place, dst_sr->mutable_value()); if (src_sr.value().place() == dst_place) {
// if (src_sr.value().place() == dst_place) { dst_sr->mutable_value()->ShareDataWith(src_sr.value());
// dst_sr->mutable_value()->ShareDataWith(src_sr.value()); } else {
// } else { Copy(src_sr.value(), dst_place, dst_sr->mutable_value());
// } }
} 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.
先完成此消息的编辑!
想要评论请 注册