Update logger.py

hariharan-devarajan · Aug 10, 2023 · 19c77e2 · 19c77e2
1 parent b808ed4
commit 19c77e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlio_profiler/logger.py b/dlio_profiler/logger.py
@@ -156,7 +156,7 @@ def wrapper(*args, **kwargs):
     def iter(self, func, iter_name="step"):
         if DLIO_PROFILER_ENABLED:
             self._arguments[iter_name] = 1
-            name = f"{inspect.stack()[1].function}.iter"
+            name = f"{self._name}.iter"
             kernal_name = f"{inspect.stack()[1].function}"
             start = dlio_logger.get_instance().get_time()
         for v in func: