MCPcopy Index your code
hub / github.com/apache/tvm / case7_normalized4

Function case7_normalized4

tests/python/tirx/test_layout.py:646–661  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

644 case6_sharded2()
645
646 def case7_normalized4():
647 # Normalized Tile Layout Test - 4 (tile < inner)
648 outer = TileLayout(S[(4, 2, 1) : (2, 1, 1)])
649 inner = TileLayout(S[(2, 4, 1) : (2, 3, 1)])
650 layout_tile = inner.tile(outer, outer_shape=(4, 2), inner_shape=(2, 4))
651
652 inner_res = outer.is_tile_outer(layout_tile, (8, 8), (4, 2))
653 assert inner_res is not None
654 assert_structural_equal(inner_res.canonicalize(), inner.canonicalize())
655
656 outer_res = inner.is_tile_inner(layout_tile, (8, 8), (2, 4))
657 assert outer_res is not None
658 assert_structural_equal(outer_res.canonicalize(), outer.canonicalize())
659
660 assert outer.is_tile_inner(layout_tile, (8, 8), (4, 2)) is None
661 assert inner.is_tile_outer(layout_tile, (8, 8), (2, 4)) is None
662
663 case7_normalized4()
664

Callers 1

test_tile_layoutFunction · 0.85

Calls 6

TileLayoutClass · 0.90
assert_structural_equalFunction · 0.90
is_tile_outerMethod · 0.80
canonicalizeMethod · 0.80
is_tile_innerMethod · 0.80
tileMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…