Skip to content

Commit b73806b

Browse files
authored
Merge pull request #12347 from jacquesqiao/fix-distribute_transpiler_for_dist_lookup_table
fix _create_prefetch_block in distribute_transpiler
2 parents f04ae97 + aa3618e commit b73806b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -887,7 +887,8 @@ def _create_table_optimize_block(self, pserver_index, pserver_program,
887887
# create table optimize block in pserver program
888888
table_opt_op = [
889889
op for op in self.optimize_ops
890-
if op.input("Param")[0] == self.table_name
890+
if 'Param' in op.input_names and op.input("Param")[0] ==
891+
self.table_name
891892
][0]
892893
table_opt_block = pserver_program.create_block(pre_block_idx)
893894
# only support sgd now

0 commit comments

Comments
 (0)