jax
260d3f0f
- [Mosaic GPU] Add SMEM inference rule for `mgpu.custom_op`.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
202 days ago
[Mosaic GPU] Add SMEM inference rule for `mgpu.custom_op`. PiperOrigin-RevId: 819092469
References
#32259 - [Mosaic GPU] Add SMEM inference rule for `mgpu.custom_op`.
Author
dimitar-asenov
Committer
Google-ML-Automation
Parents
b4e24590
Loading
