mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump up tolerance for tests using _testQdwh.

Browse Source 
PiperOrigin-RevId: 643547107
This commit is contained in:
Benjamin Chetioui 2024-06-14 23:16:46 -07:00 committed by jax authors
parent 193591b5c0
commit 4e748e1fda
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/qdwh_test.py
View File

@ -68,7 +68,7 @@ class QdwhTest(jtu.JaxTestCase):
"""Computes the polar decomposition and tests its basic properties."""
eps = jnp.finfo(a.dtype).eps
u, h, iters, conv = qdwh.qdwh(a, dynamic_shape=dynamic_shape)
tol = 11 * eps
tol = 13 * eps
if dynamic_shape is not None:
m, n = dynamic_shape
a = a[:m, :n]