def METHOD_NAME(A: T.Buffer((128, 128), "float32"), B: T.Buffer((128, 128), "float32")):
    with T.block("root"):
        s = T.int32()
        e = T.int32()
        # A0 should be remapped
        A0 = T.match_buffer(
            A[0:128, 0:128],
            shape=(128, 128),
            dtype="float32",
            # s and e should be remapped
            strides=[s, s],
            elem_offset=e,
        )
        for i, j in T.grid(128, 128):
            with T.block("B"):
                vi, vj = T.axis.remap("SS", [i, j])
                B[vi, vj] = A0[vi, vj] * 2.0