From 6f0b7cbca003c6c24ae7c892d1814c8ba39c2c1d Mon Sep 17 00:00:00 2001 From: Casey Duckering Date: Mon, 10 Nov 2025 22:37:17 -0500 Subject: [PATCH 1/2] Add pass to unroll, unsure why this works --- src/bloqade/rewrite/passes/aggressive_unroll.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/bloqade/rewrite/passes/aggressive_unroll.py b/src/bloqade/rewrite/passes/aggressive_unroll.py index eadeb56a..0708514f 100644 --- a/src/bloqade/rewrite/passes/aggressive_unroll.py +++ b/src/bloqade/rewrite/passes/aggressive_unroll.py @@ -71,6 +71,7 @@ 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 From 9d93b05f1aa2d99383be36c9179b0e995a0e16a3 Mon Sep 17 00:00:00 2001 From: Casey Duckering Date: Mon, 10 Nov 2025 23:03:27 -0500 Subject: [PATCH 2/2] Black --- src/bloqade/rewrite/passes/aggressive_unroll.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/bloqade/rewrite/passes/aggressive_unroll.py b/src/bloqade/rewrite/passes/aggressive_unroll.py index 0708514f..abb9f925 100644 --- a/src/bloqade/rewrite/passes/aggressive_unroll.py +++ b/src/bloqade/rewrite/passes/aggressive_unroll.py @@ -71,7 +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 = ( + 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