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

Fix flaky generated_fun_test.py test (#3885)

Browse Source
This commit is contained in:
Jamie Townsend 2020-07-28 16:27:07 +02:00 committed by GitHub
parent d7733c30d6
commit 7506a3e5f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/generated_fun_test.py
View File

@ -186,7 +186,7 @@ def inner_prod(xs, ys):
return sum(jnp.sum(x * y) for x, y in xys)
def jvp_fd(fun, args, tangents):
EPS = 1e-4
EPS = 1e-3
def eval_eps(eps):
return fun(*[x if t is None else x + eps * t
for x, t in zip(args, tangents)])