fix a_op update

xdslproject · Nov 13, 2023 · dc59a07 · dc59a07
1 parent 1eb103f
commit dc59a07
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xdsl/backend/riscv/lowering/convert_snitch_stream_to_snitch.py b/xdsl/backend/riscv/lowering/convert_snitch_stream_to_snitch.py
@@ -96,6 +96,7 @@ def match_and_rewrite(
             )
             set_stride_op = snitch.SsrSetDimensionStrideOp(stride_op.rd, op.dm, ints[1])
             new_ops.extend((a_inc_op, new_a_op, stride_op, set_stride_op))
+            a_op = new_a_op
 
         rewriter.insert_op_before_matched_op(new_ops)
         rewriter.erase_matched_op()