jax
0e1c6a4b - [Mosaic GPU] Add support for passing in tcgen05.mma LHS in TMEM

Commit

273 days ago

[Mosaic GPU] Add support for passing in tcgen05.mma LHS in TMEM PiperOrigin-RevId: 753560393

Author

apaszke

Committer

Google-ML-Automation

Parents