Moving Multiple Swizzle Atoms with 3D TMA
Split each 256-byte row into two 128-byte groups, then swizzle each group independently
Swizzle
None
128B
Col offset
Global Memory
16×256 fp16 (each cell = 8 fp16 = 16B)
→
3D TMA
Shared Memory (grouped into atoms and swizzled)