Polish the log of executing matmul ops. (#1117)

intelligent-machine-learning · May 13, 2024 · 2160cdc · 2160cdc
1 parent 79d1ffc
commit 2160cdc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dlrover/trainer/torch/node_check/utils.py b/dlrover/trainer/torch/node_check/utils.py
@@ -40,8 +40,9 @@ def wrapper(*args, **kwargs):
         func(*args, **kwargs)
         t = round(time.time() - start, 2)
         local_rank = int(os.environ["LOCAL_RANK"])
+        func_name = func.__name__
         logger.info(
-            f"Time to execute {func} on local rank {local_rank} is {t}."
+            f"Time to execute {func_name} on local rank {local_rank} is {t}s."
         )
 
     return wrapper