github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix localhost measurement (#4320)

pull/4359/head
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() localhost = socket.gethostname()
localaddrs = socket.getaddrinfo(localhost, port) localaddrs = socket.getaddrinfo(localhost, port)
targetaddrs = socket.getaddrinfo(hostname, port) targetaddrs = socket.getaddrinfo(hostname, port)
for (family, socktype, proto, canonname, sockaddr) in localaddrs:
for (rfamily, rsocktype, rproto, rcanonname, rsockaddr) in targetaddrs: return localaddrs == targetaddrs
if rsockaddr[0] == sockaddr[0]:
return True
return False
def __str__(self): def __str__(self):
return f'hostname: {self.hostname}, port: {self.port}' return f'hostname: {self.hostname}, port: {self.port}'