diff --git a/paddle/fluid/framework/details/multi_devices_graph_builder.cc b/paddle/fluid/framework/details/multi_devices_graph_builder.cc index 1c1b11d0e445b97469a19ca7c09ddfe249b1b63f..d1683774b4d508af469447eaa1c305f5f6a18c4b 100644 --- a/paddle/fluid/framework/details/multi_devices_graph_builder.cc +++ b/paddle/fluid/framework/details/multi_devices_graph_builder.cc @@ -230,10 +230,10 @@ std::unique_ptr MultiDevSSAGraphBuilder::Build( } CreateDistTrainOp(&result, *op, rpc_op_device_id); } - if (op->Type() == "oncat") { + if (op->Type() == "concat") { auto got = remote_vars_devices_.find(op->InputArgumentNames()[0]); - PADDLE_ENFORCE_NE(got != remote_vars_devices_.end(), - "can not find right place to concat received var."); + PADDLE_ENFORCE(got != remote_vars_devices_.end(), + "can not find right place to concat received var."); CreateDistTrainOp(&result, *op, got->second); } else { CreateDistTrainOp(&result, *op, 0);