mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update docs that jax.debug is unsupported on Cloud TPUs

Browse Source
This commit is contained in:
Rafi Witten 2023-02-01 00:06:48 +00:00
parent 8dc1dff610
commit 278ff25ae1
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