Browse Source

refactor(mge/distributed): use thread.Threading to create Server

GitOrigin-RevId: fc994411bf
tags/v1.1.0
Megvii Engine Team 5 years ago
parent
commit
02438ee635
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      imperative/python/megengine/distributed/server.py

+ 1
- 2
imperative/python/megengine/distributed/server.py View File

@@ -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,


Loading…
Cancel
Save