[https://nvbugs/5821433][fix] complete WAR for popen in QA env (#11214)

Signed-off-by: Ivy Zhang <25222398+crazydemo@users.noreply.github.com>
Signed-off-by: Wangshanshan <30051912+dominicshanshan@users.noreply.github.com>
This commit is contained in:
Ivy Zhang 2026-02-05 10:48:27 +08:00 committed by Yanchao Lu
parent 2b4ef3a014
commit 17e6062690

View File

@ -2,7 +2,6 @@ import asyncio
import os
import shutil
import subprocess
import sys
import tempfile
import traceback
import uuid
@ -187,14 +186,13 @@ def run_disagg_server(disagg_cluster_config, work_dir, port=0, save_log=False):
cmds = ["trtllm-serve", "disaggregated", "-c", disagg_server_config_path]
log_file = None
log_path = None
stdout = None
stderr = None
if save_log:
log_path = os.path.join(work_dir, "disagg_server.log")
log_file = open(log_path, "w+")
stdout = log_file
stderr = log_file
else:
stdout = sys.stdout
stderr = sys.stderr
p = subprocess.Popen(cmds, stdout=stdout, stderr=stderr)
return ProcessWrapper(p, log_file=log_file, log_path=log_path, port=port)