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

Merge pull request #26839 from Sai-Suraj-27:fix_jax.debug.print

Browse Source 
PiperOrigin-RevId: 735511953
This commit is contained in:
jax authors 2025-03-10 14:26:45 -07:00
commit b8590816bf
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/debugging/print_breakpoint.md
View File

@ -91,8 +91,8 @@ def f(x):
jax.debug.print("x: {}", x)
return x
jax.pmap(f)(xs)
# Prints: x: 1.0
# x: 0.0
# Prints: x: 0.0
# x: 1.0
# OR
# Prints: x: 1.0
# x: 0.0