jax
e189cd46 - [Mosaic GPU] Add a debug_print for TMEM refs

Commit

1 year ago

[Mosaic GPU] Add a debug_print for TMEM refs This is quite helpful while trying to debug the load/store routines. PiperOrigin-RevId: 753599963

References

Author

apaszke

Committer

Google-ML-Automation

Parents