Function test_trainium_layout_0

tests/python/tirx/test_layout.py:1214–1219

()

    ################ Trainium Layout
    def test_trainium_layout_0():
        layout = TileLayout(S[(8, 8) : (8 @ F, 1 @ P)])
        for i, j in itertools.product(range(8), range(8)):
            coord = layout.apply(i, j, shape=(8, 8))
            assert coord["P"] == j
            assert coord["F"] == i * 8

    test_trainium_layout_0()