diff --git a/src/bloqade/rewrite/passes/aggressive_unroll.py b/src/bloqade/rewrite/passes/aggressive_unroll.py index eadeb56a..abb9f925 100644 --- a/src/bloqade/rewrite/passes/aggressive_unroll.py +++ b/src/bloqade/rewrite/passes/aggressive_unroll.py @@ -71,6 +71,9 @@ def __post_init__(self): def unsafe_run(self, mt: Method) -> RewriteResult: result = RewriteResult() result = self.fold.unsafe_run(mt).join(result) + result = ( + Walk(ilist.rewrite.Unroll()).rewrite(mt.code).join(result) + ) # BUG: cduck's use case fails without this line result = self.scf_unroll.unsafe_run(mt).join(result) self.typeinfer.unsafe_run( mt