Fix immediate profiler step range boundary (#30215)

Co-authored-by: weireweire <20922698+weireweire@users.noreply.github.com>
2026-07-09 19:47:23 -07:00
parent 23390589f7
commit 5e3dc5dd5f
1 changed files with 1 additions and 1 deletions
@@ -144,7 +144,7 @@ class SchedulerProfilerManager:
                    self.profiler_start_forward_ct + num_steps
                )
            else:
-                self.profiler_target_forward_ct = self.get_forward_ct() + num_steps
+                self.profiler_target_forward_ct = self.get_forward_ct() + num_steps + 1
            # The caller will be notified when reaching profiler_target_forward_ct
        else:
            self.profiler_target_forward_ct = None