MCPcopy Create free account
hub / github.com/apache/tvm / test_group_by_logical_shape

Function test_group_by_logical_shape

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

Source from the content-addressed store, hash-verified

1329
1330
1331def test_group_by_logical_shape():
1332 def case1():
1333 layout = TileLayout(S[(8, 8) : (8, 1)])
1334 layout = layout.tile(layout, outer_shape=[8, 8], inner_shape=[8, 8])
1335 outer, seps = layout.group([64, 64])
1336 assert_structural_equal(outer, layout)
1337 assert seps[0] == 0
1338 assert seps[1] == 2
1339 assert seps[2] == 4
1340
1341 case1()
1342
1343
1344def test_permute_by_groups():

Callers

nothing calls this directly

Calls 1

case1Function · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…