.

Author: hylin2002

src/benchmark_utils.py

@@ -101,7 +101,7 @@ def get_metrics_from_trace(trace: dict[str, Any], task: str) -> list[float]:
 
 def get_metrics_from_trace_tpu(trace: dict[str, Any], task: str) -> list[float]:
     event_matcher = re.compile(task)
-
+    print("Extracting time from TPU trace...")
     if "traceEvents" not in trace:
         raise KeyError("Key 'traceEvents' not found in trace.")