This is a ParaGraph bridges that organizes support for HLO files dumped by XLA.
The C++ projects use Bazel for building binaries. To install Bazel, follow the directions at here. You need bazel-3.7.2 and a bit of patience. The bridge requires to compile good chunk of TensorFlow repo for XLA support. Use hte following command to build and testt the project
bazel test -c opt ...
To build the bridge without linter check, you can use
bazel build bridge:hlo_bridge
To dump files from the XLA-supported project, e.g. TensorFlow, JAX, or PyTorch-XLA, use XLA flags
export XLA_FLAGS="--xla_dump_to=./hlo_files --xla_dump_hlo_as_text=true "
We provide some JAX examples as a reference point.
Use ./bazel-bin/bridge/hlo_bridge --help to see available options.