jax
7192cf7d - [Mosaic GPU] Add support for warp-level f8 MMAs

Commit

4 days ago

[Mosaic GPU] Add support for warp-level f8 MMAs PiperOrigin-RevId: 875627922

References

#26604 - [ROCm][pallas:triton] Fix atomic min/max lowering for uint and float types

#34702 - [ROCm] Add script for running bazel tests on ROCm

#34722 - [ROCm] ToT ROCm Unit Test Skips

#35265 - [Mosaic GPU] Add support for warp-level f8 MMAs

#35496 - [ROCm] Use prebuilt wheels for rocm plugin deps

Author

apaszke

apaszke

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading