Skip to content

GitLab

"examples/tutorial/sequence_parallel/train.py" did not exist on "d9bf83e084dd82daaa05cd68e58fe66a24fd5a23"
Unverified Commit 03654c0c authored by LuGY's avatar LuGY Committed by GitHub
Browse files

fix localhost measurement (#4320)

parent 45b08f08
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
colossalai/cli/launcher/hostinfo.py
View file @ 03654c0c
...@@ -46,11 +46,8 @@ class HostInfo: ...@@ -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}'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment