triton.language.swizzle2d

triton.language.swizzle2d(i, j, size_i, size_j, size_g)

将行优先size_i * size_j矩阵的索引转换为每组size_g行的列优先矩阵索引。

例如，当size_i = size_j = 4且size_g = 2时，它将进行转换

[[0 , 1 , 2 , 3 ],
 [4 , 5 , 6 , 7 ],
 [8 , 9 , 10, 11],
 [12, 13, 14, 15]]

进入

[[0, 2,  4 , 6 ],
 [1, 3,  5 , 7 ],
 [8, 10, 12, 14],
 [9, 11, 13, 15]]