Tomás Longeri 8163e74e45 [Mosaic:TPU] Add relayout for adding minor implicit dim and relax some offset restrictions on similar shape cast ...

This factors out some logic from the apply-vector-layout shape cast rule where we insert a minor dimension, relaxes some offset restrictions on it, and uses it for the relayout.

PiperOrigin-RevId: 702993092

2024-12-04 23:13:26 -08:00

..

dialect

[Mosaic:TPU] Add relayout for adding minor implicit dim and relax some offset restrictions on similar shape cast

2024-12-04 23:13:26 -08:00

gpu

[Mosaic GPU] Improve default kernel name and add option to customize

2024-12-02 22:22:11 -05:00

python

[MOSAIC:GPU] Add async_load, async_store, and supporting attributes to the MLIR Mosaic GPU Dialect.

2024-11-08 14:34:23 -08:00

BUILD

[Mosaic] Add target core type parameter to tpu.sem_signal

2024-11-19 13:40:13 -08:00