Skip to content

Commit e0b396e

Browse files
committed
update by comment
1 parent 39277e9 commit e0b396e

File tree

1 file changed

+2
-12
lines changed

1 file changed

+2
-12
lines changed

python/paddle/fluid/distribute_transpiler.py

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -412,12 +412,7 @@ def _get_splited_name_and_shape(varname):
412412
tmpvar = s_prog.global_block().create_var(
413413
name=var.name, persistable=var.persistable, type=var.type)
414414
else:
415-
tmpvar = s_prog.global_block().create_var(
416-
name=var.name,
417-
persistable=var.persistable,
418-
type=var.type,
419-
dtype=var.dtype,
420-
shape=var.shape)
415+
tmpvar = s_prog.global_block().clone_variable(var)
421416
created_var_map[var.name] = tmpvar
422417

423418
# 2. rename op outputs
@@ -713,18 +708,13 @@ def _append_pserver_non_opt_ops(self, optimize_block, opt_op):
713708
varlist = [varlist]
714709

715710
for var in varlist:
716-
print("##### deal var: ", var)
717711
if var.type == core.VarDesc.VarType.STEP_SCOPES:
718712
program.global_block().create_var(
719713
name=var.name,
720714
persistable=var.persistable,
721715
type=var.type)
722716
else:
723-
program.global_block().create_var(
724-
name=var.name,
725-
persistable=var.persistable,
726-
dtype=var.dtype,
727-
shape=var.shape)
717+
program.global_block().clone_variable(var)
728718

729719
optimize_block.append_op(
730720
type=opt_op.type,

0 commit comments

Comments
 (0)