Function func

tests/python/relax/test_utils.py:188–201

(n: R.Prim("int64"))

    def define_function():
        @R.function
        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)

        return func