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

Function test_swizzle_layout_0

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

Source from the content-addressed store, hash-verified

1144
1145 ################ Swizzle Layout
1146 def test_swizzle_layout_0():
1147 layout = SwizzleLayout(per_element=0, swizzle_len=3, atom_len=3)
1148 # assert layout.size == 64
1149 for i, j in itertools.product(range(8), range(8)):
1150 assert layout.apply(i * 8 + j)["m"] == i * 8 + i ^ j
1151
1152 test_swizzle_layout_0()
1153

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…