diff --git a/test/test_examples.py b/test/test_examples.py index 287f5d4db..bb35334b2 100644 --- a/test/test_examples.py +++ b/test/test_examples.py @@ -413,8 +413,8 @@ def test_rms_norm_bwd_dx(self): block_size=32, num_warps=4, num_stages=3, - rtol=1e-3, - atol=1e-3, + rtol=2e-3, + atol=5e-3, ) )