From 02438ee63582a107ee08f4b09460b1e24b6c0aad Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 9 Nov 2020 15:49:37 +0800 Subject: [PATCH] refactor(mge/distributed): use thread.Threading to create Server GitOrigin-RevId: fc994411bf775566284e9caada79d3dc40890b4b --- imperative/python/megengine/distributed/server.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/imperative/python/megengine/distributed/server.py b/imperative/python/megengine/distributed/server.py index 581a85c81..e7875851b 100644 --- a/imperative/python/megengine/distributed/server.py +++ b/imperative/python/megengine/distributed/server.py @@ -6,7 +6,6 @@ # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -import multiprocessing as mp import threading import time from collections import defaultdict @@ -142,7 +141,7 @@ class Server: def __init__(self, port): self.py_server_port = get_free_ports(1)[0] if port == 0 else port self.mm_server_port = create_mm_server("0.0.0.0", 0) - self.proc = mp.Process( + self.proc = threading.Thread( target=start_server, args=(self.py_server_port, self.mm_server_port), daemon=True, -- GitLab