cpu reduce mode did not need to broadcast test=develop

b99db0e2 · Qiao Longfei · bec68fa0 · b99db0e2
隐藏空白更改
内联并排

Showing with 1 addition and 3 deletion

paddle/fluid/framework/details/multi_devices_graph_pass.cc paddle/fluid/framework/details/multi_devices_graph_pass.cc +1 -3

未找到文件。
--- a/paddle/fluid/framework/details/multi_devices_graph_pass.cc
+++ b/paddle/fluid/framework/details/multi_devices_graph_pass.cc
@@ -925,9 +925,7 @@ void DistSSAGraphBuilder::InsertCollectiveOp(ir::Graph *result,
 }
 void DistSSAGraphBuilder::InsertPostprocessOps(ir::Graph *result) const {
-  if (need_broadcast_var_ ||
+  if (UseGPU() && strategy_.reduce_ == BuildStrategy::ReduceStrategy::kReduce) {
-      (UseGPU() &&
-       strategy_.reduce_ == BuildStrategy::ReduceStrategy::kReduce)) {
    if (strategy_.fuse_broadcast_op_) {
      CreateFusedBroadcastOp(result, bcast_var_name_set_);
    } else {