mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-02-18 16:55:08 +08:00
[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:
parent
2b4ef3a014
commit
17e6062690
@ -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)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user