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

Remove the doctest skip now that jit and pjit have been merged

Browse Source 
PiperOrigin-RevId: 508162840
This commit is contained in:
Yash Katariya 2023-02-08 13:09:24 -08:00 committed by jax authors
parent 1254d44dbd
commit e4d551a217
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/jaxpr.rst
View File

@ -412,7 +412,7 @@ which the computation should run. For example
... return x + arg * jnp.ones(1) # Include a constant in the inner function
... return arg + inner(arg - 2.)
...
>>> print(make_jaxpr(func12)(1.)) # doctest:+ELLIPSIS, +SKIP
>>> print(make_jaxpr(func12)(1.)) # doctest:+ELLIPSIS
{ lambda ; a:f32[]. let
b:f32[] = sub a 2.0
c:f32[1] = pjit[