From a767e90a12c6a205e42abbb363719615e768e4fe Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 28 May 2026 13:36:30 +0000 Subject: [PATCH] test: B=2.0 to understand TS MMA scale factor --- tests/unit/test_mma_ts.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_mma_ts.cu b/tests/unit/test_mma_ts.cu index 3f3bb926..94037723 100644 --- a/tests/unit/test_mma_ts.cu +++ b/tests/unit/test_mma_ts.cu @@ -38,7 +38,7 @@ test_mma_ts(float* o_out) int r = i / 16, c = i % 16; int ck = c / 8, lc = c % 8; int tmn = r / 8, lr = r % 8; - sV[ck * 2 * 64 + tmn * 64 + lr * 8 + lc] = f32_to_bf16(1.0f); + sV[ck * 2 * 64 + tmn * 64 + lr * 8 + lc] = f32_to_bf16(2.0f); // Use 2.0 to distinguish from A=1.0 } __syncthreads();