tvm_tirx.backend.cuda.layout.mma_shared_layout#

tvm_tirx.backend.cuda.layout.mma_shared_layout(dtype, swizzle_mode, shape)#

Generate the MMA-compatible shared-memory layout for shape and dtype.

It uses a default tiling strategy to tile the TMA atom layout into the shared memory.