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 #14236 from rwitten:rwitten_debug_docs

Browse Source 
PiperOrigin-RevId: 506154997
This commit is contained in:
jax authors 2023-01-31 16:52:16 -08:00
commit fcb9dfb080
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/debugging/print_breakpoint.md
View File

@ -223,6 +223,7 @@ Furthermore, when using `jax.debug.print` with `jax.pjit`, a global synchronizat
#### Limitations
* Adding print statements is a manual process
* Can have performance impacts
* Unsupported on Cloud TPUs
## Interactive inspection with `jax.debug.breakpoint()`
@ -293,3 +294,4 @@ Because `jax.debug.breakpoint` is a just an application of `jax.debug.callback`,
#### Limitations
* Need to potentially use many breakpoints to pinpoint the source of an error
* Materializes many intermediates
* Unsupported on Cloud TPUs