InfiniBench/dashboard/utils/visualizations/__init__.py at f4a8e4d9bf1a286db7a7cf0ef27db1adcce0f390 · InfiniTensor/InfiniBench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#!/usr/bin/env python3
"""Visualization functions for InfiniMetrics dashboard.

This package provides visualization utilities organized by test type:
- base: Common/legacy visualization functions
- hardware: Hardware test visualizations (memory sweep, cache bandwidth)
- communication: Communication test visualizations
- inference: Inference test visualizations
- operator: Operator test visualizations
- training: Training test visualizations
- summary_tables: Summary tables for different test types
"""

# Base functions (common)
from .base import (
    create_gauge_chart,
    plot_timeseries_auto,
)

# Communication functions
from .communication import (
    plot_metric_vs_size,
    plot_comparison_matrix,
)

# Inference functions
from .inference import (
    render_inference_metrics,
    render_memory_gauge,
)

# Summary tables
from .summary_tables import (
    create_summary_table_comm,
    create_summary_table_infer,
    create_summary_table_ops,
)

# Hardware functions
from .hardware import (
    create_summary_table_hw,
    plot_hw_mem_sweep,
    plot_hw_cache,
)

# Operator functions
from .operator import (
    extract_operator_metrics,
    render_operator_performance_charts,
)

# Training functions
from .training import (
    render_performance_curves,
    render_throughput_comparison,
    render_data_tables,
    render_config_details,
)

__all__ = [
    # Base
    "create_gauge_chart",
    "plot_timeseries_auto",
    # Communication
    "plot_metric_vs_size",
    "plot_comparison_matrix",
    # Inference
    "render_inference_metrics",
    "render_memory_gauge",
    # Summary tables
    "create_summary_table_comm",
    "create_summary_table_infer",
    "create_summary_table_ops",
    # Hardware
    "create_summary_table_hw",
    "plot_hw_mem_sweep",
    "plot_hw_cache",
    # Operator
    "extract_operator_metrics",
    "render_operator_performance_charts",
    # Training
    "render_performance_curves",
    "render_throughput_comparison",
    "render_data_tables",
    "render_config_details",
]