XLA HLO Navigator
A lightweight VS Code extension for reading and exploring OpenXLA HLO files. It recognizes .hlo, .hlo.txt, and files named hlo.txt without requiring a language server.
Core features
- Syntax highlighting: highlights HLO modules, computations, instructions, opcodes, types, shapes, attributes, comments, and matching values. Unused outputs are shown in gray.
- Code navigation: Ctrl/Cmd+click jumps between values and their producers/consumers, or between computation definitions and call sites. Double-click highlights a value and all of its uses.
- HLO computation graph: Alt/Option+click a computation or value to open its graph in the bundled Model Explorer. You can also run
HLO: Show Computation Graph at Cursor or use Alt/Option+G.
- Compilation Artifact Navigator: press Ctrl+Shift+R on Windows/Linux or Cmd+Shift+R on macOS to explore the computation under the cursor. Compile with
xla_compile or load an existing pass-dump folder, then inspect HLO/MLIR passes, changed IR, diffs, codegen, and AOT artifacts.
All shortcuts can be changed in VS Code's Keyboard Shortcuts editor.
Install
code --install-extension xla-hlo-navigator-<version>.vsix
Open an HLO file after installation. Model Explorer is bundled in the extension, so no separate server or package is required.
For pass exploration, set an AOT-capable xla_compile path in the explorer. The expected build artifact is:
<XLA workspace>/bazel-bin/xla/service/xla_compile
CPU is the default target, CUDA is the second preset, and another compiler-supported target can be entered manually.
License
Apache-2.0. Bundled third-party components retain their original licenses.
