Implement ilist any and all methods

src/kirin/dialects/ilist/__init__.py

@@ -27,6 +27,8 @@
 from .runtime import IList as IList
 from ._dialect import dialect as dialect
 from ._wrapper import (  # careful this is not the builtin range
+    all as all,
+    any as any,
     map as map,
     scan as scan,
     foldl as foldl,

src/kirin/dialects/ilist/_wrapper.py

@@ -65,3 +65,11 @@ def for_each(
     fn: typing.Callable[[ElemT], typing.Any],
     collection: IList[ElemT, LenT] | list[ElemT],
 ) -> None: ...
+
+
+@lowering.wraps(stmts.Any)
+def any(collection: IList[bool, LenT] | list[bool]) -> bool: ...
+
+
+@lowering.wraps(stmts.All)
+def all(collection: IList[bool, LenT] | list[bool]) -> bool: ...

src/kirin/dialects/ilist/interp.py

@@ -3,7 +3,7 @@
 from kirin.dialects.py.len import Len
 from kirin.dialects.py.binop import Add
 
-from .stmts import Map, New, Push, Scan, Foldl, Foldr, Range, ForEach
+from .stmts import All, Any, Map, New, Push, Scan, Foldl, Foldr, Range, ForEach
 from .runtime import IList
 from ._dialect import dialect
 
@@ -96,3 +96,13 @@ def for_each(self, interp: Interpreter, frame: Frame, stmt: ForEach):
             # NOTE: assume fn has been type checked
             interp.call(fn.code, fn, elem)
         return
+
+    @impl(Any)
+    def any(self, interp: Interpreter, frame: Frame, stmt: Any):
+        coll: IList = frame.get(stmt.collection)
+        return (any(coll),)
+
+    @impl(All)
+    def all(self, interp: Interpreter, frame: Frame, stmt: All):
+        coll: IList = frame.get(stmt.collection)
+        return (all(coll),)

src/kirin/dialects/ilist/stmts.py

@@ -112,3 +112,17 @@ class ForEach(ir.Statement):
     purity: bool = info.attribute(default=False)
     fn: ir.SSAValue = info.argument(types.Generic(ir.Method, [ElemT], types.NoneType))
     collection: ir.SSAValue = info.argument(IListType[ElemT])
+
+
+@statement(dialect=dialect)
+class Any(ir.Statement):
+    traits = frozenset({ir.Pure(), lowering.FromPythonCall()})
+    collection: ir.SSAValue = info.argument(IListType[types.Bool, ListLen])
+    result: ir.ResultValue = info.result(types.Bool)
+
+
+@statement(dialect=dialect)
+class All(ir.Statement):
+    traits = frozenset({ir.Pure(), lowering.FromPythonCall()})
+    collection: ir.SSAValue = info.argument(IListType[types.Bool, ListLen])
+    result: ir.ResultValue = info.result(types.Bool)

test/dialects/test_ilist_wrapper.py

@@ -75,3 +75,38 @@ def scan_wrap():
         10 + 1 + 1 + 1 + 3,
         10 + 1 + 1 + 1 + 1 + 4,
     ]
+
+
+def test_any_all_wrapper():
+
+    @basic
+    def test_any_all():
+        ls = [True, False, False]
+        return ls, ilist.any(ls), ilist.all(ls)
+
+    test_any_all.print()
+
+    ls, any_val, all_val = test_any_all()
+
+    assert isinstance(ls, ilist.IList)
+    assert ls.data == [True, False, False]
+    assert any_val
+    assert not all_val
+
+    @basic
+    def test_any_all2():
+        ls = [False, False]
+        return ilist.any(ls), ilist.all(ls)
+
+    any_val, all_val = test_any_all2()
+    assert not any_val
+    assert not all_val
+
+    @basic
+    def test_any_all3():
+        ls = [True, True, True, True, True]
+        return ilist.any(ls), ilist.all(ls)
+
+    any_val, all_val = test_any_all3()
+    assert any_val
+    assert all_val