def METHOD_NAME(a: T.handle, b: T.handle, c: T.handle) -> None:
    A = T.match_buffer(a, (4,), dtype=in_dtype, offset_factor=1)
    B = T.match_buffer(b, (4, 4), dtype=in_dtype, offset_factor=1)
    C = T.match_buffer(c, (4,), dtype=out_dtype, offset_factor=1)
    with T.block("root"):
        T.reads(C[0:4], A[0:4], B[0:4, 0:4])
        T.writes(C[0:4])
        for i in T.serial(0, 4):
            for k in T.serial(0, 4):
                with T.block("update"):
                    vi, vk = T.axis.remap("SR", [i, k])
                    C[vi] = C[vi] + T.cast(A[vk], dtype=out_dtype) * T.cast(
                        B[vi, vk], dtype=out_dtype
                    )