fix aggressive rules (#123)

`ConstFold` should `Walk`

Author: Roger-luo

src/kirin/ir/dialect.py

@@ -34,25 +34,7 @@ def __post_init__(self) -> None:
         self.lowering["default"] = NoSpecialLowering()
 
     def __repr__(self) -> str:
-        stmts = ", ".join([stmt.__name__ for stmt in self.stmts])
-        attrs = ", ".join([attr.__name__ for attr in self.attrs])
-        interps = ", ".join(
-            [f"{key} = {type(interp).__name__}" for key, interp in self.interps.items()]
-        )
-        lowering = ", ".join(
-            [f"{key} = {type(lower).__name__}" for key, lower in self.lowering.items()]
-        )
-        codegen = ", ".join(
-            [f"{key} = {type(emit).__name__}" for key, emit in self.codegen.items()]
-        )
-        return f"""Dialect(\
-name={self.name},\
-stmts=[{stmts}], \
-attrs=[{attrs}], \
-interps=[{interps}], \
-lowering=[{lowering}]\
-codegen=[{codegen}]\
-)"""
+        return f"Dialect(name={self.name}, ...)"
 
     def __hash__(self) -> int:
         return hash(self.name)

src/kirin/rules/aggressive/fold.py

@@ -22,8 +22,8 @@ def __init__(self, cfg: analysis.CFG, results: dict[ir.SSAValue, Any]):
             Chain(
                 [
                     Walk(Inline(lambda _: True)),
-                    ConstantFold(results),
-                    Call2Invoke(results),
+                    Walk(ConstantFold(results)),
+                    Walk(Call2Invoke(results)),
                     Fixpoint(
                         Walk(
                             Chain(

test/program/py/test_aggressive.py

@@ -32,13 +32,13 @@ def main():
     return x()
 
 
-@basic.add(dialect)
-def target():
-    x = DummyStmt2(1, option="hello")
-    DummyStmt2(x, option="hello")
-    DummyStmt2(x, option="hello")
-    return
-
-
 def test_aggressive_pass():
-    assert target.callable_region.is_structurally_equal(main.callable_region)
+    const_count = 0
+    dummy_count = 0
+    for stmt in main.callable_region.walk():
+        if isinstance(stmt, DummyStmt2):
+            dummy_count += 1
+        elif stmt.has_trait(ir.ConstantLike):
+            const_count += 1
+    assert dummy_count == 3
+    assert const_count == 2