mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 00:56:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/lib/mlir
History
Aart Bik c1261ccd27 Adds a wrapper to sparse tensor dialect, as part of an 
an initial prototype of an alternate JAX compilation path
that emits the MLIR MHLO/CHLO dialects instead of classic XLA HLO
together with sparse tensor types.

PiperOrigin-RevId: 443438043
2022-04-21 11:48:44 -07:00
..
dialects
Adds a wrapper to sparse tensor dialect, as part of an
2022-04-21 11:48:44 -07:00
__init__.py
Adds an initial prototype of an alternate JAX compilation path that emits the MLIR MHLO/CHLO dialects instead of classic XLA HLO.
2021-11-11 06:37:12 -08:00