Function func

tests/python/codegen/test_target_codegen_cuda.py:901–903

(A: T.Buffer((4,), "uint32"), B: T.Buffer((4,), "uint8"))

def test_invalid_reinterpret():
    @T.prim_func(s_tir=True)
    def func(A: T.Buffer((4,), "uint32"), B: T.Buffer((4,), "uint8")) -> None:
        for tx in T.thread_binding(4, "threadIdx.x"):
            B[tx] = T.call_intrin("uint8", "tirx.reinterpret", A[tx])

    with pytest.raises(RuntimeError):
        tvm.compile(func, target="cuda")