fix format

pytorch · Oct 10, 2024 · ec68dea · ec68dea
1 parent f0e04be
commit ec68dea
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/torchbenchmark/util/experiment/metrics.py b/torchbenchmark/util/experiment/metrics.py
@@ -4,16 +4,18 @@
 
 import copy
 import dataclasses
+import os
 import pathlib
 import time
 from typing import List, Optional, Tuple, Union
 
+import psutil
+
 import torch
+
 from torchbenchmark import ModelTask
 from torchbenchmark.util.experiment.instantiator import TorchBenchModelConfig
 from torchbenchmark.util.model import BenchmarkModel
-import psutil
-import os
 
 WARMUP_ROUNDS = 10
 BENCHMARK_ITERS = 15
@@ -32,10 +34,12 @@ class TorchBenchModelMetrics:
     pt2_graph_breaks: Optional[float]
     model_flops: Optional[float]
 
+
 def maybe_synchronize(device: str):
     if device == "cuda":
         torch.cuda.synchronize()
 
+
 def get_latencies(
     func, device: str, nwarmup=WARMUP_ROUNDS, num_iter=BENCHMARK_ITERS
 ) -> List[float]:
@@ -84,8 +88,12 @@ def get_peak_memory(
         from torchbenchmark._components.model_analyzer.TorchBenchAnalyzer import (
             ModelAnalyzer,
         )
+
         mem_model_analyzer = ModelAnalyzer(
-            export_metrics_file, new_metrics_needed, metrics_gpu_backend, cpu_monitored_pid
+            export_metrics_file,
+            new_metrics_needed,
+            metrics_gpu_backend,
+            cpu_monitored_pid,
         )
     else:
         mem_model_analyzer = None