Revert "debug: test 12w identity softmax with n=256 to verify multi-tile pipeline"

This reverts commit 6cf8702e3c.
2026-05-22 10:25:48 +00:00
parent 6cf8702e3c
commit 5a08b79364
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_fmha_v3_12w.py
+++ b/tests/unit/test_fmha_v3_12w.py
@@ -295,7 +295,7 @@ class FmhaV3:

 def test():
    torch.manual_seed(42)
-    for n in [128, 256]:
+    for n in [128]:
        m, hd = 128, HEAD_DIM
        q = torch.randn(m, hd, 1, dtype=torch.bfloat16, device='cuda')
        k = torch.randn(n, hd, 1, dtype=torch.bfloat16, device='cuda')