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

Function case11_normalized8

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

Source from the content-addressed store, hash-verified

717 case10_normalized7()
718
719 def case11_normalized8():
720 # Normalized Tile Layout Test - 8 (tile = inner w/ device)
721 outer = TileLayout(S[(8, 8, 4) : (32, 4, 1)])
722 inner = TileLayout(S[(8, 8, 1, 4, 2) : (4, 4 @ laneid, 2, 1 @ laneid, 1)])
723 layout_tile = inner.tile(outer, (8, 8, 4), (8, 8, 8))
724
725 outer_res = inner.is_tile_inner(layout_tile, (64, 64, 32), (8, 8, 8))
726 assert outer_res is not None
727 assert_structural_equal(outer_res.canonicalize(), outer.canonicalize())
728 assert inner.is_tile_inner(layout_tile.canonicalize(), (64, 64, 32), (8, 8, 8))
729 assert not outer.canonicalize().is_tile_inner(
730 layout_tile.canonicalize(), (64, 64, 32), (8, 8, 4)
731 )
732
733 case11_normalized8()
734

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…