diff --git a/paddle/operators/parallel_do_op.cc b/paddle/operators/parallel_do_op.cc index eb6308d306af24894207705d6413ac1fec55577b..6c85ca6cde7d1be94609d976554dd89175e67f4b 100644 --- a/paddle/operators/parallel_do_op.cc +++ b/paddle/operators/parallel_do_op.cc @@ -152,7 +152,9 @@ class ParallelDoOp : public framework::OperatorBase { auto *sub_scope = sub_scopes[i]; auto *dst = sub_scope->Var(param)->GetMutable(); framework::Copy(src, place, dst); - dst->set_lod(src.lod()); + framework::LoD lod(src.lod()); + lod.CopyToPeer(place); + dst->set_lod(lod); } } WaitOnPlaces(places);