Tooling

Tooling#

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.

—
Contents