From 11da4daa01c0defe567d2b31e77b451dfd33c4d1 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 14:25:47 +0000
Subject: [PATCH] Debug: single PV K-tile

---
 tests/unit/test_fmha_smem_p.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/test_fmha_smem_p.cu b/tests/unit/test_fmha_smem_p.cu
index 726eebac..4c49fffe 100644
--- a/tests/unit/test_fmha_smem_p.cu
+++ b/tests/unit/test_fmha_smem_p.cu
@@ -126,7 +126,7 @@ test_fmha_smem_p(const bf16_t* __restrict__ q, const bf16_t* __restrict__ k,
     // Starting at offset kt * 2 * 1024 = kt * 2048 BF16
     {
         uint32_t idesc_pv = make_idesc(BLOCK_MN, HD);
-        for (int kt = 0; kt < NKT_PV; kt++) {
+        for (int kt = 0; kt < 1; kt++) {
             bf16_t* sp = sP + kt * 2048;
             bf16_t* sv = sV + kt * 256;
             uint64_t dp = make_umma_desc_kmajor_none(__cvta_generic_to_shared(sp), BLOCK_MN);