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

Function test_swizzle_layout_2

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

Source from the content-addressed store, hash-verified

1166 test_swizzle_layout_1()
1167
1168 def test_swizzle_layout_2():
1169 layout = SwizzleLayout(per_element=0, swizzle_len=3, atom_len=3, swizzle_inner=False)
1170 assert layout.size() == 64
1171 for i, j in itertools.product(range(8), range(8)):
1172 assert layout.apply(i * 8 + j)["m"] == (i ^ j) * 8 + j
1173
1174 test_swizzle_layout_2()
1175

Callers 1

test_applyFunction · 0.85

Calls 4

SwizzleLayoutClass · 0.90
productMethod · 0.80
sizeMethod · 0.45
applyMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…