From 53f399ce0223adeee96ca671409b192f32e7760e Mon Sep 17 00:00:00 2001
From: lc_pro <gyang_lu@foxmail.com>
Date: Mon, 31 Jul 2023 17:59:38 +0800
Subject: [PATCH] fix localhost measurement

---
 colossalai/cli/launcher/hostinfo.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/colossalai/cli/launcher/hostinfo.py b/colossalai/cli/launcher/hostinfo.py
index d1b88b229fb8..2a6a111e4d72 100644
--- a/colossalai/cli/launcher/hostinfo.py
+++ b/colossalai/cli/launcher/hostinfo.py
@@ -46,11 +46,8 @@ def is_host_localhost(hostname: str, port: str = None) -> None:
         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}'