jax
51d35912
- [Mosaic GPU] Reenable the use of redux.sync, but only when all lanes participate
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
141 days ago
[Mosaic GPU] Reenable the use of redux.sync, but only when all lanes participate This seems to be the only case when this instruction is actually profitable. PiperOrigin-RevId: 857092650
References
#34399 - [Mosaic GPU] Reenable the use of redux.sync, but only when all lanes participate
Author
apaszke
Committer
Google-ML-Automation
Parents
7a5e13d7
Loading