Merge pull request #17 from wffpy/dev0

fix logger without start
wffpy · Jan 7, 2025 · 83dcaa2 · 83dcaa2
2 parents 5859188 + 27521dc
commit 83dcaa2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()))