Skip to content

Commit e443d03

Browse files
author
Yang Yang
committed
bring back share data
1 parent 0ed3f35 commit e443d03

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

paddle/operators/parallel_do_op.cc

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -79,11 +79,11 @@ inline void CopyOrShare(const framework::Variable& src,
7979
auto *dst_sr = dst->GetMutable<SelectedRows>();
8080
dst_sr->set_rows(src_sr.rows());
8181
dst_sr->set_height(src_sr.height());
82-
Copy(src_sr.value(), dst_place, dst_sr->mutable_value());
83-
// if (src_sr.value().place() == dst_place) {
84-
// dst_sr->mutable_value()->ShareDataWith(src_sr.value());
85-
// } else {
86-
// }
82+
if (src_sr.value().place() == dst_place) {
83+
dst_sr->mutable_value()->ShareDataWith(src_sr.value());
84+
} else {
85+
Copy(src_sr.value(), dst_place, dst_sr->mutable_value());
86+
}
8787
} else {
8888
PADDLE_THROW("Expect LoDTensor/SelectedRows, get %s", src.Type().name());
8989
}

0 commit comments

Comments
 (0)