Function case_compose_slice_2d

tests/python/tirx/test_layout.py:1643–1653

()

    case_compose_slice()

    def case_compose_slice_2d():
        # ComposeLayout slice with 2D shape
        compose = ComposeLayout(
            SwizzleLayout(per_element=3, swizzle_len=3, atom_len=3),
            TileLayout(S[(8, 64) : (64, 1)]),
        )
        shape = [8, 64]
        region = [(2, 4), (0, 64)]
        sliced = compose.slice(shape, region)
        assert sliced is not None
        verify_slice(compose, shape, region, sliced)

    case_compose_slice_2d()