dist train use split_by_ref

7 years ago · ed89b7b7e6
parent 0c6eef3e58
commit ed89b7b7e6
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/distribute_transpiler.py
+++ b/python/paddle/fluid/distribute_transpiler.py
@ -824,7 +824,7 @@ class DistributeTranspiler:
                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