Skip to content

Commit 59a25fd

Browse files
committed
update
1 parent 74701b2 commit 59a25fd

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,8 @@ def transpile(self,
410410
attrs={"axis": 0})
411411

412412
if self.has_distributed_lookup_table:
413-
self._replace_lookup_table_op_with_prefetch(program, pserver_endpoints)
413+
self._replace_lookup_table_op_with_prefetch(program,
414+
pserver_endpoints)
414415
self._split_table_grad_and_add_send_vars(program, pserver_endpoints)
415416

416417
def get_trainer_program(self):
@@ -631,7 +632,8 @@ def _get_splited_name_and_shape(varname):
631632
return s_prog
632633

633634
# transpiler function for dis lookup_table
634-
def _replace_lookup_table_op_with_prefetch(self, program, pserver_endpoints):
635+
def _replace_lookup_table_op_with_prefetch(self, program,
636+
pserver_endpoints):
635637
# 1. replace lookup_table_op with split_ids_op -> prefetch_op -> sum_op
636638
self.prefetch_input_vars = None
637639
self.prefetch_output_vars = None

0 commit comments

Comments
 (0)