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 #10978 from jakevdp:x64-modf

Browse Source 
PiperOrigin-RevId: 452867169
This commit is contained in:
jax authors 2022-06-03 16:38:30 -07:00
commit b665c43799
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/_src/numpy/ufuncs.py
View File

@ -591,6 +591,7 @@ def real(val):
@jit
def modf(x, out=None):
_check_arraylike("modf", x)
x, = _promote_dtypes_inexact(x)
if out is not None:
raise NotImplementedError("The 'out' argument to jnp.modf is not supported.")
whole = _where(lax.ge(x, lax_internal._zero(x)), floor(x), ceil(x))