From 9a3b43c42bc5ceddd75f0709a2636b44b0d2608b Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 14:47:10 +0000
Subject: [PATCH] Fix reference to also use uniform P

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

diff --git a/tests/unit/test_fmha_v5.cu b/tests/unit/test_fmha_v5.cu
index 6bdfbc31..db049639 100644
--- a/tests/unit/test_fmha_v5.cu
+++ b/tests/unit/test_fmha_v5.cu
@@ -152,7 +152,7 @@ test_fmha_v5(const bf16_t* q, const bf16_t* k, const bf16_t* v,
         for (int j=0;j<SK;j++) mx = fmaxf(mx, s[j]);
         float sm = 0.0f;
         for (int j=0;j<SK;j++) { s[j] = expf(s[j]-mx); sm += s[j]; }
-        for (int j=0;j<SK;j++) s[j] /= sm;
+        for (int j=0;j<SK;j++) s[j] = 1.0f / SK;  // Uniform P (matching override)
         for (int d=0;d<HD;d++) {
             float ov = 0.0f;
             for (int j=0;j<SK;j++) ov += s[j] * bf16_to_f32(v[d*SK+j]);