move range implementation to IList (#154)

now range is conditionally enabled by IList.

cc: @kaihsin

Author: Roger-luo

src/kirin/dialects/ilist/interp.py

@@ -2,6 +2,7 @@
 from kirin.interp import Err, Frame, Interpreter, MethodTable, impl
 from kirin.dialects.py.len import Len
 from kirin.dialects.py.binop import Add
+from kirin.dialects.py.range import Range
 
 from .stmts import Map, New, Push, Scan, Foldl, Foldr, ForEach
 from .runtime import IList
@@ -11,6 +12,10 @@
 @dialect.register
 class IListInterpreter(MethodTable):
 
+    @impl(Range)
+    def _range(self, interp, frame: Frame, stmt: Range):
+        return (IList(range(*frame.get_values(stmt.args))),)
+
     @impl(New)
     def new(self, interp, frame: Frame, stmt: New):
         return (IList(list(frame.get_values(stmt.values))),)

src/kirin/dialects/ilist/runtime.py

@@ -1,6 +1,7 @@
 # TODO: replace with something faster
 from typing import Any, Generic, TypeVar, overload
 from dataclasses import dataclass
+from collections.abc import Sequence
 
 T = TypeVar("T")
 L = TypeVar("L")
@@ -10,7 +11,7 @@
 class IList(Generic[T, L]):
     """A simple immutable list."""
 
-    data: list[T]
+    data: Sequence[T]
 
     def __len__(self) -> int:
         return len(self.data)

src/kirin/dialects/py/range.py

@@ -1,7 +1,7 @@
 import ast
 from dataclasses import dataclass
 
-from kirin import ir, types, interp, lowering, exceptions
+from kirin import ir, types, lowering, exceptions
 from kirin.decl import info, statement
 
 dialect = ir.Dialect("py.range")
@@ -35,14 +35,6 @@ def lower_Call_range(
         return _lower_range(state, node)
 
 
-@dialect.register
-class Concrete(interp.MethodTable):
-
-    @interp.impl(Range)
-    def _range(self, interp, frame: interp.Frame, stmt: Range):
-        return (range(*frame.get_values(stmt.args)),)
-
-
 def _lower_range(state: lowering.LoweringState, node: ast.Call) -> lowering.Result:
     if len(node.args) == 1:
         start = state.visit(ast.Constant(0)).expect_one()

test/dialects/test_ilist.py

@@ -145,3 +145,17 @@ def foldr(xs: ilist.IList[int, Literal[3]]):
     assert isinstance(stmt, py.Constant)
     assert stmt.value == 0
     assert isinstance(foldr.callable_region.blocks[0].stmts.at(10), func.Call)
+
+
+def test_ilist_range():
+    @basic.add(py.range)
+    def map():
+        return ilist.Map(add1, range(0, 3))  # type: ignore
+
+    assert map() == ilist.IList([1, 2, 3])
+
+    @basic.add(py.range)
+    def const_range():
+        return range(0, 3)
+
+    assert const_range() == ilist.IList(range(0, 3))