From d98e7bf7ce13022c0a56963222add09549e40d0a Mon Sep 17 00:00:00 2001 From: ao jia Date: Fri, 5 Sep 2025 07:56:39 +0800 Subject: [PATCH] Fix comment in mma_atom.hpp (#2579) --- include/cute/atom/mma_atom.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/cute/atom/mma_atom.hpp b/include/cute/atom/mma_atom.hpp index 6367fc0c..57338e2a 100644 --- a/include/cute/atom/mma_atom.hpp +++ b/include/cute/atom/mma_atom.hpp @@ -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