tcgen05.mma
Transpose A
否
是
Transpose B
否
是
M（output rows）
128
N（output cols）
16
32
64
128
K
16
Row group
Col group
A (SMEM)
×
B (SMEM)
Tensor Core
tcgen05.mma
→
C (TMEM)
128 lanes（M）
Col（N）— 由
tmem.alloc
分配
K iteration：
SMEM Descriptor（A, B）
smem_desc =
base_addr
|
leading_byte_offset
|
stride_byte_offset
|
start_addr_off
tcgen05.mma PTX
tcgen05.mma.cta_group::1.kind::f16 [
taddr
],
a_desc
,
b_desc
,
idesc
,
{
mask0, mask1, mask2, mask3
},
enable
;