jax
7746f50f - [Mosaic GPU] Add support for f16/bf16 atomics

Commit

77 days ago

[Mosaic GPU] Add support for f16/bf16 atomics PiperOrigin-RevId: 878493131

References

#35596 - [Mosaic GPU] Add support for f16/bf16 atomics

Author

apaszke

apaszke

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading