fix: add FromPythonCall to examples

Author: Roger-luo

example/beer/interp.py

@@ -8,7 +8,7 @@
 
 
 @dialect.register
-class BeerInterpreter(MethodTable):
+class BeerMethods(MethodTable):
 
     @impl(NewBeer)
     def new_beer(self, interp: Interpreter, frame: Frame, stmt: NewBeer):

example/beer/script.py

@@ -1,7 +1,7 @@
 from stmts import Pour, Puke, Drink, NewBeer
 from dialect import dialect
 
-from interp import BeerInterpreter as BeerInterpreter
+from interp import BeerMethods as BeerMethods
 from rewrite import RandomWalkBranch
 from kirin.ir import dialect_group
 from kirin.prelude import basic_no_opt
@@ -24,10 +24,15 @@ def run_pass(mt):
 # type: ignore
 @beer
 def main(x):
+    def some_closure(beer, amount):
+        Pour(beer, amount + 1)
+        Puke()
+
     beer = NewBeer("budlight")
     Drink(beer)
-    Pour(beer, 12)
+    Pour(beer, 12 + x)
     Puke()
+    some_closure(beer, 1 + 1)
     if x > 1:
         Drink(NewBeer("heineken"))
     else:

example/beer/stmts.py

@@ -8,20 +8,22 @@
 @statement(dialect=dialect)
 class NewBeer(ir.Statement):
     name = "new_beer"
-    traits = frozenset({ir.Pure()})
+    traits = frozenset({ir.Pure(), ir.FromPythonCall()})
     brand: ir.SSAValue = info.argument(types.String)
     result: ir.ResultValue = info.result(types.PyClass(Beer))
 
 
 @statement(dialect=dialect)
 class Drink(ir.Statement):
     name = "drink"
+    traits = frozenset({ir.FromPythonCall()})
     beverage: ir.SSAValue = info.argument(types.PyClass(Beer))
 
 
 @statement(dialect=dialect)
 class Pour(ir.Statement):
     name = "pour"
+    traits = frozenset({ir.FromPythonCall()})
     beverage: ir.SSAValue = info.argument(types.PyClass(Beer))
     amount: ir.SSAValue = info.argument(types.Int)
 
@@ -40,3 +42,4 @@ class RandomBranch(ir.Statement):
 @statement(dialect=dialect)
 class Puke(ir.Statement):
     name = "puke"
+    traits = frozenset({ir.FromPythonCall()})

example/simple.py

@@ -0,0 +1,34 @@
+"""A minimal language example with a single pass that does nothing.
+"""
+
+from kirin import ir
+from kirin.dialects import cf, py, func
+
+
+@ir.dialect_group([func, py.base, py.constant, py.assign, py.binop, py.unary])
+def simple(self):
+    def run_pass(mt):
+        return mt
+
+    return run_pass
+
+
+@simple
+def main(x):
+    y = x + 1
+    return y
+
+
+main.print()
+
+
+@simple.add(cf).add(py.cmp)
+def main2(x):
+    y = x + 1
+    if y > 0:  # errors
+        return y
+    else:
+        return -y
+
+
+main2.print()

src/kirin/ir/nodes/block.py

@@ -13,8 +13,8 @@
 from kirin.ir.nodes.view import View, MutableSequenceView
 
 if TYPE_CHECKING:
-    from kirin.ir.types import TypeAttribute
     from kirin.ir.nodes.stmt import Statement
+    from kirin.ir.attrs.types import TypeAttribute
     from kirin.ir.nodes.region import Region