Function recursive_lambda

tests/python/relax/test_utils.py:190–199

(i_arg: R.Prim(value="i"))

        def func(n: R.Prim("int64")):
            @R.function
            def recursive_lambda(i_arg: R.Prim(value="i")) -> R.Prim("int64"):
                i = T.int64()
                if R.prim_value(i == 0):
                    output = R.prim_value(T.int64(0))
                else:
                    remainder_relax = recursive_lambda(R.prim_value(i - 1))
                    remainder_tir = T.int64()
                    _ = R.match_cast(remainder_relax, R.Prim(value=remainder_tir))
                    output = R.prim_value(i + remainder_tir)
                return output

            return recursive_lambda(n)