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)