Fix test: normalize=False for un-normalized O comparison

2026-05-27 05:06:52 +00:00
parent 2e262d2b99
commit a858ed1c14
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_smem_acc.py
+++ b/tests/unit/test_smem_acc.py
@@ -9,7 +9,7 @@ import cuda.bindings.driver as cuda
 from dsv4.kernels.attention.fmha_smem_acc import FmhaKernel


-def test_smem_acc(hd=64, s_k=256, use_smem_p=False, normalize=True):
+def test_smem_acc(hd=64, s_k=256, use_smem_p=False, normalize=False):
    m = 128
    n_kv_tiles = s_k // 128
    torch.manual_seed(42)