From 670480df3abe2ba959f2c4d66b0137a2eba62731 Mon Sep 17 00:00:00 2001 From: Aidan Do Date: Tue, 6 Jan 2026 20:55:21 -0800 Subject: [PATCH] Fix SFB Layout scale granularity representation (#2924) --- examples/81_blackwell_gemm_blockwise/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/81_blackwell_gemm_blockwise/README.md b/examples/81_blackwell_gemm_blockwise/README.md index 9fe03bab..786d036d 100644 --- a/examples/81_blackwell_gemm_blockwise/README.md +++ b/examples/81_blackwell_gemm_blockwise/README.md @@ -19,7 +19,7 @@ and SFB. This leads to a GEMM $D = \alpha \text{SFA} * A \text{ SFB} * B + \beta These can be represented in CuTe as: - *SFA Layout*: $((\text{scale granularity M}, M / \text{scale granularity M}), (\text{scale granularity K}, K / \text{scale granularity K})) : ((0, int), (0, int))$ -- *SFB Layout*: $((\text{scale granularity N}, M / \text{scale granularity M}), (\text{scale granularity K}, K / \text{scale granularity K})) : ((0, int), (0, int))$ +- *SFB Layout*: $((\text{scale granularity N}, N / \text{scale granularity N}), (\text{scale granularity K}, K / \text{scale granularity K})) : ((0, int), (0, int))$ The 0 element stride ensures the same group of coordinates to map to the same element in the scale factors.