github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-26 12:14:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix port exception type (#2925)

Browse Source
This commit is contained in:
YH
2023-02-28 12:00:43 +09:00
committed by GitHub
parent 61e687831d
commit a848091141
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
colossalai/utils/common.py
View File

@@ -50,16 +50,20 @@ def ensure_path_exists(filename: str):
Path(dirpath).mkdir(parents=True, exist_ok=True) Path(dirpath).mkdir(parents=True, exist_ok=True)
def free_port(): def free_port() -> int:
"""Get a free port on localhost.
Returns:
int: A free port on localhost.
"""
while True: while True:
port = random.randint(20000, 65000)
try: try:
sock = socket.socket() with socket.socket() as sock:
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
port = random.randint(20000, 65000) sock.bind(("localhost", port))
sock.bind(('localhost', port)) return port
sock.close() except OSError:
return port
except Exception:
continue continue