m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix op name typo

Browse Source
revert-11610-move_hooks
Yancey1989 7 years ago
parent 6d69ae0c6e
commit 82d741c4b9
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File

6
paddle/fluid/framework/details/multi_devices_graph_builder.cc
Unescape View File

@ -230,10 +230,10 @@ std::unique_ptr<SSAGraph> MultiDevSSAGraphBuilder::Build(
} }
CreateDistTrainOp(&result, *op, rpc_op_device_id); CreateDistTrainOp(&result, *op, rpc_op_device_id);
} }
if (op->Type() == "oncat") { if (op->Type() == "concat") {
auto got = remote_vars_devices_.find(op->InputArgumentNames()[0]); 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."); "can not find right place to concat received var.");
CreateDistTrainOp(&result, *op, got->second); CreateDistTrainOp(&result, *op, got->second);
} else { } else {
CreateDistTrainOp(&result, *op, 0); CreateDistTrainOp(&result, *op, 0);

Write Preview
Loading…
Cancel
Save