Skip to content

Commit

Permalink
[jax2tf] Adjust tolerance for asinh test.
Browse files Browse the repository at this point in the history
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
  • Loading branch information
gnecula authored and jax authors committed Apr 19, 2024
1 parent c7517b8 commit cea36a0
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions jax/experimental/jax2tf/tests/jax2tf_limitations.py
Original file line number Diff line number Diff line change
Expand Up @@ -281,6 +281,10 @@ def asinh(cls, harness: test_harnesses.Harness):
custom_numeric(dtypes=[np.complex64], devices=("cpu", "gpu", "tpu"),
tol=1e-3),
custom_numeric(dtypes=[np.complex128], devices=("cpu", "gpu"), tol=1e-12),
custom_numeric(dtypes=[np.complex128], devices=("cpu",),
modes=("compiled",),
tol=1e-13,
native_serialization=Jax2TfLimitation.FOR_NATIVE | Jax2TfLimitation.FOR_NON_NATIVE),
cls.helper_get_trig_custom_limitation(np.sinh)
]

Expand Down

0 comments on commit cea36a0

Please sign in to comment.