From 27521dc766b0a17369ba1d1fa462c645c822a0db Mon Sep 17 00:00:00 2001 From: wffpy Date: Tue, 7 Jan 2025 11:37:33 +0800 Subject: [PATCH] fix logger without start --- python/module_logging/perf/logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/module_logging/perf/logger.py b/python/module_logging/perf/logger.py index 6d0faf2..fe10164 100644 --- a/python/module_logging/perf/logger.py +++ b/python/module_logging/perf/logger.py @@ -188,7 +188,7 @@ def __torch_dispatch__(self, op, types, args=(), kwargs=None): if self.enable_profiling: torch.cuda.synchronize() # insert pre-op delimiter - # print("[START_SYMBOL]: {} ns".format(str(op)), flush=True) + print("[START_SYMBOL]: {} ns".format(str(op)), flush=True) # if cpp_extend: # from .. import Hook # print("{} start at: {}".format(str(op), Hook.get_current_time()))