further simplify interpreter and use a more clarified name (#118)

This PR removes the fallback attached to each method table for
associated dialects but instead the fallback is implemented inside the
corresponding interpreter. This is because I don't see any use case that
different dialect will use different fallback, and in fact no dialect is
using this feature right now.

Author: Roger-luo

example/beer/interp.py

@@ -4,11 +4,11 @@
 from stmts import Pour, Puke, Drink, NewBeer, RandomBranch
 from dialect import dialect
 
-from kirin.interp import Successor, Interpreter, DialectInterpreter, impl
+from kirin.interp import Successor, Interpreter, MethodTable, impl
 
 
 @dialect.register
-class BeerInterpreter(DialectInterpreter):
+class BeerInterpreter(MethodTable):
 
     @impl(NewBeer)
     def new_beer(self, interp: Interpreter, stmt: NewBeer, values: tuple):

example/beer/script.py

@@ -1,11 +1,11 @@
+from stmts import Pour, Puke, Drink, NewBeer
 from dialect import dialect
-from stmts import Drink, NewBeer, Pour, Puke
 
 from interp import BeerInterpreter as BeerInterpreter
+from rewrite import RandomWalkBranch
 from kirin.ir import dialect_group
 from kirin.prelude import basic_no_opt
-from kirin.rewrite import Fixpoint, Walk
-from rewrite import RandomWalkBranch
+from kirin.rewrite import Walk, Fixpoint
 
 
 # create our own beer dialect, it runs a random walk on the branches
@@ -38,6 +38,6 @@ def main(x):
 main.code.print()
 main(1)  # execute the function
 
-# for i in range(10):
-#     print("iteration", i)
-#     main(i)  # now drink a random beer!
+for i in range(10):
+    print("iteration", i)
+    main(i)  # now drink a random beer!

src/kirin/analysis/typeinfer.py

@@ -1,4 +1,4 @@
-from kirin import ir
+from kirin import ir, interp
 from kirin.ir import types
 from kirin.ir.method import Method
 from kirin.ir.nodes.stmt import Statement
@@ -7,7 +7,7 @@
 
 
 class TypeInference(Forward[types.TypeAttribute]):
-    keys = ["typeinfer", "typeinfer.default"]
+    keys = ["typeinfer", "empty"]
     lattice = types.TypeAttribute
 
     def build_signature(self, stmt: Statement, args: tuple):
@@ -26,6 +26,14 @@ def build_signature(self, stmt: Statement, args: tuple):
             tuple(_args),
         )
 
+    def eval_stmt(
+        self, stmt: Statement, args: tuple[types.TypeAttribute, ...]
+    ) -> interp.Result:
+        method = self.lookup_registry(stmt, args)
+        if method is not None:
+            return method(self, stmt, args)
+        return tuple(result.type for result in stmt.results)
+
     def run_method_region(
         self, mt: Method, body: Region, args: tuple[types.TypeAttribute, ...]
     ) -> types.TypeAttribute:

src/kirin/dialects/cf/constprop.py

@@ -1,11 +1,11 @@
-from kirin.interp import Successor, DialectInterpreter, impl
+from kirin.interp import Successor, MethodTable, impl
 from kirin.analysis import const
 from kirin.dialects.cf.stmts import Assert, Branch, ConditionalBranch
 from kirin.dialects.cf.dialect import dialect
 
 
 @dialect.register(key="constprop")
-class DialectConstProp(DialectInterpreter):
+class DialectConstProp(MethodTable):
 
     @impl(Assert)
     def assert_stmt(self, interp: const.Propagate, stmt: Assert, values):

src/kirin/dialects/cf/interp.py

@@ -1,10 +1,10 @@
-from kirin.interp import Err, Successor, Interpreter, DialectInterpreter, impl
+from kirin.interp import Err, Successor, Interpreter, MethodTable, impl
 from kirin.dialects.cf.stmts import Assert, Branch, ConditionalBranch
 from kirin.dialects.cf.dialect import dialect
 
 
 @dialect.register
-class CfInterpreter(DialectInterpreter):
+class CfInterpreter(MethodTable):
 
     @impl(Assert)
     def assert_stmt(self, interp: Interpreter, stmt: Assert, values):

src/kirin/dialects/cf/typeinfer.py

@@ -1,12 +1,12 @@
 from kirin.ir import types
-from kirin.interp import Successor, DialectInterpreter, impl
+from kirin.interp import Successor, MethodTable, impl
 from kirin.dialects.cf.stmts import Assert, Branch, ConditionalBranch
 from kirin.analysis.typeinfer import TypeInference
 from kirin.dialects.cf.dialect import dialect
 
 
 @dialect.register(key="typeinfer")
-class TypeInfer(DialectInterpreter):
+class TypeInfer(MethodTable):
 
     @impl(Assert)
     def assert_stmt(self, interp: TypeInference, stmt: Assert, values):

src/kirin/dialects/fcf/interp.py

@@ -1,11 +1,11 @@
 from kirin import ir
-from kirin.interp import Interpreter, DialectInterpreter, impl
+from kirin.interp import Interpreter, MethodTable, impl
 from kirin.dialects.fcf.stmts import Map, Scan, Foldl, Foldr
 from kirin.dialects.fcf.dialect import dialect
 
 
 @dialect.register
-class FCFInterpreter(DialectInterpreter):
+class FCFInterpreter(MethodTable):
 
     @impl(Foldl)
     def foldl(self, interp: Interpreter, stmt: Foldl, values: tuple):

src/kirin/dialects/fcf/typeinfer.py

@@ -1,14 +1,14 @@
 from typing import Callable, Iterable
 
 from kirin import ir
-from kirin.interp import DialectInterpreter, impl
+from kirin.interp import MethodTable, impl
 from kirin.analysis.typeinfer import TypeInference
 from kirin.dialects.fcf.stmts import Map, Scan, Foldl, Foldr
 from kirin.dialects.fcf.dialect import dialect
 
 
 @dialect.register(key="typeinfer")
-class TypeInfer(DialectInterpreter):
+class TypeInfer(MethodTable):
 
     @impl(Foldl)
     def foldl(

src/kirin/dialects/func/constprop.py

@@ -1,14 +1,14 @@
 from typing import Iterable
 
 from kirin import ir
-from kirin.interp import Result, ReturnValue, DialectInterpreter, impl
+from kirin.interp import Result, MethodTable, ReturnValue, impl
 from kirin.analysis import const
 from kirin.dialects.func.stmts import Call, Invoke, Lambda, Return, GetField
 from kirin.dialects.func.dialect import dialect
 
 
 @dialect.register(key="constprop")
-class DialectConstProp(DialectInterpreter):
+class DialectConstProp(MethodTable):
 
     @impl(Return)
     def return_(

src/kirin/dialects/func/interp.py

@@ -1,5 +1,5 @@
 from kirin.ir import Method
-from kirin.interp import ReturnValue, DialectInterpreter, impl, concrete
+from kirin.interp import MethodTable, ReturnValue, impl, concrete
 from kirin.dialects.func.stmts import (
     Call,
     Invoke,
@@ -12,7 +12,7 @@
 
 
 @dialect.register
-class Interpreter(DialectInterpreter):
+class Interpreter(MethodTable):
 
     @impl(Call)
     def call(self, interp: concrete.Interpreter, stmt: Call, values: tuple):