Skip to content

Commit 76c96b0

Browse files
authored
Fix incorrect shapes in copy_atom doc comments. (#2575)
1 parent d98e7bf commit 76c96b0

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

include/cute/atom/copy_atom.hpp

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -208,11 +208,11 @@ struct TiledCopy : Copy_Atom
208208
// Tile a tensor or a layout from shape
209209
// (M,N,...)
210210
// to shape
211-
// ((ThrV,ThrX),FrgV,(RestM,RestN,...))
211+
// (Thr,(FrgV,FrgX),(RestM,RestN,...))
212212
// where
213-
// ThrV: The threads local to a COPY_ATOM Src.
214-
// ThrX: The threads tiled across COPY_ATOMs Src.
213+
// Thr: The logical threads within the tiled copy.
215214
// FrgV: The values local to a COPY_ATOM Src.
215+
// FrgX: The values tiled across COPY_ATOMs Src.
216216
// RestM: The values tiled in M.
217217
// RestN: The values tiled in N.
218218
template <class STensor>
@@ -229,11 +229,11 @@ struct TiledCopy : Copy_Atom
229229
// Tile a tensor or a layout from shape
230230
// (M,N,...)
231231
// to shape
232-
// ((ThrV,ThrX),FrgV,(RestM,RestN,...))
232+
// (Thr,(FrgV,FrgX),(RestM,RestN,...))
233233
// where
234-
// ThrV: The threads local to a COPY_ATOM Dst.
235-
// ThrX: The threads tiled across COPY_ATOMs Dst.
234+
// Thr: The logical threads within the tiled copy.
236235
// FrgV: The values local to a COPY_ATOM Dst.
236+
// FrgX: The values tiled across COPY_ATOMs Dst.
237237
// RestM: The values tiled in M.
238238
// RestN: The values tiled in N.
239239
template <class DTensor>
@@ -250,7 +250,7 @@ struct TiledCopy : Copy_Atom
250250
// Tile a tensor or a layout from shape
251251
// ((TileM,TileN,...), (RestM,RestN,...))
252252
// to shape
253-
// ((ThrV,ThrX),FrgV,(RestM,RestN,...))
253+
// (Thr,(FrgV,FrgX),(RestM,RestN,...))
254254
template <class Tensor, class Ref2TrgLayout>
255255
CUTE_HOST_DEVICE constexpr static
256256
auto

0 commit comments

Comments
 (0)