tvm_tirx.backend.cuda.layout.mma_atom_shape

tvm_tirx.backend.cuda.layout.mma_atom_shape(dtype, swizzle_mode, shape=None)

Generate the MMA-compatible shared-memory atom shape.

Parameters:

• dtype (str)

• swizzle_mode (SwizzleMode | int)

• shape (list[int] | None)