Skip to content

Commit d0de2d8

Browse files
author
lilong12
authored
fix the bug in processing subblock in pipeline (#32727)
1 parent a0f4ac5 commit d0de2d8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/paddle/fluid/optimizer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4604,13 +4604,13 @@ def _add_sub_blocks(self, main_block, program_list):
46044604
origin_sub_block_id = op.attr('sub_block').id
46054605
origin_sub_block = main_program.block(origin_sub_block_id)
46064606
new_sub_block = prog._create_block(parent_idx=0)
4607-
for op in origin_sub_block.ops:
4608-
op_desc = op.desc
4607+
for sub_op in origin_sub_block.ops:
4608+
op_desc = sub_op.desc
46094609
ap_op = new_sub_block.desc.append_op()
46104610
ap_op.copy_from(op_desc)
46114611
new_sub_block._sync_with_cpp()
46124612
self._create_vars(new_sub_block, origin_sub_block)
4613-
op._set_attr('sub_block:', new_sub_block)
4613+
op._set_attr('sub_block', new_sub_block)
46144614

46154615
def _get_device_info(self, block):
46164616
for op in block.ops:

0 commit comments

Comments
 (0)