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

set en empty optimize block if pserver has no optimize block

上级 70ce6dcd
...@@ -767,6 +767,13 @@ in a single call.") ...@@ -767,6 +767,13 @@ in a single call.")
prefetch_var_name_to_block_id.extend( prefetch_var_name_to_block_id.extend(
lookup_table_var_name_to_block_id) lookup_table_var_name_to_block_id)
if optimize_blocks.size() == 0:
pre_block_idx = pserver_program.num_blocks - 1
empty_block = pserver_program._create_block(pre_block_idx)
optimize_blocks.append(empty_block)
# In some case, some parameter server will have no parameter to optimize
# So we give an empty optimize block to parameter server.
attrs = { attrs = {
"optimize_blocks": optimize_blocks, "optimize_blocks": optimize_blocks,
"endpoint": endpoint, "endpoint": endpoint,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册