mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[XLA] More error tolerance adjustments due to incoming change in XLA:CPU

Browse Source 
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
This commit is contained in:
Diego Caballero 2021-11-29 06:26:12 -08:00 committed by jax authors
parent c1ce85203c
commit 4f80e6af6e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/lax_scipy_test.py
View File

@ -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."""