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

Function case12_normalized9

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

Source from the content-addressed store, hash-verified

733 case11_normalized8()
734
735 def case12_normalized9():
736 # Normalized Tile Layout Test - 9 (tile = inner w/ device + diff major-dim)
737 outer = TileLayout(S[(16, 8, 4) : (1, 64, 16)])
738 inner = TileLayout(S[(2, 4, 2, 2) : (4, 1, 4, 3)])
739 layout_tile = inner.tile(outer, (16, 8, 4), (8, 2, 2))
740
741 outer_res = inner.is_tile_inner(layout_tile, (128, 16, 8), (8, 2, 2))
742 assert outer_res is not None
743 assert_structural_equal(outer_res.canonicalize(), outer.canonicalize())
744 assert inner.is_tile_inner(layout_tile.canonicalize(), (128, 16, 8), (8, 2, 2))
745 assert not outer.canonicalize().is_tile_inner(
746 layout_tile.canonicalize(), (128, 16, 8), (16, 8, 4)
747 )
748
749 case12_normalized9()
750

Callers 1

test_tile_layoutFunction · 0.85

Calls 5

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

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…