dist train use split_by_ref

ed89b7b7 · typhoonzero · 0c6eef3e · ed89b7b7
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/distribute_transpiler.py python/paddle/fluid/distribute_transpiler.py +1 -1

未找到文件。
--- 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