[Misc] Rework process titles (#21780)
Signed-off-by: Nick Hill <nhill@redhat.com>
This commit is contained in:
@@ -10,11 +10,10 @@ import zmq
|
||||
|
||||
from vllm.config import ParallelConfig
|
||||
from vllm.logger import init_logger
|
||||
from vllm.utils import get_mp_context, make_zmq_socket
|
||||
from vllm.utils import get_mp_context, make_zmq_socket, set_process_title
|
||||
from vllm.v1.engine import EngineCoreOutputs, EngineCoreRequestType
|
||||
from vllm.v1.serial_utils import MsgpackDecoder
|
||||
from vllm.v1.utils import (bind_process_name, get_engine_client_zmq_addr,
|
||||
shutdown)
|
||||
from vllm.v1.utils import get_engine_client_zmq_addr, shutdown
|
||||
|
||||
logger = init_logger(__name__)
|
||||
|
||||
@@ -119,7 +118,7 @@ class DPCoordinatorProc:
|
||||
def __init__(self,
|
||||
engine_count: int,
|
||||
min_stats_update_interval_ms: int = 100):
|
||||
bind_process_name(self.__class__.__name__)
|
||||
set_process_title("DPCoordinator")
|
||||
self.ctx = zmq.Context()
|
||||
|
||||
self.engines = [EngineState() for _ in range(engine_count)]
|
||||
|
||||
Reference in New Issue
Block a user