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

Update 0x_gemm_tutorial.md (#1982)

Browse Source 
Shouldn't this be BLK_M, BLK_**K**, k
This commit is contained in:
Andrew O'Neill
2025-01-06 22:04:35 -05:00
committed by GitHub
parent bf9da7b76c
commit cffd5d32b7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
media/docs/cute/0x_gemm_tutorial.md
+1 -1
View File
@@ -154,7 +154,7 @@ Tensor gA_mk = zipped_divide(mA, select<0,2>(cta_tiler));
``` ```
2. apply the coord to the second mode, the "Rest" mode, to extract out the correct tiles for this CTA. 2. apply the coord to the second mode, the "Rest" mode, to extract out the correct tiles for this CTA.
```cpp ```cpp
// (BLK_M,BLK_N,k) // (BLK_M,BLK_K,k)
Tensor gA = gA_mk(make_coord(_,_), select<0,2>(cta_coord)); Tensor gA = gA_mk(make_coord(_,_), select<0,2>(cta_coord));
``` ```
Because the projections of the tiler and coord are symmetric and the two steps (apply a tiler and then slice into the rest-mode to produce a partition) are so common, they are wrapped together into the projective `local_tile` interface. Because the projections of the tiler and coord are symmetric and the two steps (apply a tiler and then slice into the rest-mode to produce a partition) are so common, they are wrapped together into the projective `local_tile` interface.