!8680 Bug Fix For Step Trace Time

Merge pull request !8680 from 周先琪/master
Ascend · Dec 27, 2023 · c047c84 · c047c84
1 parent 2f3e003
commit c047c84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_npu/profiler/analysis/prof_view/trace_step_time_parser.py b/torch_npu/profiler/analysis/prof_view/trace_step_time_parser.py
@@ -34,7 +34,7 @@ def count_time(cls, addtype, addtime, durtime, step_list, save_time):
         addtime = float(addtime)
         durtime = float(durtime)
         for step in step_list:
-            if step[1] < addtime and step[2] > addtime:
+            if step[1] <= addtime <= step[2]:
                 cur_step = step[0]
                 break
         for step in step_list: