jax
55830196 - [Pallas/Mosaic GPU] Make `collective_matmul_mgpu` use warpgroup lowering semantics.

Commit

147 days ago

[Pallas/Mosaic GPU] Make `collective_matmul_mgpu` use warpgroup lowering semantics. There is no meaningful performance difference. PiperOrigin-RevId: 918058564

References

#37809 - [Pallas/Mosaic GPU] Make `collective_matmul_mgpu` use warpgroup lowering semantics.

Author

bchetioui

Committer

Google-ML-Automation

Parents

44bdc919

jax 55830196 - [Pallas/Mosaic GPU] Make `collective_matmul_mgpu` use warpgroup lowering semantics.

jax
55830196 - [Pallas/Mosaic GPU] Make `collective_matmul_mgpu` use warpgroup lowering semantics.