EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix scheduler proctitle suffix is ​​None (#4326)

Browse Source 
Co-authored-by: wenfeng.wf <wenfeng.wf@alibaba-inc.com>
This commit is contained in:
文峰
2025-03-13 10:29:35 +08:00
committed by GitHub
parent e35a93fa8a
commit c550e52f8b
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
python/sglang/srt/managers/scheduler.py
View File

@@ -1677,9 +1677,16 @@ def run_scheduler_process(
dp_rank: Optional[int], dp_rank: Optional[int],
pipe_writer, pipe_writer,
): ):
# Generate the prefix
if dp_rank is None:
prefix = f" TP{tp_rank}"
else:
prefix = f" DP{dp_rank} TP{tp_rank}"
# Config the process # Config the process
# kill_itself_when_parent_died() # This is disabled because it does not work for `--dp 2` # kill_itself_when_parent_died() # This is disabled because it does not work for `--dp 2`
setproctitle.setproctitle(f"sglang::scheduler_{dp_rank}") setproctitle.setproctitle(f"sglang::scheduler{prefix.replace(' ', '_')}")
faulthandler.enable() faulthandler.enable()
parent_process = psutil.Process().parent() parent_process = psutil.Process().parent()
@@ -1688,10 +1695,6 @@ def run_scheduler_process(
dp_rank = int(os.environ["SGLANG_DP_RANK"]) dp_rank = int(os.environ["SGLANG_DP_RANK"])
# Configure the logger # Configure the logger
if dp_rank is None:
prefix = f" TP{tp_rank}"
else:
prefix = f" DP{dp_rank} TP{tp_rank}"
configure_logger(server_args, prefix=prefix) configure_logger(server_args, prefix=prefix)
suppress_other_loggers() suppress_other_loggers()