jax
60e44d27 - [Mosaic GPU] Add pallas and dialect WG lowering for multimem_load_reduce.

Commit

21 days ago

[Mosaic GPU] Add pallas and dialect WG lowering for multimem_load_reduce. PiperOrigin-RevId: 899526327

References

#36630 - [Mosaic GPU] Add pallas and dialect WG lowering for multimem_load_reduce.

#37071 - [ROCm] Switch to hermetic llvm

#37085 - Upgrade upstream ROCm CI from 7.2.0 to 7.2.2

#37186 - [ROCm] aiter mha kernels (ASM+CK) integration (#747)

#37198 - [ROCm] Introduce shim script for rocm jax upstream CI

Author

golechwierowicz

golechwierowicz

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading