fix lack split_byref op in parallel executor (#10770)

7ae03ec0 · Yancey · GitHub · 0e4467ee · 7ae03ec0
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/framework/details/multi_devices_graph_builder.cc ...le/fluid/framework/details/multi_devices_graph_builder.cc +1 -1

未找到文件。
--- a/paddle/fluid/framework/details/multi_devices_graph_builder.cc
+++ b/paddle/fluid/framework/details/multi_devices_graph_builder.cc
@@ -98,7 +98,7 @@ bool MultiDevSSAGraphBuilder::IsDistTrainOp(const OpDesc &op,
    return false;
  };

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