Function compose_layout_size

tests/python/tirx/test_layout.py:998–1003

()

    swizzle_layout_size()

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

    compose_layout_size()