()
| 623 | case5_sharded1() |
| 624 | |
| 625 | def case6_sharded2(): |
| 626 | # Tile over a sharded layout - 2 |
| 627 | inner = TileLayout(S[(8, 4) : (4 @ laneid, 1 @ laneid)]) |
| 628 | outer = TileLayout(S[(8, 8) : (8, 1)]) |
| 629 | layout_tile = inner.tile(outer=outer, outer_shape=(8, 8), inner_shape=(8, 4)) |
| 630 | layout_expected = TileLayout(S[(8, 8, 8, 4) : (8, 4 @ laneid, 1, 1 @ laneid)]) |
| 631 | assert_structural_equal(layout_expected, layout_tile) |
| 632 | |
| 633 | outer_res = inner.is_tile_inner(layout_tile, (64, 32), (8, 4)) |
| 634 | assert outer_res is not None |
| 635 | assert_structural_equal(outer_res.canonicalize(), outer.canonicalize()) |
| 636 | |
| 637 | inner_res = outer.is_tile_outer(layout_tile, (64, 32), (8, 8)) |
| 638 | assert inner_res is not None |
| 639 | assert_structural_equal(inner_res.canonicalize(), inner.canonicalize()) |
| 640 | |
| 641 | assert outer.is_tile_inner(layout_tile, (64, 32), (8, 8)) is None |
| 642 | assert inner.is_tile_outer(layout_tile, (64, 32), (8, 4)) is None |
| 643 | |
| 644 | case6_sharded2() |
| 645 |
no test coverage detected
searching dependent graphs…