From 8e7e7e78994e21778d803ac724a302100961ad7b Mon Sep 17 00:00:00 2001 From: Seyyed Hossein Hasanpour Date: Sat, 8 Feb 2020 22:18:16 +0330 Subject: [PATCH] Fixed process termination routine in windows (#4844) * Fixed process termination routine in windows addresses and Fixes AttributeError: module 'os' has no attribute 'killpg' error in #4821 * Update server.py --- python/tvm/rpc/server.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/tvm/rpc/server.py b/python/tvm/rpc/server.py index ea9c2da31e996..ea677350d171f 100644 --- a/python/tvm/rpc/server.py +++ b/python/tvm/rpc/server.py @@ -403,7 +403,10 @@ def terminate(self): """Terminate the server process""" if self.use_popen: if self.proc: - os.killpg(self.proc.pid, signal.SIGTERM) + if platform.system() == "Windows": + os.kill(self.proc.pid, signal.CTRL_C_EVENT) + else: + os.killpg(self.proc.pid, signal.SIGTERM) self.proc = None else: if self.proc: