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
apache · Feb 8, 2020 · e7a3c47 · e7a3c47
1 parent e33de24
commit e7a3c47
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: