Skip to content

Commit 4c7769b

Browse files
QiJunelancelly
authored andcommitted
chore: remove duplicate should_stop_processing check (NVIDIA#6242)
Signed-off-by: junq <[email protected]> Signed-off-by: Lanyu Liao <[email protected]>
1 parent fe1c5fc commit 4c7769b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensorrt_llm/_torch/pyexecutor/py_executor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -643,7 +643,7 @@ def _executor_loop_pp(self):
643643
with self._profiler() as profile_step:
644644
iter_start_time = time.time()
645645
iter_stats = None
646-
while not self.should_stop_processing:
646+
while True:
647647
profile_step()
648648
if self.enable_iter_perf_stats:
649649
iter_start_time = time.time()
@@ -801,7 +801,7 @@ def _executor_loop(self):
801801
sample_state = None
802802
iter_start_time = time.time()
803803
iter_stats = None
804-
while not self.should_stop_processing:
804+
while True:
805805
profile_step()
806806
if self.enable_iter_perf_stats:
807807
iter_start_time = time.time()
@@ -945,7 +945,7 @@ def _executor_loop_overlap(self):
945945
with self._profiler() as profile_step:
946946
iter_start_time = time.time()
947947
iter_stats = None
948-
while not self.should_stop_processing:
948+
while True:
949949
profile_step()
950950
if self.enable_iter_perf_stats:
951951
iter_start_time = time.time()

0 commit comments

Comments
 (0)