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

Fix Typos (#2021)

Browse Source 
* Fix Typo

* Fix Typo
This commit is contained in:
Lei Mao
2025-01-08 20:46:28 -08:00
committed by GitHub
parent 24f991e879
commit 52b35e90ce
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/cute/tutorial/sgemm_1.cu
View File

@@ -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)