Adam Paszke e0415c1865 [Mosaic TPU] Don't fold the accumulator into matmul if it has multiple uses ...

PiperOrigin-RevId: 657967724

2024-07-31 05:19:52 -07:00

..

dialect/tpu

[Mosaic TPU] Don't fold the accumulator into matmul if it has multiple uses

2024-07-31 05:19:52 -07:00

gpu

Added argument validation to mosaic_gpu_init_tma_desc

2024-07-25 13:16:34 -07:00

python

[Pallas/Mosaic] Add support for returning values from run_scoped

2024-07-11 18:37:09 -07:00

BUILD

[XLA:Mosaic] Support memref shapecast.

2024-07-12 17:05:03 -07:00