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
alexwong · Feb 26, 2020 · 8e7e7e7 · 8e7e7e7
1 parent 812b612
commit 8e7e7e7
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: