[mmf][search] Leverage the TrainerProfilingMixin to figure out the de…

…tailed latency breakdown

Differential Revision: D46013756

fbshipit-source-id: d01a6668b40e77c3adaddf0e93a7948da94657d8

mmf/trainers/core/profiling.py

@@ -1,6 +1,7 @@
 # Copyright (c) Facebook, Inc. and its affiliates.
 
 import logging
+import threading
 from abc import ABC
 from typing import Type
 
@@ -16,5 +17,7 @@ class TrainerProfilingMixin(ABC):
     def profile(self, text: str) -> None:
         if self.training_config.logger_level != "debug":
             return
-        logging.debug(f"{text}: {self.profiler.get_time_since_start()}")
+        logging.debug(
+            f"tid={threading.current_thread().ident}, {text}: {self.profiler.get_time_since_start()}"
+        )
         self.profiler.reset()