jax

875099b2 - [Mosaic GPU] Enable the new transform inference pass in the warpgroup lowering.

[Mosaic GPU] Enable the new transform inference pass in the warpgroup lowering. A couple of dummy transform inference rules needed to be added in order to contend with parts of the lowering that do not use the dialect yet, along with a transform inference rule for `memref.view`. PiperOrigin-RevId: 738089782