trinity.perf package

Submodules

• trinity.perf.report_metrics module
  • compute_global_token_throughput_metrics()
• trinity.perf.report_viewer module
  • PerfReportViewer
    • PerfReportViewer.run_viewer()
  • launch_report_viewer()
  • has_streamlit_context()
  • configure_streamlit_page()
  • parse_args()
  • load_report()
  • infer_module_name()
  • get_step_metric_prefixes()
  • format_timestamp()
  • format_metric_value()
  • metric_label()
  • gpu_series_label()
  • render_metric_card()
  • build_elapsed_series()
  • build_scalar_timeline_series()
  • build_gpu_timeline_series()
  • render_line_chart()
  • render_step_metric_chart()
  • render_header()
  • compute_global_token_throughput_metrics()
  • render_global_metrics()
  • render_step_metrics()
  • render_resource_utilization()
  • main()
• trinity.perf.resource_backends module
  • GPUSample
    • GPUSample.gpu_id
    • GPUSample.name
    • GPUSample.gpu_util_percent
    • GPUSample.gpu_memory_used_mb
    • GPUSample.gpu_memory_total_mb
    • GPUSample.to_dict()
    • GPUSample.__init__()
  • ResourceSample
    • ResourceSample.timestamp
    • ResourceSample.cpu_percent
    • ResourceSample.memory_rss_mb
    • ResourceSample.memory_percent
    • ResourceSample.gpu_metrics
    • ResourceSample.to_dict()
    • ResourceSample.__init__()
  • SystemResourceBackend
    • SystemResourceBackend.__init__()
    • SystemResourceBackend.open()
    • SystemResourceBackend.close()
    • SystemResourceBackend.sample()
• trinity.perf.resource_sampler module
  • ResourceSampler
    • ResourceSampler.__init__()
    • ResourceSampler.start()
    • ResourceSampler.stop()
    • ResourceSampler.samples()
• trinity.perf.stage_perf module
  • ExplorerPerfOptions
    • ExplorerPerfOptions.config_path
    • ExplorerPerfOptions.output_path
    • ExplorerPerfOptions.monitor_interval
    • ExplorerPerfOptions.total_steps
    • ExplorerPerfOptions.timeout
    • ExplorerPerfOptions.__init__()
  • validate_explorer_perf_config()
  • build_explorer_perf_payload()
  • write_explorer_perf_output()
  • run_explorer_perf()
• trinity.perf.tensorboard_metrics module
  • TensorBoardScalarReader
    • TensorBoardScalarReader.__init__()
  • extract_raw_metrics_for_step()
  • collect_step_metrics()

Module contents

Performance tooling package for Trinity.

class trinity.perf.ExplorerPerfOptions(config_path: 'str', output_path: 'str', monitor_interval: 'float' = 2.0, total_steps: 'int' = 5, timeout: 'Optional[float]' = None)

Bases: object

config_path: str

output_path: str

monitor_interval: float

total_steps: int

timeout: float | None

__init__(config_path: str, output_path: str, monitor_interval: float = 2.0, total_steps: int = 5, timeout: float | None = None) → None

class trinity.perf.ResourceSampler(interval_seconds: float, backend: SystemResourceBackend | None = None)

Bases: object

Periodically collect system resource samples in a background thread.

__init__(interval_seconds: float, backend: SystemResourceBackend | None = None) → None

samples() → list[ResourceSample]

Return a snapshot of all collected samples.

start() → None

Start sampling in the background.

stop() → list[ResourceSample]

Stop sampling and return the collected samples.

class trinity.perf.TensorBoardScalarReader(log_dir: str)

Bases: object

Read scalar metrics from TensorBoard event files.

__init__(log_dir: str)

trinity.perf.collect_step_metrics(metric_map: dict[str, dict[int, float]]) → list[dict[str, Any]]

Build per-step metrics from TensorBoard scalars.

trinity.perf.run_explorer_perf(options: ExplorerPerfOptions) → dict[str, Any]

Run Explorer perf collection and return the result payload.

trinity.perf.write_explorer_perf_output(output_path: str, payload: dict[str, Any]) → None

Write the final payload to disk.