jax
d2bf034c
- [Mosaic GPU] Test the wgmma_op lowering when a is in registers.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
323 days ago
[Mosaic GPU] Test the wgmma_op lowering when a is in registers. I had to add support for wgmma layout in vector_load. Not sure if this is useful outside the test. PiperOrigin-RevId: 735384104
References
#26990 - [Mosaic GPU] Test the wgmma_op lowering when a is in registers.
#271 - CI: 03/11/25 upstream sync
Author
dimitar-asenov
Committer
Google-ML-Automation
Parents
5a7ef404
Loading