From cffd5d32b760f95f39d263c59bb8e34a882e13f4 Mon Sep 17 00:00:00 2001 From: Andrew O'Neill Date: Mon, 6 Jan 2025 19:04:35 -0800 Subject: [PATCH] Update 0x_gemm_tutorial.md (#1982) Shouldn't this be BLK_M, BLK_**K**, k --- media/docs/cute/0x_gemm_tutorial.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/media/docs/cute/0x_gemm_tutorial.md b/media/docs/cute/0x_gemm_tutorial.md index a2f4dd2e..b9ace8b1 100644 --- a/media/docs/cute/0x_gemm_tutorial.md +++ b/media/docs/cute/0x_gemm_tutorial.md @@ -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. ```cpp -// (BLK_M,BLK_N,k) +// (BLK_M,BLK_K,k) 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.