Adjusted atol to fix golden tests broken by the change of the loop un…

…rolling default in IrCompiler.

PiperOrigin-RevId: 725656881

jax_cfd/base/equations_test.py

@@ -97,7 +97,7 @@ class SemiImplicitNavierStokesTest(test_util.TestCase):
            dt=1e-3,
            time_steps=1000,
            divergence_atol=1e-3,
-           momentum_atol=2e-3),
+           momentum_atol=1.5e-2),
       dict(testcase_name='semi_implicit_gaussian_force_upwind',
            velocity=zero_velocity_field,
            forcing=gaussian_forcing,
@@ -180,7 +180,7 @@ class ImplicitDiffusionNavierStokesTest(test_util.TestCase):
           dt=1e-3,
           time_steps=1000,
           divergence_atol=1e-3,
-          momentum_atol=3e-3),
+          momentum_atol=1.5e-2),
       dict(
           testcase_name='implicit_gaussian_force_upwind',
           velocity=zero_velocity_field,
@@ -207,7 +207,7 @@ class ImplicitDiffusionNavierStokesTest(test_util.TestCase):
           convect=advection.convect_linear,
           diffusion_solve=diffusion.solve_fast_diag,
           pressure_solve=pressure.solve_fast_diag,
-          dt=1e-3,
+          dt=5e-3,
           time_steps=1000,
           divergence_atol=1e-3,
           momentum_atol=1e-3),