Function func

tests/python/tirx-base/test_tir_base.py:107–110

(a: T.float32, b: T.float32)

def test_control_flow_jump():
    @T.prim_func(s_tir=True)
    def func(a: T.float32, b: T.float32):
        if True:
            T.evaluate(T.ret(a))
        T.evaluate(T.ret(b))

    func = build_tir_func(func)
    out = func(1.0, 2.0)