Benjamin Chetioui
ba2f7c9ad9
[Mosaic GPU] Add transform inference rule for mgpu.slice_smem.
...
PiperOrigin-RevId: 737957778
2025-03-18 04:53:54 -07:00
Benjamin Chetioui
9a686e0bf3
[Mosaic GPU] Add initial transform inference rules for vector.{load,store}.
...
PiperOrigin-RevId: 737703568
2025-03-17 12:08:07 -07:00
Benjamin Chetioui
d09df7c8ab
[Mosaic GPU] Add transform inference rules for mgpu.async_{load,store}.
...
PiperOrigin-RevId: 736795784
2025-03-14 02:37:55 -07:00
Benjamin Chetioui
d028354abb
[Mosaic GPU] Introduce an initial transform inference pass.
...
For now, propagate transforms for `wgmma`. We do not handle `transpose` for
either operand yet.
The pass isn't called anywhere yet.
PiperOrigin-RevId: 736758754
2025-03-13 23:22:59 -07:00
