Skip to content

Commit acb54ca

Browse files
authored
Intialize io_thread_pool attribute in the beginning. (#18331)
Signed-off-by: rabi <[email protected]>
1 parent 6e0fd34 commit acb54ca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/v1/executor/multiproc_executor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ def _init_executor(self) -> None:
5050
self.is_failed = False
5151
self.shutdown_event = threading.Event()
5252
self.failure_callback: Optional[FailureCallback] = None
53+
self.io_thread_pool: Optional[ThreadPoolExecutor] = None
5354

5455
self.world_size = self.parallel_config.world_size
5556
tensor_parallel_size = self.parallel_config.tensor_parallel_size
@@ -107,7 +108,6 @@ def _init_executor(self) -> None:
107108

108109
# For pipeline parallel, we use a thread pool for asynchronous
109110
# execute_model.
110-
self.io_thread_pool: Optional[ThreadPoolExecutor] = None
111111
if self.max_concurrent_batches > 1:
112112
# Note: must use only 1 IO thread to keep dequeue sequence
113113
# from the response queue

0 commit comments

Comments
 (0)