jax
9ba1fd28 - [Pallas TPU] Add vector support to `pl.debug_print`

Commit

1 year ago

[Pallas TPU] Add vector support to `pl.debug_print` PiperOrigin-RevId: 715085454

References

#25099 - [Pallas TPU] Add vector support to `pl.debug_print`

#213 - CI: 01/27/25 upstream sync

Author

ayaka14732

ayaka14732

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading