Skip to content

Commit 03654c0

Browse files
authored
fix localhost measurement (#4320)
1 parent 45b08f0 commit 03654c0

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

colossalai/cli/launcher/hostinfo.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,8 @@ def is_host_localhost(hostname: str, port: str = None) -> None:
4646
localhost = socket.gethostname()
4747
localaddrs = socket.getaddrinfo(localhost, port)
4848
targetaddrs = socket.getaddrinfo(hostname, port)
49-
for (family, socktype, proto, canonname, sockaddr) in localaddrs:
50-
for (rfamily, rsocktype, rproto, rcanonname, rsockaddr) in targetaddrs:
51-
if rsockaddr[0] == sockaddr[0]:
52-
return True
53-
return False
49+
50+
return localaddrs == targetaddrs
5451

5552
def __str__(self):
5653
return f'hostname: {self.hostname}, port: {self.port}'

0 commit comments

Comments
 (0)