debug: test multi-tile with s_k=128 only

2026-05-29 19:53:02 +00:00
parent 8e99bd50e6
commit 7598d548ee
1 changed files with 1 additions and 1 deletions
--- 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));