optimized the diagonal transpose in N = 64, M = 64