Robert Suderman 499a4e733c Expose ml_program dialect for MLIR builder ...

We now have an ml_program dialect that describes global variables
including load and store operations. Expose this dialect to allow
exporting variables and constants.

2022-06-28 20:29:41 +00:00

..

dialects

Expose ml_program dialect for MLIR builder

2022-06-28 20:29:41 +00: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