From 3d15f5bb218af1679c47e18e606a686bd99a5fb3 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 28 May 2026 14:43:01 +0000 Subject: [PATCH] Debug: 1 PV K-tile --- tests/unit/test_fmha_v5.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_fmha_v5.cu b/tests/unit/test_fmha_v5.cu index 54f48434..cddc687b 100644 --- a/tests/unit/test_fmha_v5.cu +++ b/tests/unit/test_fmha_v5.cu @@ -102,7 +102,7 @@ test_fmha_v5(const bf16_t* q, const bf16_t* k, const bf16_t* v, // ===== PV GEMM (SS) — per K-tile ===== { uint32_t idesc_pv = make_idesc(BLOCK_MN, HD); - for (int kt = 0; kt < NKT_PV; kt++) { + for (int kt = 0; kt < 1; kt++) { // Fill sPk from s_p_vals[kt*16..kt*16+15] for (int i = tid; i < TILE_SZ; i += 128) sPk[i] = 0; if (tid < 16) {