Skip to content

Commit a8253b9

Browse files
authored
chore: remove duplicate should_stop_processing check (#6242)
Signed-off-by: junq <[email protected]>
1 parent 83c3ed1 commit a8253b9

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
@@ -653,7 +653,7 @@ def _executor_loop_pp(self):
653653
with self._profiler() as profile_step:
654654
iter_start_time = time.time()
655655
iter_stats = None
656-
while not self.should_stop_processing:
656+
while True:
657657
profile_step()
658658
if self.enable_iter_perf_stats:
659659
iter_start_time = time.time()
@@ -811,7 +811,7 @@ def _executor_loop(self):
811811
sample_state = None
812812
iter_start_time = time.time()
813813
iter_stats = None
814-
while not self.should_stop_processing:
814+
while True:
815815
profile_step()
816816
if self.enable_iter_perf_stats:
817817
iter_start_time = time.time()
@@ -955,7 +955,7 @@ def _executor_loop_overlap(self):
955955
with self._profiler() as profile_step:
956956
iter_start_time = time.time()
957957
iter_stats = None
958-
while not self.should_stop_processing:
958+
while True:
959959
profile_step()
960960
if self.enable_iter_perf_stats:
961961
iter_start_time = time.time()

0 commit comments

Comments
 (0)