fix tests

Author: Roger-luo

src/kirin/rewrite/alias.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.dialects.py import stmts
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass

src/kirin/rewrite/apply_type.py

@@ -8,7 +8,7 @@
     CallableStmtInterface,
     types,
 )
-from kirin.rewrite import RewriteRule, RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 from kirin.dialects.func import Signature
 
 

src/kirin/rewrite/call2invoke.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.analysis import const
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 from kirin.dialects.func import Call, Invoke
 
 

src/kirin/rewrite/cfg_compactify.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.dialects import cf
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 from kirin.analysis.cfg import CFG
 
 

src/kirin/rewrite/cse.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 
 from kirin.ir import Pure, Block, Statement
-from kirin.rewrite import RewriteRule, RewriteResult
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass

src/kirin/rewrite/dce.py

@@ -1,9 +1,9 @@
 from dataclasses import field, dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.analysis import const
 from kirin.dialects import func
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass

src/kirin/rewrite/fold.py

@@ -1,10 +1,10 @@
 from dataclasses import field, dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.analysis import const
 from kirin.dialects import cf, func
 from kirin.dialects.py import stmts
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass

src/kirin/rewrite/getfield.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.dialects import func
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass

src/kirin/rewrite/getitem.py

@@ -1,9 +1,9 @@
 from dataclasses import dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.analysis import const
 from kirin.dialects.py import stmts
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 
 @dataclass

src/kirin/rewrite/inline.py

@@ -2,11 +2,11 @@
 from dataclasses import dataclass
 
 from kirin import ir
-from kirin.rewrite import RewriteRule, RewriteResult
 from kirin.dialects import cf, func
 
 # TODO: use func.Constant instead of kirin.dialects.py.stmts.Constant
 from kirin.dialects.py import stmts
+from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 # NOTE: this only inlines func dialect