Function test_compose_layout_0

tests/python/tirx/test_layout.py:1186–1195

()

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

    test_compose_layout_0()