fix op name typo

revert-11610-move_hooks
Yancey1989 7 years ago
parent 6d69ae0c6e
commit 82d741c4b9

@ -230,9 +230,9 @@ std::unique_ptr<SSAGraph> 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(),
PADDLE_ENFORCE(got != remote_vars_devices_.end(),
"can not find right place to concat received var.");
CreateDistTrainOp(&result, *op, got->second);
} else {

Loading…
Cancel
Save