Skip to content

Commit

Permalink
Update debugging Pallas g3doc to remove text about scalar printing re…
Browse files Browse the repository at this point in the history
…striction

PiperOrigin-RevId: 718560406
  • Loading branch information
sharadmv authored and Google-ML-Automation committed Jan 22, 2025
1 parent 23d360b commit 64e9b07
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jax/experimental/pallas/g3doc/debugging.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ Note that interpret mode will not be able to fully replicate the behavior or pro

### debug_print

The `pl.debug_print` function can be used to print runtime values inside of a kernel. The implementation is currently limited to scalar values, but we are working on lifting this limitation.
The `pl.debug_print` function can be used to print runtime values inside of a kernel.

For TPUs only, the kernel must be compiled with the 'xla_tpu_enable_log_recorder' option.
<!--internal:1-->
Expand Down

0 comments on commit 64e9b07

Please sign in to comment.