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

fix localhost measurement (#4320)

Browse Source
This commit is contained in:
LuGY 2023-08-01 10:14:00 +08:00 committed by GitHub
parent 45b08f08cb
commit 03654c0ce2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
colossalai/cli/launcher/hostinfo.py
View File

@ -46,11 +46,8 @@ class HostInfo:
localhost = socket.gethostname()
localaddrs = socket.getaddrinfo(localhost, port)
targetaddrs = socket.getaddrinfo(hostname, port)
for (family, socktype, proto, canonname, sockaddr) in localaddrs:
for (rfamily, rsocktype, rproto, rcanonname, rsockaddr) in targetaddrs:
if rsockaddr[0] == sockaddr[0]:
return True
return False
return localaddrs == targetaddrs
def __str__(self):
return f'hostname: {self.hostname}, port: {self.port}'