get_backend_colors() — pytorch Function Reference

Architecture documentation for the get_backend_colors() function in utils.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration called by 1

Entity Profile

DynamoBenchmarks→ Configuration→ get_backend_colors() — pytorch Function Reference

Dependency Diagram

graph TD
  ee2a0377_fc0a_6dcc_ff4c_9e0e6201c7cc["get_backend_colors()"]
  3d2c6e7f_0e61_a579_1388_3d494d9ef856["visualize_comparison()"]
  3d2c6e7f_0e61_a579_1388_3d494d9ef856 -->|calls| ee2a0377_fc0a_6dcc_ff4c_9e0e6201c7cc
  style ee2a0377_fc0a_6dcc_ff4c_9e0e6201c7cc fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/utils.py lines 228–240

def get_backend_colors() -> dict[str, str]:
    """Get consistent color scheme for different backends."""
    return {
        "eager": "#1f77b4",  # blue
        "compiled": "#ff7f0e",  # orange
        "quack": "#2ca02c",  # green
        "liger": "#d62728",  # red
        "helion": "#9467bd",  # purple
        "triton": "#8c564b",  # brown
        "cutlass": "#e377c2",  # pink
        "flash_attn": "#7f7f7f",  # gray
        "default": "#000000",  # black
    }