Skip to content

Commit 7ae03ec

Browse files
author
Yancey
authored
fix lack split_byref op in parallel executor (#10770)
1 parent 0e4467e commit 7ae03ec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/framework/details/multi_devices_graph_builder.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ bool MultiDevSSAGraphBuilder::IsDistTrainOp(const OpDesc &op,
9898
return false;
9999
};
100100

101-
if (op.Type() == "split") {
101+
if (op.Type() == "split" || op.Type() == "split_byref") {
102102
return checker(op.OutputArgumentNames(), send_op->InputArgumentNames());
103103
} else if (op.Type() == "concat") {
104104
return checker(op.InputArgumentNames(), send_op->OutputArgumentNames());

0 commit comments

Comments
 (0)