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

Relax some test tolerances. (#2917)

Browse Source
This commit is contained in:
Peter Hawkins 2020-05-01 11:45:28 -04:00 committed by GitHub
parent ac023bf28f
commit 25e8280d8a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/lax_test.py
View File

@ -115,7 +115,8 @@ LAX_OPS = [
op_record("atan", 1, float_dtypes, jtu.rand_small),
op_record("asinh", 1, float_dtypes, jtu.rand_default),
op_record("acosh", 1, float_dtypes, jtu.rand_positive),
op_record("atanh", 1, float_dtypes, jtu.rand_small),
# TODO(b/155331781): atanh has only ~float precision
op_record("atanh", 1, float_dtypes, jtu.rand_small, {onp.float64: 1e-9}),
op_record("sinh", 1, float_dtypes + complex_dtypes, jtu.rand_default),
op_record("cosh", 1, float_dtypes + complex_dtypes, jtu.rand_default),
op_record("lgamma", 1, float_dtypes, jtu.rand_positive,
@ -1773,7 +1774,7 @@ LAX_GRAD_OPS = [
dtypes=grad_float_dtypes, tol=1e-3),
grad_test_spec(lax.acos, nargs=1, order=2,
rng_factory=partial(jtu.rand_uniform, -1.3, 1.3),
dtypes=grad_float_dtypes, tol=1e-3),
dtypes=grad_float_dtypes, tol=2e-2),
# TODO(proteneer): atan2 input is already a representation of a
# complex number. Need to think harder about what this even means
# if each input itself is a complex number.