cleanup rewrite exports (#372)

fix #364

Author: Roger-luo

src/kirin/dialects/ilist/passes.py

@@ -1,7 +1,7 @@
 from kirin import ir, types
 from kirin.rewrite import Walk, Chain, Fixpoint
 from kirin.passes.abc import Pass
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteResult
 from kirin.dialects.ilist.rewrite import List2IList, ConstList2IList
 
 

src/kirin/dialects/ilist/rewrite/const.py

@@ -1,7 +1,6 @@
 from kirin import ir, types
 from kirin.analysis import const
-from kirin.rewrite.abc import RewriteRule
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 from kirin.dialects.py.constant import Constant
 
 from ..stmts import IListType

src/kirin/dialects/ilist/rewrite/list.py

@@ -1,7 +1,6 @@
 from kirin import ir, types
 from kirin.dialects.py import constant
-from kirin.rewrite.abc import RewriteRule
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 from kirin.dialects.ilist.stmts import IListType
 from kirin.dialects.ilist.runtime import IList
 

src/kirin/dialects/ilist/rewrite/unroll.py

@@ -1,6 +1,5 @@
 from kirin import ir, types
-from kirin.rewrite.abc import RewriteRule
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 from kirin.dialects.py.tuple import New as TupleNew
 from kirin.dialects.func.stmts import Call
 from kirin.dialects.ilist.stmts import Map, New, Scan, Foldl, Foldr, ForEach, IListType

src/kirin/dialects/scf/trim.py

@@ -1,6 +1,5 @@
 from kirin import ir
-from kirin.rewrite.abc import RewriteRule
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 from .stmts import For, Yield, IfElse
 from ._dialect import dialect

src/kirin/dialects/scf/unroll.py

@@ -1,8 +1,7 @@
 from kirin import ir
 from kirin.analysis import const
 from kirin.dialects import func
-from kirin.rewrite.abc import RewriteRule
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 from kirin.dialects.py.constant import Constant
 
 from .stmts import For, Yield, IfElse

src/kirin/rewrite/abc.py

@@ -1,8 +1,21 @@
 from abc import ABC
-from dataclasses import dataclass
+from dataclasses import field, dataclass
 
 from kirin.ir import Pure, Block, IRNode, Region, MaybePure, Statement
-from kirin.rewrite.result import RewriteResult
+
+
+@dataclass
+class RewriteResult:
+    terminated: bool = field(default=False, kw_only=True)
+    has_done_something: bool = field(default=False, kw_only=True)
+    exceeded_max_iter: bool = field(default=False, kw_only=True)
+
+    def join(self, other: "RewriteResult") -> "RewriteResult":
+        return RewriteResult(
+            terminated=self.terminated or other.terminated,
+            has_done_something=self.has_done_something or other.has_done_something,
+            exceeded_max_iter=self.exceeded_max_iter or other.exceeded_max_iter,
+        )
 
 
 @dataclass(repr=False)

src/kirin/rewrite/chain.py

@@ -2,8 +2,7 @@
 from dataclasses import dataclass
 
 from kirin.ir import IRNode
-from kirin.rewrite.abc import RewriteRule
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass

src/kirin/rewrite/fixpoint.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass
 
 from kirin.ir import IRNode
-from kirin.rewrite.abc import RewriteRule
-from kirin.rewrite.result import RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass