How to convert hlo.pb to hlo text? about xla HOT 3 OPEN

core has hidden module xla_builder which has computation_from_module_proto and get_computation_hlo

import torch_xla.core.xla_builder as xb

fname = "model.hlo"
with open(fname, mode="rb") as f:
  comp = xb.computation_from_module_proto("foo", f.read())
  
print(xb.get_computation_hlo(comp))

BTW, Looks like xla_builder is hidden , at least dir(core) and "tab-tab" do not show it, but it still can be imported

import torch_xla.core as core
>>> dir(core)
['__builtins__', '__cached__', '__doc__', '__file__', '__loader__', '__name__', '__package__', '__path__', '__spec__', 'dynamo_bridge', 'xla_env_vars', 'xla_model']

from xla.

Looking at the JAX implementation it seems like we just need to expose a pybind(in https://github.com/pytorch/xla/blob/master/torch_xla/csrc/init_python_bindings.cpp) for XlaComputation, its constructor should be able to convert proto to HLO.

I don't know if we currently have bandwidth for that but we are happy to take contributions.

from xla.

oh nice finding. We added that python xla builder couple years ago but never really use it that much.

from xla.