def METHOD_NAME(
    M: size,
    K: size,
    N: size,
    src1: [i8][M, 4 * K] @ AMX_TILE,
    src2: [i8][K, 4 * N] @ AMX_TILE,
    dst: [i32][M, N] @ AMX_TILE,
):
    assert M <= 16
    assert K <= 16
    assert M <= 16
    for m in seq(0, M):
        for n in seq(0, N):
            for k in seq(0, K):
                for byte in seq(0, 4):
                    a: i32
                    b: i32
                    a = src1[m, 4 * k + byte]
                    b = src2[k, 4 * n + byte]
                    dst[m, n] += a * b