mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic
History
jax authors a6fcb7415f [TPU][Mosaic][Easy] Add verification for AssumeMultipleOp. 
A user must use AssumeMultipleOp to annotate integer constants that are divisible by the given multiple.

PiperOrigin-RevId: 727699186
2025-02-16 21:16:05 -08:00
..
dialect
[TPU][Mosaic][Easy] Add verification for AssumeMultipleOp.
2025-02-16 21:16:05 -08:00
gpu
[Mosaic GPU] Handle the swizzle attribute in the lowering of async_store and async_load
2025-01-27 05:18:16 -08:00
python
Fix OSS build for the Mosaic GPU dialect
2025-01-02 15:55:03 +00:00
BUILD
[Pallas] Reductions with replicated axes.
2025-02-15 07:41:16 -08:00
serde.cc
[mosaic] Extracted serialization pass traversal logic into a reusable function
2025-01-16 02:58:06 -08:00
serde.h
[mosaic] Extracted serialization pass traversal logic into a reusable function
2025-01-16 02:58:06 -08:00