Fixed process termination routine in windows (apache#4844)

* Fixed process termination routine in windows addresses and Fixes AttributeError: module 'os' has no attribute 'killpg' error in apache#4821 * Update server.py
neo-ai · Mar 2, 2020 · f54865f · f54865f
1 parent eb97328
commit f54865f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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: