Function func

tests/python/codegen/test_target_codegen_vulkan.py:352–357

(A: T.Buffer((N, 2), "int32"))

    @T.prim_func(s_tir=True)
    def func(A: T.Buffer((N, 2), "int32")):
        for i in T.thread_binding(N, thread="threadIdx.x"):
            with T.sblock("A"):
                v_i = T.axis.spatial(N, i)
                A[v_i, 0] = T.floordiv(v_i - offset, divisor)
                A[v_i, 1] = T.floormod(v_i - offset, divisor)

    built = tvm.compile(func, target=target)