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.
| 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. | — |
- 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.
- Micro-benchmarking—running the benchmark suites the tools summarize.