Function test_tile_layout

tests/python/tirx/test_layout.py:529–869

()

def test_tile_layout():
    def case1():
        # (8):(1)x(8):(1) -> (64):(1)
        inner = TileLayout(S[8:1])
        outer = inner
        layout_tile = TileLayout(S[64:1])
        assert_structural_equal(layout_tile, inner.tile(outer, [8], [8]))

        outer_res = inner.is_tile_inner(layout_tile, [64], [8])
        assert outer_res is not None
        assert_structural_equal(outer_res.canonicalize(), outer.canonicalize())

        inner_res = outer.is_tile_outer(layout_tile, [64], [8])
        assert inner_res is not None
        assert_structural_equal(inner_res.canonicalize(), inner.canonicalize())

    case1()

    def case2():
        # (8,8):(8,1)x(8,8):(8,1) -> (8,8,8,8):(512,8,64,1)
        inner = TileLayout(S[(8, 8) : (8, 1)])
        outer = inner
        layout_tile = TileLayout(S[(8, 8, 8, 8) : (512, 8, 64, 1)])
        assert_structural_equal(layout_tile, inner.tile(outer, [8, 8], [8, 8]))

        outer_res = inner.is_tile_inner(layout_tile, [64, 64], [8, 8])
        assert outer_res is not None
        assert_structural_equal(outer_res.canonicalize(), outer.canonicalize())

        inner_res = outer.is_tile_outer(layout_tile, [64, 64], [8, 8])
        assert inner_res is not None
        assert_structural_equal(inner_res.canonicalize(), inner.canonicalize())

    case2()

    def case3():
        # (2,4):(1,2)x(8,8):(8,1) -> (8,2,8,4):(64,1,8,2)
        inner = TileLayout(S[(2, 4) : (1, 2)])
        outer = TileLayout(S[(8, 8) : (8, 1)])
        layout_tile = TileLayout(S[(8, 2, 32) : (64, 1, 2)])
        assert_structural_equal(layout_tile, inner.tile(outer, [8, 8], [2, 4]))

        outer_res = inner.is_tile_inner(layout_tile, [16, 32], [2, 4])
        assert outer_res is not None
        assert_structural_equal(outer_res.canonicalize(), outer.canonicalize())

        inner_res = outer.is_tile_outer(layout_tile, [16, 32], [8, 8])
        assert inner_res is not None
        assert_structural_equal(inner_res.canonicalize(), inner.canonicalize())

        assert outer.is_tile_inner(layout_tile, [16, 32], [8, 8]) is None
        assert inner.is_tile_outer(layout_tile, [16, 32], [2, 4]) is None

    case3()

    def case4():
        # ((4,2),(2,4)):((16,8),(1,2))x(8,8):(8,1) -> (8,4,2,8,2,4):(512,16,8,64,1,2)
        inner = TileLayout(S[(4, 2, 2, 4) : (16, 8, 1, 2)])