Skip to content

T
tic
Commit 01e53935 by Haozheng Fan Committed by Tianqi Chen
Options

hide psutil (#4013)

parent d7998d39
Showing with 1 additions and 1 deletions
python/tvm/rpc/server.py
...@@ -39,7 +39,6 @@ import subprocess ...@@ -39,7 +39,6 @@ import subprocess
import time import time
import sys import sys
import signal import signal
import psutil
from .._ffi.function import register_func from .._ffi.function import register_func
from .._ffi.base import py_str from .._ffi.base import py_str
...@@ -211,6 +210,7 @@ def _listen_loop(sock, port, rpc_key, tracker_addr, load_library, custom_addr): ...@@ -211,6 +210,7 @@ def _listen_loop(sock, port, rpc_key, tracker_addr, load_library, custom_addr):
server_proc.join(opts.get("timeout", None)) server_proc.join(opts.get("timeout", None))
if server_proc.is_alive(): if server_proc.is_alive():
logger.info("Timeout in RPC session, kill..") logger.info("Timeout in RPC session, kill..")
import psutil
parent = psutil.Process(server_proc.pid) parent = psutil.Process(server_proc.pid)
# terminate worker childs # terminate worker childs
for child in parent.children(recursive=True): for child in parent.children(recursive=True):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment