Method arange

tests/python/runtime/test_runtime_extension.py:28–32

(A: T.handle)

    class Module:
        @T.prim_func(s_tir=True)
        def arange(A: T.handle):
            n = T.int32()
            Ab = T.match_buffer(A, (n,), "int64")
            for i in T.serial(n - 1):
                Ab[i + 1] = Ab[i] + T.int64(1)

    mod = Module
    f = tvm.compile(mod, target="llvm")