jax
7ad15086 - [Mosaic GPU] Do not predicate collective ops in the lowering.

Commit

242 days ago

[Mosaic GPU] Do not predicate collective ops in the lowering. Users should manually predicate the op. PiperOrigin-RevId: 796802662

References

Author

allanrenucci

Committer

Google-ML-Automation

Parents