未验证 提交 b73806b2 编写于 作者: Q Qiao Longfei 提交者: GitHub

Merge pull request #12347 from jacquesqiao/fix-distribute_transpiler_for_dist_lookup_table

fix _create_prefetch_block in distribute_transpiler
...@@ -887,7 +887,8 @@ class DistributeTranspiler(object): ...@@ -887,7 +887,8 @@ class DistributeTranspiler(object):
# create table optimize block in pserver program # create table optimize block in pserver program
table_opt_op = [ table_opt_op = [
op for op in self.optimize_ops 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] ][0]
table_opt_block = pserver_program.create_block(pre_block_idx) table_opt_block = pserver_program.create_block(pre_block_idx)
# only support sgd now # only support sgd now
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册