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

Merge pull request #15673 from jakevdp:fix-i0e-test

Browse Source 
PiperOrigin-RevId: 525556536
This commit is contained in:
jax authors 2023-04-19 14:48:29 -07:00
commit 968dbaf8f3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/lax_scipy_special_functions_test.py
View File

@ -88,7 +88,8 @@ JAX_SPECIAL_FUNCTION_RECORDS = [
"i0", 1, float_dtypes, jtu.rand_default, True
),
op_record(
"i0e", 1, float_dtypes, jtu.rand_default, True
# Note: values near zero can fail numeric gradient tests.
"i0e", 1, float_dtypes, functools.partial(jtu.rand_not_small, offset=1E-3), True
),
op_record(
"i1", 1, float_dtypes, jtu.rand_default, True