File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
paddle/fluid/framework/details Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -431,7 +431,9 @@ std::unique_ptr<ir::Graph> MultiDevSSAGraphBuilder::ApplyImpl(
431
431
CreateReduceOp (&result, g_name, cur_device_id);
432
432
graph->Get <ShardedVarDevice>(kShardedVarDevice )
433
433
.emplace (g_name, cur_device_id);
434
- bcast_var_name_set[cur_device_id].emplace (p_name);
434
+ if (!is_dist_train) {
435
+ bcast_var_name_set[cur_device_id].emplace (p_name);
436
+ }
435
437
break ;
436
438
case BuildStrategy::ReduceStrategy::kAllReduce :
437
439
if (IsSparseGradient (g_name)) {
@@ -461,7 +463,7 @@ std::unique_ptr<ir::Graph> MultiDevSSAGraphBuilder::ApplyImpl(
461
463
if ((use_gpu &&
462
464
strategy_.reduce_ == BuildStrategy::ReduceStrategy::kReduce ) ||
463
465
is_dist_train) {
464
- // Insert BCast Ops
466
+ // allways broadcast receieved parameters for distributed training
465
467
for (size_t dev_id = 0 ; dev_id < bcast_var_name_set.size (); ++dev_id) {
466
468
auto &to_bcast_set = bcast_var_name_set[dev_id];
467
469
for (auto &bcast_name : to_bcast_set) {
You can’t perform that action at this time.
0 commit comments