Merge remote-tracking branch 'pr/7754' into benchmark_parallel_do

emailweixu-patch-1
Yang Yang 7 years ago
commit 03acc45400

@ -65,9 +65,9 @@ static void SplitTensorAndMoveTensorToScopes(
}
}
inline void CopyOrShare(const framework::Variable& src,
const platform::Place& dst_place,
framework::Variable* dst) {
inline void CopyOrShare(const framework::Variable &src,
const platform::Place &dst_place,
framework::Variable *dst) {
if (src.IsType<LoDTensor>()) {
if (src.Get<LoDTensor>().place() == dst_place) {
dst->GetMutable<LoDTensor>()->ShareDataWith(src.Get<LoDTensor>());

Loading…
Cancel
Save