We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 392ae69 + 23b12c6 commit 7f692b8Copy full SHA for 7f692b8
paddle/fluid/framework/details/multi_devices_graph_pass.cc
@@ -442,8 +442,7 @@ std::unique_ptr<ir::Graph> MultiDevSSAGraphBuilder::ApplyImpl(
442
use_gpu = nccl_ctxs_ != nullptr;
443
#endif
444
445
- if (use_gpu ||
446
- strategy_.reduce_ == BuildStrategy::ReduceStrategy::kAllReduce) {
+ if (use_gpu && strategy_.reduce_ == BuildStrategy::ReduceStrategy::kReduce) {
447
// Insert BCast Ops
448
for (size_t dev_id = 0; dev_id < bcast_var_name_set.size(); ++dev_id) {
449
auto &to_bcast_set = bcast_var_name_set[dev_id];
0 commit comments