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 716f4f8119 [Mosaic] Allow users to opt out of window prefetching 
PiperOrigin-RevId: 552797922
2023-08-01 07:36:59 -07:00
..
dialect/tpu
[Mosaic] Allow users to opt out of window prefetching
2023-08-01 07:36:59 -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