Function test_normalize_tile_layout

tests/python/tirx/test_layout.py:273–526

()

def test_normalize_tile_layout():
    def case1():
        layout = TileLayout(S[(8, 8, 8, 4, 2) : (512, 64, 8, 2, 1)])
        layout_expected = TileLayout(S[4096:1])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case1()

    def case2():
        layout = TileLayout(S[(8, 8, 1, 8, 4, 2) : (512, 64, 160, 8, 2, 1)])
        layout_expected = TileLayout(S[4096:1])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case2()

    def case3():
        layout = TileLayout(S[(8, 8, 8, 4, 1, 1) : (512, 64, 8, 2, 1, 1)])
        layout_expected = TileLayout(S[2048:2])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case3()

    def case4():
        layout = TileLayout(S[(8, 8, 1, 1, 1, 4, 1, 1) : (512, 64, 1, 1, 1, 2, 1, 1)])
        layout_expected = TileLayout(S[(64, 4) : (64, 2)])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case4()

    def case5():
        layout = TileLayout(S[(2, 3, 6) : (18, 6, 1)])
        layout_expected = TileLayout(S[36:1])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case5()

    def case6():
        layout = TileLayout(S[(8, 2, 3, 6) : (6, 18, 6, 1)])
        layout_expected = TileLayout(S[(8, 36) : (6, 1)])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case6()

    def case7():
        layout = TileLayout(S[(8, 2, 3, 6) : (6, 24, 6, 1)])
        layout_expected = TileLayout(S[(8, 2, 18) : (6, 24, 1)])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case7()

    def case8():
        layout = TileLayout(S[(8, 2, 4, 2, 3, 6) : (2, 1, 4, 24, 6, 1)])
        layout_expected = TileLayout(S[(16, 4, 2, 18) : (1, 4, 24, 1)])
        assert_structural_equal(layout_expected, layout.canonicalize())

    case8()

    def case9():