dist train use split_by_ref

PaddlePaddle · Apr 17, 2018 · ed89b7b · ed89b7b
1 parent 0c6eef3
commit ed89b7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/distribute_transpiler.py b/python/paddle/fluid/distribute_transpiler.py
@@ -824,7 +824,7 @@ def _append_split_op(self, program, gradblocks):
                 for v in splited_vars:
                     sections.append(v.shape[0])
                 program.global_block().append_op(
-                    type="split",
+                    type="split_byref",
                     inputs={"X": orig_var},
                     outputs={"Out": splited_vars},
                     attrs={"sections": sections}  # assume split evenly