From 10d9d77e414986ce84561912745152a970aef19d Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 18 Jan 2021 14:20:56 +0800
Subject: [PATCH] refactor(mge/distributed): remove get_free_ports func

GitOrigin-RevId: b85c4885be74f3135c6707831dedf31b51356e8f
---
 .../python/megengine/distributed/__init__.py  |  1 -
 .../python/megengine/distributed/launcher.py  |  1 -
 .../python/megengine/distributed/server.py    |  1 -
 .../python/megengine/distributed/util.py      | 26 -------------------
 4 files changed, 29 deletions(-)
 delete mode 100644 imperative/python/megengine/distributed/util.py

diff --git a/imperative/python/megengine/distributed/__init__.py b/imperative/python/megengine/distributed/__init__.py
index 79993da9..818829b1 100644
--- a/imperative/python/megengine/distributed/__init__.py
+++ b/imperative/python/megengine/distributed/__init__.py
@@ -22,4 +22,3 @@ from .group import (
 from .helper import bcast_list_, make_allreduce_cb, synchronized
 from .launcher import launcher
 from .server import Client, Server
-from .util import get_free_ports
diff --git a/imperative/python/megengine/distributed/launcher.py b/imperative/python/megengine/distributed/launcher.py
index 31fceebc..f904babe 100644
--- a/imperative/python/megengine/distributed/launcher.py
+++ b/imperative/python/megengine/distributed/launcher.py
@@ -13,7 +13,6 @@ from ..core._imperative_rt.core2 import sync
 from .group import group_barrier, init_process_group
 from .helper import get_device_count_by_fork
 from .server import Client, Server
-from .util import get_free_ports
 
 
 def _run_wrapped(
diff --git a/imperative/python/megengine/distributed/server.py b/imperative/python/megengine/distributed/server.py
index 433c9017..c686fad2 100644
--- a/imperative/python/megengine/distributed/server.py
+++ b/imperative/python/megengine/distributed/server.py
@@ -17,7 +17,6 @@ from xmlrpc.server import SimpleXMLRPCServer
 
 from ..core._imperative_rt.utils import create_mm_server
 from ..utils.future import Future
-from .util import get_free_ports
 
 
 class Methods:
diff --git a/imperative/python/megengine/distributed/util.py b/imperative/python/megengine/distributed/util.py
deleted file mode 100644
index e6eaacee..00000000
--- a/imperative/python/megengine/distributed/util.py
+++ /dev/null
@@ -1,26 +0,0 @@
-# -*- coding: utf-8 -*-
-# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
-#
-# Copyright (c) 2014-2020 Megvii Inc. All rights reserved.
-#
-# 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 functools
-import socket
-from typing import List
-
-
-def get_free_ports(num: int) -> List[int]:
-    """
-    Get one or more free ports.
-    """
-    socks, ports = [], []
-    for i in range(num):
-        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        sock.bind(("", 0))
-        socks.append(sock)
-        ports.append(sock.getsockname()[1])
-    for sock in socks:
-        sock.close()
-    return ports
-- 
GitLab