visualize() — pytorch Function Reference

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

Function python DynamoBenchmarks Configuration calls 1

Entity Profile

DynamoBenchmarks→ Configuration→ visualize() — pytorch Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/utils.py lines 196–203

    def visualize(self) -> None:
        device_name = torch.cuda.get_device_name(0)
        visualize_comparison(
            self.profiling_results,
            title=f"{self.name} ({device_name})",
            output_path=f"{self.name}_bench",
        )
        return