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

Function test_tile_layout_3

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

Source from the content-addressed store, hash-verified

1126 test_tile_layout_2()
1127
1128 def test_tile_layout_3():
1129 layout = TileLayout(S[(8, 1, 4, 2) : (4 @ laneid, 2, 1 @ laneid, 1)])
1130 for i0, i1 in itertools.product(range(8), range(8)):
1131 res = layout.apply(i0, i1, shape=(8, 8))
1132 assert res["m"] == i1 % 2
1133 assert res["laneid"] == i0 * 4 + i1 // 2
1134
1135 test_tile_layout_3()
1136

Callers 1

test_applyFunction · 0.85

Calls 3

TileLayoutClass · 0.90
productMethod · 0.80
applyMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…