From 7598d548ee205ccc2af42eba4983f1c328fd14f4 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Fri, 29 May 2026 19:53:02 +0000
Subject: [PATCH] debug: test multi-tile with s_k=128 only

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

diff --git a/tests/unit/test_fmha_6warp_tma_multitile.cu b/tests/unit/test_fmha_6warp_tma_multitile.cu
index e3494c1f..2d8ea96e 100644
--- a/tests/unit/test_fmha_6warp_tma_multitile.cu
+++ b/tests/unit/test_fmha_6warp_tma_multitile.cu
@@ -75,7 +75,7 @@ int main() {
 
     int total_fail = 0;
 
-    for (int s_k : {128, 256, 384, 512}) {
+    for (int s_k : {128}) {
         printf("\n--- s_k=%d (%d KV tiles) ---\n", s_k, (s_k + 127) / 128);
 
         bf16_t* h_q = (bf16_t*)calloc(HD, sizeof(bf16_t));