From 1e0805ad6079d29fca2d33648cd1e9ebcb6ae412 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Fri, 22 May 2026 18:07:07 +0000
Subject: [PATCH] Diag: test n=384 (3 tiles) to find crash boundary

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

diff --git a/tests/unit/test_fmha_v3_diag.py b/tests/unit/test_fmha_v3_diag.py
index c6be5559..95ff71ca 100644
--- a/tests/unit/test_fmha_v3_diag.py
+++ b/tests/unit/test_fmha_v3_diag.py
@@ -266,7 +266,7 @@ class FmhaV3Diag:
 
 
 def test():
-    for n in [128, 256, 512, 1024]:
+    for n in [128, 256, 384]:
         torch.manual_seed(42)
         m, hd = 128, HEAD_DIM
         q = torch.randn(m, hd, 1, dtype=torch.bfloat16, device='cuda')