Function get_ind_matrix

cutlass.py/swizzle.py:32–33

(rows, cols)

def test_swizzle():

    def get_ind_matrix(rows, cols):
        return [[(x, y) for y in range(cols)] for x in range(rows)]

    def get_row_major_ind(x, y, rows, cols):
        return x * cols + y