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

Fix cute doc (#1529)

Browse Source
This commit is contained in:
Wilber 2024-10-08 00:38:32 +08:00 committed by GitHub
parent e2b0789927
commit b27c49e84a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
media/docs/cute/0x_gemm_tutorial.md
View File

@ -150,7 +150,7 @@ This `local_tile` is simply shorthand for
1. apply the tiler via [`zipped_divide`](./02_layout_algebra.md#zipped-tiled-flat-divides) 1. apply the tiler via [`zipped_divide`](./02_layout_algebra.md#zipped-tiled-flat-divides)
```cpp ```cpp
// ((BLK_M,BLK_K),(m,k)) // ((BLK_M,BLK_K),(m,k))
Tensor gA_mk = zipped_divide(gA, select<0,2>(cta_tiler)); 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