From fa84ba23507f3211f48ca142619c83843cf2f106 Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Tue, 30 Oct 2018 19:29:17 +0800 Subject: [PATCH] set en empty optimize block if pserver has no optimize block --- python/paddle/fluid/transpiler/distribute_transpiler.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 8daac0f43b..b71bd48baf 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -767,6 +767,13 @@ in a single call.") prefetch_var_name_to_block_id.extend( 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 = { "optimize_blocks": optimize_blocks, "endpoint": endpoint, -- GitLab