From a858ed1c145478d8be9dfab0a17e627729e5c949 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Wed, 27 May 2026 05:06:52 +0000 Subject: [PATCH] Fix test: normalize=False for un-normalized O comparison --- tests/unit/test_smem_acc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_smem_acc.py b/tests/unit/test_smem_acc.py index 97288564..643257e8 100644 --- 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)