[Misc] Rework process titles (#21780)

Signed-off-by: Nick Hill <nhill@redhat.com>
This commit is contained in:
Nick Hill
2025-07-29 06:14:47 +01:00
committed by GitHub
parent f1e2c095ec
commit 7234fe2685
7 changed files with 48 additions and 26 deletions

View File

@@ -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)]