Tomás Longeri 4452960947 [Mosaic:TPU] In infer ext rule, avoid assigning offsets outside of dst first tile ...

Note that offsets outside of first tile are still disabled (for both infer and apply), and once we support it we will want to assign offsets differently, this is mostly to avoid assigning invalid layouts (that may not just be outside the first tile, but outside the vreg slice)

PiperOrigin-RevId: 709168368

2024-12-23 15:49:39 -08:00

..

dialect

[Mosaic:TPU] In infer ext rule, avoid assigning offsets outside of dst first tile

2024-12-23 15:49:39 -08:00

gpu

[Mosaic GPU] Add end-to-end lowering example for a pointwise kernel using the dialect and layout inference.

2024-12-17 09:28:05 -08: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_gpu] Include Mosaic GPU dialect fiels into jaxlib

2024-12-23 13:46:25 +00:00