Jevin Jiang 59e944dadf [XLA:Mosaic] Pass rewrite ctx of apply-vector-layout pass to relayout function. ...

We will implement a more efficient relayout according to the configs in rewrite ctx, such as `hardware_generation`, `max_sublanes_in_scratch` and so on. So it makes sense to change the relayout interface to take ctx (including python bindings). Now we can define rewrite ctx in `apply_vector_layout_test` as well. It makes it easier to test some advanced stuff (eg., mxu_shape change, max_sublanes_in_scratch change for rotate and relayout).

PiperOrigin-RevId: 655350013

2024-07-23 16:50:45 -07:00

..

_mlir_libs

[XLA:Mosaic] Pass rewrite ctx of apply-vector-layout pass to relayout function.

2024-07-23 16:50:45 -07:00

BUILD.bazel

#sdy Initial set of changes to allow for lowering to the Shardy dialect.

2024-07-23 05:32:06 -07:00