jax
5a7325eb
- [Mosaic GPU] Only allow extracting a non-replicated layout assignment out of a `LeastReplicatedExpression` very conservatively.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
242 days ago
[Mosaic GPU] Only allow extracting a non-replicated layout assignment out of a `LeastReplicatedExpression` very conservatively. It's not correct to propagate replicated layouts upwards in the general case. PiperOrigin-RevId: 781431727
References
#30048 - [Mosaic GPU] Only allow extracting a non-replicated layout assignment out of a `LeastReplicatedExpression` very conservatively.
Author
bchetioui
Committer
Google-ML-Automation
Parents
5c6d5411
Loading