Function verify_slice

tests/python/tirx/test_layout.py:1549–1570

(layout, shape, region, sliced)

def test_slice():
    def verify_slice(layout, shape, region, sliced):
        r_shape = [r[1] - r[0] for r in region]
        r_size = functools.reduce(operator.mul, [r[1] - r[0] for r in region])

        def get_region_coord(u):
            coord = []
            for r in reversed(region):
                coord.append(u % (r[1] - r[0]))
                u //= r[1] - r[0]
            return coord[::-1]

        def get_shape_coord(r_coord, region):
            return [region[i][0] + r_coord[i] for i in range(len(region))]

        analyzer = Analyzer()

        for u in range(r_size):
            r_coord = get_region_coord(u)
            s_coord = get_shape_coord(r_coord, region)
            a = layout.apply(*s_coord, shape=shape)["m"]
            b = sliced.apply(*r_coord, shape=r_shape)["m"]
            assert analyzer.simplify(a == b)

    def case1():
        layout = TileLayout(S[(8, 8) : (8, 1)])