diff --git a/profiler/msprof_analyze/compare_tools/compare_backend/profiling_parser/base_profiling_parser.py b/profiler/msprof_analyze/compare_tools/compare_backend/profiling_parser/base_profiling_parser.py
index fa24a844c24d709b3eaaffa606abd4605be9abc1..bb7dfe7d90a543f36d6bcf0e4705cd00eef7dffb 100644
--- a/profiler/msprof_analyze/compare_tools/compare_backend/profiling_parser/base_profiling_parser.py
+++ b/profiler/msprof_analyze/compare_tools/compare_backend/profiling_parser/base_profiling_parser.py
@@ -124,7 +124,8 @@ class BaseProfilingParser(ABC):
         if self._step_id == Constant.VOID_STEP:
             return self._step_range
         step_list = []
-        for event in self._result_data.torch_op_data:
+        events = self._result_data.torch_op_data or self._trace_event_generator(self._profiling_type)
+        for event in events:
             if event.is_step_profiler():
                 step_id = event.name.split("#")[-1]
                 step_list.append(step_id)