From 4f80e6af6eb6afb68ec68e1bedf4829cabf144e2 Mon Sep 17 00:00:00 2001 From: Diego Caballero Date: Mon, 29 Nov 2021 06:26:12 -0800 Subject: [PATCH] [XLA] More error tolerance adjustments due to incoming change in XLA:CPU We are about to change the vectorization strategy for XLA:CPU. As a result, we have to increase the error tolerance of the impacted tests. PiperOrigin-RevId: 412872724 --- tests/lax_scipy_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/lax_scipy_test.py b/tests/lax_scipy_test.py index 525b85019..702d89033 100644 --- a/tests/lax_scipy_test.py +++ b/tests/lax_scipy_test.py @@ -406,7 +406,7 @@ class LaxBackedScipyTests(jtu.JaxTestCase): actual = jnp.real( lsp_special.sph_harm(jnp.array([0]), jnp.array([1]), theta, phi, n_max)) - self.assertAllClose(actual, expected, rtol=7e-8, atol=1.5e-8) + self.assertAllClose(actual, expected, rtol=2e-7, atol=6e-8) def testSphHarmOrderOneDegreeOne(self): """Tests the spherical harmonics of order one and degree one."""