()
| 1327 | # fmt: off |
| 1328 | @T.prim_func |
| 1329 | def func() -> None: |
| 1330 | T.device_entry() |
| 1331 | A = T.alloc_buffer([2], dtype="float16", scope="local") |
| 1332 | A_layout = T.TileLayout(T.S[(1, 2) : (2, 1)]) |
| 1333 | B = A.view(8, 8, layout=A_layout.tile(L_LANE, (8, 4), (1, 2))) |
| 1334 | B_local = B.local() |
| 1335 | B_local[0] = T.float16(0) |
| 1336 | # fmt: on |
| 1337 | |
| 1338 | bufs = _collect_buffers(func) |
| 1339 | b_local = bufs["B_local"] |