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

Function test_swizzle_layout_3

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

Source from the content-addressed store, hash-verified

1174 test_swizzle_layout_2()
1175
1176 def test_swizzle_layout_3():
1177 layout = SwizzleLayout(per_element=0, swizzle_len=2, atom_len=3)
1178 for i, j in itertools.product(range(8), range(8)):
1179 _outer_i, inner_i = i // 4, i % 4
1180 outer_j, inner_j = j // 4, j % 4
1181 assert layout.apply(i * 8 + j)["m"] == i * 8 + outer_j * 4 + (inner_i ^ inner_j)
1182
1183 test_swizzle_layout_3()
1184

Callers 1

test_applyFunction · 0.85

Calls 3

SwizzleLayoutClass · 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…