jax
5a3605d3
- [pallas:mgpu] Allow WGMMA accuculator to cond yielded values.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
285 days ago
[pallas:mgpu] Allow WGMMA accuculator to cond yielded values. PiperOrigin-RevId: 754112101
References
#28479 - [pallas:mgpu] Allow WGMMA accuculator to cond yielded values.
Author
cperivol
Committer
Google-ML-Automation
Parents
c7f3d1c0
Loading
