Merge pull request #12445 from jacquesqiao/fix-dis-trans

Fix dist trans

Author: jacquesqiao

python/paddle/fluid/transpiler/distribute_transpiler.py

@@ -495,6 +495,7 @@ def __clone_lr_op_sub_block__(op, program, lr_block):
             pserver_index = self.pserver_endpoints.index(endpoint)
             table_opt_block = self._create_table_optimize_block(
                 pserver_index, pserver_program, pre_block_idx, grad_to_block_id)
+            optimize_blocks.append(table_opt_block)
             prefetch_var_name_to_block_id = self._create_prefetch_block(
                 pserver_index, pserver_program, table_opt_block)
             checkpoint_block_id = self._create_checkpoint_save_block(