mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 20:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic
History
Sharad Vikram f08df0f0c3 [Pallas] Add Mosaic lowering for pow2 
PiperOrigin-RevId: 559291137
2023-08-22 19:42:17 -07:00
..
dialect/tpu
[Mosaic] Relax return type checks for vector.contract
2023-08-04 13:34:12 -07:00
python
[Pallas] Add Mosaic lowering for pow2
2023-08-22 19:42:17 -07:00
BUILD
Enable building jaxlib w/ Mosaic
2023-07-26 03:59:30 -07:00