()
| 1152 | test_swizzle_layout_0() |
| 1153 | |
| 1154 | def test_swizzle_layout_1(): |
| 1155 | layout = SwizzleLayout(per_element=3, swizzle_len=3, atom_len=3) |
| 1156 | assert layout.size() == 512 |
| 1157 | for i, j, k in itertools.product(range(8), range(8), range(8)): |
| 1158 | assert layout.apply((i * 8 + j) * 8 + k)["m"] == (i * 8 + (i ^ j)) * 8 + k |
| 1159 | # apply can accept coord larger than size |
| 1160 | for p in range(4096): |
| 1161 | outer = p // 512 |
| 1162 | inner = p % 512 |
| 1163 | i, j, k = inner // 64, (inner % 64) // 8, inner % 8 |
| 1164 | assert layout.apply(p)["m"] == outer * 512 + (i * 8 + (i ^ j)) * 8 + k |
| 1165 | |
| 1166 | test_swizzle_layout_1() |
| 1167 |
no test coverage detected
searching dependent graphs…