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

Fix comment in mma_atom.hpp (#2579)

Browse Source
This commit is contained in:
ao jia
2025-09-05 07:56:39 +08:00
committed by GitHub
parent b6ccf34aef
commit d98e7bf7ce
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/cute/atom/mma_atom.hpp
View File

@ -262,7 +262,7 @@ struct TiledMMA : MMA_Atom
make_layout(size<1>(AtomShape_MNK{})));
auto c_tensor = zipped_divide(t_tensor, c_tile); // ((AtomM,AtomN),(RestM,RestN))
// Transform the Atom mode from (M,K) to (Thr,Val)
// Transform the Atom mode from (M,N) to (Thr,Val)
auto tv_tensor = c_tensor.compose(AtomLayoutC_TV{},_); // ((ThrV,FrgV),(RestM,RestN))
// Tile the tensor for the C-threads
@ -340,7 +340,7 @@ struct TiledMMA : MMA_Atom
make_layout(size<2>(AtomShape_MNK{})));
auto b_tensor = zipped_divide(t_tensor, b_tile); // ((AtomN,AtomK),(RestN,RestK))
// Transform the Atom mode from (M,K) to (Thr,Val)
// Transform the Atom mode from (N,K) to (Thr,Val)
auto tv_tensor = b_tensor.compose(AtomLayoutB_TV{},_); // ((ThrV,FrgV),(RestN,RestK))
// Tile the tensor for the Thread