From 52b35e90ced06069899c0c211b7f76b1b1d30070 Mon Sep 17 00:00:00 2001 From: Lei Mao Date: Wed, 8 Jan 2025 20:46:28 -0800 Subject: [PATCH] Fix Typos (#2021) * Fix Typo * Fix Typo --- examples/cute/tutorial/sgemm_1.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/cute/tutorial/sgemm_1.cu b/examples/cute/tutorial/sgemm_1.cu index e5bf9a92..cdd23503 100644 --- a/examples/cute/tutorial/sgemm_1.cu +++ b/examples/cute/tutorial/sgemm_1.cu @@ -134,9 +134,9 @@ gemm_device(ProblemShape shape_MNK, CtaTiler cta_tiler, // TUTORIAL: Example of partitioning via projections of a ThreadLayout tC - // Partition sA (M,K) by the rows of tC + // Partition sA (BLK_M, BLK_K) by the rows of tC Tensor tCsA = local_partition(sA, tC, threadIdx.x, Step<_1, X>{}); // (THR_M,BLK_K) - // Partition sB (N,K) by the cols of tC + // Partition sB (BLK_N, BLK_K) by the cols of tC Tensor tCsB = local_partition(sB, tC, threadIdx.x, Step< X,_1>{}); // (THR_N,BLK_K) // Partition gC (M,N) by the tile of tC Tensor tCgC = local_partition(gC, tC, threadIdx.x, Step<_1,_1>{}); // (THR_M,THR_N)