Merge pull request #12347 from jacquesqiao/fix-distribute_transpiler_for_dist_lookup_table

fix _create_prefetch_block in distribute_transpiler
distributed_test
Qiao Longfei 7 years ago committed by GitHub
commit b73806b208
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -887,7 +887,8 @@ class DistributeTranspiler(object):
# create table optimize block in pserver program
table_opt_op = [
op for op in self.optimize_ops
if op.input("Param")[0] == self.table_name
if 'Param' in op.input_names and op.input("Param")[0] ==
self.table_name
][0]
table_opt_block = pserver_program.create_block(pre_block_idx)
# only support sgd now

Loading…
Cancel
Save