Skip to content

Commit cea36a0

Browse files
gneculajax authors
authored andcommitted
[jax2tf] Adjust tolerance for asinh test.
This test has started to fail in compiled mode, for complex128, but with small errors (1e-14). Adjust the tolerance for both the native and non-native serialization mode. PiperOrigin-RevId: 626373781
1 parent c7517b8 commit cea36a0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

jax/experimental/jax2tf/tests/jax2tf_limitations.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,10 @@ def asinh(cls, harness: test_harnesses.Harness):
281281
custom_numeric(dtypes=[np.complex64], devices=("cpu", "gpu", "tpu"),
282282
tol=1e-3),
283283
custom_numeric(dtypes=[np.complex128], devices=("cpu", "gpu"), tol=1e-12),
284+
custom_numeric(dtypes=[np.complex128], devices=("cpu",),
285+
modes=("compiled",),
286+
tol=1e-13,
287+
native_serialization=Jax2TfLimitation.FOR_NATIVE | Jax2TfLimitation.FOR_NON_NATIVE),
284288
cls.helper_get_trig_custom_limitation(np.sinh)
285289
]
286290

0 commit comments

Comments
 (0)