jax authors eeca8d81b9 Fix example in mosaic tpu dialect layout.h ...

PiperOrigin-RevId: 629424833

2024-04-30 08:42:54 -07:00

..

dialect/tpu

Fix example in mosaic tpu dialect layout.h

2024-04-30 08:42:54 -07:00

gpu

[Mosaic GPU] Stop using the MLIR CUDA runtime

2024-04-29 08:04:51 -07:00

python

Fix some load orderings for buildifier

2024-03-27 10:28:57 -07:00

BUILD

Adds rewrite patterns to LinalgVectorizationPass to eliminate transfer_read and transfer_write ops.

2024-04-26 13:51:04 -07:00