Skip to content

Commit fa84ba2

Browse files
committed
set en empty optimize block if pserver has no optimize block
1 parent 70ce6dc commit fa84ba2

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -767,6 +767,13 @@ def __clone_lr_op_sub_block__(op, program, lr_block):
767767
prefetch_var_name_to_block_id.extend(
768768
lookup_table_var_name_to_block_id)
769769

770+
if optimize_blocks.size() == 0:
771+
pre_block_idx = pserver_program.num_blocks - 1
772+
empty_block = pserver_program._create_block(pre_block_idx)
773+
optimize_blocks.append(empty_block)
774+
775+
# In some case, some parameter server will have no parameter to optimize
776+
# So we give an empty optimize block to parameter server.
770777
attrs = {
771778
"optimize_blocks": optimize_blocks,
772779
"endpoint": endpoint,

0 commit comments

Comments
 (0)