Function test_compose_layout_1

tests/python/tirx/test_layout.py:1199–1209

()

    test_compose_layout_0()

    def test_compose_layout_1():
        layoutA = SwizzleLayout(per_element=3, swizzle_len=3, atom_len=3)
        layoutB = TileLayout(S[(16, 64, 8) : (64, 1, 1024)])
        layout = ComposeLayout(layoutA, layoutB)
        assert layout.size() == 16 * 64 * 8
        assert layout.span() == 16 * 64 * 8
        for i, j, k in itertools.product(range(16), range(64), range(8)):
            assert (
                layout.apply(i * 64 * 8 + j * 8 + k)["m"]
                == layoutA.apply(layoutB.apply(i * 64 * 8 + j * 8 + k)["m"])["m"]
            )

    test_compose_layout_1()