jax authors 1471702adc [Mosaic TPU] Support 1D concat: set implicit_dim to kSecondMinor to treat 1D (N,) as (1, N) and then tile it as (1, 128) ...

PiperOrigin-RevId: 696870258

2024-11-15 06:41:57 -08:00

..

dialect

[Mosaic TPU] Support 1D concat: set implicit_dim to kSecondMinor to treat 1D (N,) as (1, N) and then tile it as (1, 128)

2024-11-15 06:41:57 -08:00

gpu

[mosaic_gpu] The profiler now uses FFI calls for creating events and computing elapsed time

2024-11-12 11:01:59 -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] Add extension files for infer/apply vector layout.

2024-10-31 11:08:37 -07:00