Tooling
July 15, 2026 · View on GitHub
Primus ships a set of auxiliary tools for analysis, benchmarking, visualization, installation, and diagnostics. They live under tools/ in the repository; each tool keeps its own README with detailed usage instructions. This page is a lightweight index so the tooling is discoverable from the documentation.
Available tools
| Tool | Directory | What it does | Docs |
|---|---|---|---|
| IRLens | tools/IRLens/ | Parses XLA HLO text dumps and prints an execution skeleton with control flow, separating communication vs compute ops. | README |
| model_stats | tools/model_stats/ | Generates charts from the model config registry under primus/configs/models. | README |
| Pipeline Visualization | tools/visualization/pp_vis/ | Visualizes pipeline-parallelism schedules from dumped data or PP-simulator JSON via a local web UI. | README |
| Auto Benchmark | tools/auto_benchmark/ | Interactive benchmark menu for Megatron/TorchTitan on MI300X/MI355X with metrics collection. | README |
| Backend Gap Report / Engineering Dashboard | tools/backend_gap_report/ | Generation and publishing toolchain for the shared Primus engineering dashboard and backend-gap reports. | README |
| Installation (venv) | tools/installation/ | Reproduces the Primus training Docker environment in a Python virtual environment (no Docker, no sudo). | README |
| Daily Report | tools/daily/ | Benchmark summary CSV generation used by CI workflows. | — |
| Docker Helpers | tools/docker/ | Container startup and proxy scripts. | — |
| Profile Trace | tools/profile_trace/ | Trace-file merging utility. | — |
Related documentation
- Primus tools—the full catalog of Primus tools (CLI, tuning agent, ecosystem) with how-to starting points.
- Tools overview—the top-level index maintained alongside the code.
- Profiling and observability—how these tools fit into performance analysis.
- Benchmarking—running the benchmark suites the tools summarize.