Function check

tests/python/tirx-base/test_tir_ops.py:34–38

(f, *args)

def test_const_fold():
    def check(f, *args):
        x = f(*[tvm.tirx.const(x, "int32") for x in args])
        y = f(*args)
        if not isinstance(x, tvm.tirx.IntImm) or x.value != int(y):
            raise ValueError(f"check error: {x} vs {y} ")

    tmod = tvm.tirx.truncmod
    check(lambda x, y: x + y, 3, 4)