mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic
History
Adam Paszke a184b5e4af [Mosaic] Add a reinterpret cast for memrefs 
This allows us to override the inferred tiling of the values, which makes it possible to
e.g. preswizzle the data into a more efficient format before the kernel.

PiperOrigin-RevId: 553402946
2023-08-03 01:50:19 -07:00
..
dialect/tpu
[Mosaic] Add a reinterpret cast for memrefs
2023-08-03 01:50:19 -07:00
python
[Mosaic] Add support for sublane strided store.
2023-07-31 14:40:27 -07:00
BUILD
Enable building jaxlib w/ Mosaic
2023-07-26 03:59:30 -07:00