提交 f3bbd3b4 编写于 作者: Q Qiao Longfei

code style format

test=develop
上级 2d461cb0
...@@ -1065,10 +1065,12 @@ to transpile() call.") ...@@ -1065,10 +1065,12 @@ to transpile() call.")
continue_search_lookup_table_op = False continue_search_lookup_table_op = False
all_ops = program.global_block().ops all_ops = program.global_block().ops
for op in all_ops: for op in all_ops:
if op.type == LOOKUP_TABLE_TYPE and self.table_name == op.input("W")[0]: if op.type == LOOKUP_TABLE_TYPE and self.table_name == op.input(
"W")[0]:
if not op.attr('is_distributed'): if not op.attr('is_distributed'):
raise RuntimeError("lookup_table_op that lookup an distributed embedding table" raise RuntimeError(
"should set is_distributed to true") "lookup_table_op that lookup an distributed embedding table"
"should set is_distributed to true")
continue_search_lookup_table_op = True continue_search_lookup_table_op = True
lookup_table_op_index = lookup_table_op_index if lookup_table_op_index != -1 else list( lookup_table_op_index = lookup_table_op_index if lookup_table_op_index != -1 else list(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册