youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix incorrect K dim in CuTe MMA Atom doc. (#2544)

Browse Source
This commit is contained in:
Lifu Huang
2025-08-10 19:40:56 -07:00
committed by GitHub
parent 6dd13d4278
commit 23139309e9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
media/docs/cpp/cute/0t_mma_atom.md
View File

@ -426,7 +426,7 @@ where we see 16 copies of the 64x8 tile.
GMMA atoms that consume A and B sources directly from shared memory are a bit interesting. The GMMA Descriptor is constructed on an entire tile of A and/or B data in shared memory rather than being partitioned by threads. That is, every thread sees the entire tile of data and the tile is not reordered so that the descriptor can be constructed on it. In `ALayout` form, this can be expressed
```cpp
// (T128,V64x8) -> (M64,K16)
// (T128,V64x16) -> (M64,K16)
using ALayout = Layout<Shape <_128, Shape <_64,_16>>,
Stride< _0, Stride< _1,_64>>>;
```