From 223a4e485b6cd20e68af76d6432dc360f0748ed9 Mon Sep 17 00:00:00 2001 From: Roopak Srivastava Date: Mon, 16 Mar 2026 09:32:01 +0530 Subject: [PATCH] [test] Fix bfloat16 tolerance in test_selective_state_update (#20345) --- test/registered/layers/mamba/test_mamba_ssm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/registered/layers/mamba/test_mamba_ssm.py b/test/registered/layers/mamba/test_mamba_ssm.py index e2532e9c6..0178bfa2c 100644 --- a/test/registered/layers/mamba/test_mamba_ssm.py +++ b/test/registered/layers/mamba/test_mamba_ssm.py @@ -99,7 +99,7 @@ def test_selective_state_update(dim, dstate, has_z, itype): rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (5e-3, 1e-2) if itype == torch.bfloat16: - rtol, atol = 1e-2, 5e-2 + rtol, atol = 1e-2, 1e-1 if torch.version.hip: atol *= 2 # set seed