From 7df17384fd7a207e85796c0516010dbb538db3bb Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Fri, 29 May 2026 19:59:21 +0000
Subject: [PATCH] test: multi-tile s_k=128/256/384/512

---
 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 2d8ea96e..e3494c1f 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}) {
+    for (int s_k : {128, 256, 384, 512}) {
         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));