Enable all 8 PV K-tiles

2026-05-28 14:45:13 +00:00
parent 6f5be8a4e4
commit af93c283c7
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_fmha_v5.cu
+++ b/tests/unit/test_fmha_v5.cu
@@ -103,7 +103,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 < 1; kt++) {
+        for (int kt = 0; kt < NKT_PV; 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) {