mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic/dialect
History
Tzu-Wei Sung e235fb9760 [Mosaic] Allow part of x2 int casts. 
This should at least allow int2 -> int4 for native tiling vregs. Skip many tests due to XLA compatibility.

PiperOrigin-RevId: 736710186
2025-03-13 18:57:36 -07:00
..
gpu
[Mosaic GPU] Replace WGMMAFragLayout with TiledLayout in the mlir dialect and use it in layout inference.
2025-03-11 13:50:42 -07:00
tpu
[Mosaic] Allow part of x2 int casts.
2025-03-13 18:57:36 -07:00