Adjust AggressiveUnroll to fix type error during unroll (#593)

I'm not sure what was causing the error but rearranging the passes with
the help of @weinbe58 fixed the problem.

For context, this was the error.  I did not find a MWE.
```python
Traceback (most recent call last):
  File "my_script.py", line 49, in compile_to_stim
    PhysicalAndSquinToStim(main.dialects, no_raise=no_raise)(main)
  File "site-packages/kirin/passes/abc.py", line 30, in __call__
    result = self.unsafe_run(mt)
  File "squin_and_to_stim.py", line 99, in unsafe_run
    .unsafe_run(mt)
  File "site-packages/bloqade/rewrite/passes/canonicalize_ilist.py", line 29, in unsafe_run
    ).rewrite(mt.code)
  File "site-packages/kirin/rewrite/fixpoint.py", line 24, in rewrite
    result = self.rule.rewrite(node)
  File "site-packages/kirin/rewrite/walk.py", line 40, in rewrite
    result = self.rule.rewrite(subnode)
  File "site-packages/kirin/rewrite/chain.py", line 29, in rewrite
    result = rule.rewrite(node)
  File "site-packages/kirin/rewrite/abc.py", line 38, in rewrite
    return self.rewrite_Statement(cast(Statement, node))
  File "site-packages/kirin/dialects/ilist/rewrite/flatten_add.py", line 49, in rewrite_Statement
    assert isinstance(rhs_type := rhs.type, types.Generic), "Impossible"
AssertionError: Impossible
```

---------

Co-authored-by: Phillip Weinberg <[email protected]>

Author: cduck

src/bloqade/rewrite/passes/aggressive_unroll.py

@@ -14,13 +14,15 @@
     InlineGetItem,
     InlineGetField,
     DeadCodeElimination,
+    CommonSubexpressionElimination,
 )
 from kirin.dialects import scf, ilist
 from kirin.ir.method import Method
 from kirin.rewrite.abc import RewriteResult
-from kirin.rewrite.cse import CommonSubexpressionElimination
 from kirin.passes.aggressive import UnrollScf
 
+from .canonicalize_ilist import CanonicalizeIList
+
 
 @dataclass
 class Fold(Pass):
@@ -55,30 +57,36 @@ class AggressiveUnroll(Pass):
     fold: Fold = field(init=False)
     typeinfer: TypeInfer = field(init=False)
     scf_unroll: UnrollScf = field(init=False)
+    canonicalize_ilist: CanonicalizeIList = field(init=False)
 
     def __post_init__(self):
         self.fold = Fold(self.dialects, no_raise=self.no_raise)
         self.typeinfer = TypeInfer(self.dialects, no_raise=self.no_raise)
         self.scf_unroll = UnrollScf(self.dialects, no_raise=self.no_raise)
+        self.canonicalize_ilist = CanonicalizeIList(
+            self.dialects, no_raise=self.no_raise
+        )
 
     def unsafe_run(self, mt: Method) -> RewriteResult:
         result = RewriteResult()
+        result = self.fold.unsafe_run(mt).join(result)
         result = self.scf_unroll.unsafe_run(mt).join(result)
+        self.typeinfer.unsafe_run(
+            mt
+        )  # Do not join the result of typeinfer or fixpoint will waste time
         result = (
             Walk(Chain(ilist.rewrite.ConstList2IList(), ilist.rewrite.Unroll()))
             .rewrite(mt.code)
             .join(result)
         )
-        self.typeinfer.unsafe_run(mt)
-        result = self.fold.unsafe_run(mt).join(result)
         result = Walk(Inline(self.inline_heuristic)).rewrite(mt.code).join(result)
         result = Walk(Fixpoint(CFGCompactify())).rewrite(mt.code).join(result)
-
+        result = self.canonicalize_ilist.fixpoint(mt).join(result)
         rule = Chain(
             CommonSubexpressionElimination(),
             DeadCodeElimination(),
         )
-        result = Fixpoint(Walk(rule)).rewrite(mt.code).join(result)
+        result = Walk(rule).rewrite(mt.code).join(result)
 
         return result