mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 20:26:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic
History
Dimitar (Mitko) Asenov f57d603c45 [Mosaic GPU] Simplify enums in the MLIR Mosaic GPU dialect. 
This enables us to use them more simply in the current and upcoming Python code. The Python bindings for enum and enum attributes leave much to be desired.

PiperOrigin-RevId: 718795667
2025-01-23 03:38:26 -08:00
..
dialect
[Mosaic GPU] Simplify enums in the MLIR Mosaic GPU dialect.
2025-01-23 03:38:26 -08:00
gpu
Remove CUDA rpaths from jaxlib build.
2025-01-17 17:09:30 -08:00
python
Fix OSS build for the Mosaic GPU dialect
2025-01-02 15:55:03 +00:00
BUILD
[mosaic] Extracted serialization pass traversal logic into a reusable function
2025-01-16 02:58:06 -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