XLA HLO graph dump

harishch4 · January 15, 2021, 4:21am

Is there a way to dump XLA HLO graph to file which can be send as a input to XLA compiler?

mehdi_amini · January 15, 2021, 4:26am

Are you trying to use MHLO or directly XLA independently of MLIR? (this forum is about the MLIR ecosystem)

harishch4 · January 15, 2021, 4:27am

I’m interested in using MHLO.

mehdi_amini · January 15, 2021, 4:29am

Looking at the tests in the repo may be the best way to get started, unfortunately we are still missing a good end-to-end story there. We have some project to plumb thing these days with IREE (and possibly np-comp / JAX). I can try to write some docs about all this next week!

harishch4 · January 15, 2021, 4:44am

That would be really helpful, Thanks!