Skip to content

Commit 7f692b8

Browse files
author
Yan Xu
authored
Merge pull request #13347 from Yancey1989/fix_reduce_bast
fix bcast in reduce strategy
2 parents 392ae69 + 23b12c6 commit 7f692b8

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

paddle/fluid/framework/details/multi_devices_graph_pass.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -442,8 +442,7 @@ std::unique_ptr<ir::Graph> MultiDevSSAGraphBuilder::ApplyImpl(
442442
use_gpu = nccl_ctxs_ != nullptr;
443443
#endif
444444

445-
if (use_gpu ||
446-
strategy_.reduce_ == BuildStrategy::ReduceStrategy::kAllReduce) {
445+
if (use_gpu && strategy_.reduce_ == BuildStrategy::ReduceStrategy::kReduce) {
447446
// Insert BCast Ops
448447
for (size_t dev_id = 0; dev_id < bcast_var_name_set.size(); ++dev_id) {
449448
auto &to_bcast_set = bcast_var_name_set[dev_id];

0 commit comments

Comments
 (0)