Function compose_layout_span

tests/python/tirx/test_layout.py:1025–1030

()

    swizzle_layout_span()

    def compose_layout_span():
        layout = ComposeLayout(
            SwizzleLayout(per_element=3, swizzle_len=3, atom_len=3),
            TileLayout(S[(8, 64) : (64, 1)]),
        )
        assert layout.span() == 512

    compose_layout_span()