From 278ff25ae1d9e1397547170081f490a88b6ff77c Mon Sep 17 00:00:00 2001 From: Rafi Witten Date: Wed, 1 Feb 2023 00:06:48 +0000 Subject: [PATCH] Update docs that jax.debug is unsupported on Cloud TPUs --- docs/debugging/print_breakpoint.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/debugging/print_breakpoint.md b/docs/debugging/print_breakpoint.md index 929f30e4d..69267b506 100644 --- a/docs/debugging/print_breakpoint.md +++ b/docs/debugging/print_breakpoint.md @@ -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