snsd0805/gpu-provider
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: use host as IP

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-05-29 18:43:42 +08:00
parent b7372420bc
commit 061b0e504e
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/communication.py
View File

@ -11,7 +11,6 @@ class ClusterCommunicationModule():
self.client_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.client_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.host = host self.host = host
self.port = port self.port = port
self.IP = socket.gethostbyname(socket.gethostname())
self.node_manager = node_manager self.node_manager = node_manager
self.worker_conns = [] self.worker_conns = []
@ -66,7 +65,7 @@ class ClusterCommunicationModule():
def handle_command(self, data): def handle_command(self, data):
command, args = data.split() command, args = data.split()
if command == '[INFO]': if command == '[INFO]':
data = {'host': self.IP, 'GPU': self.node_manager.GPU, 'GPU_num': self.node_manager.GPU_num} data = {'host': self.host, 'GPU': self.node_manager.GPU, 'GPU_num': self.node_manager.GPU_num}
self.client_sock.send(json.dumps(data).encode()) self.client_sock.send(json.dumps(data).encode())
return True return True
@ -118,7 +117,6 @@ class ServiceExplorationModule():
self.host = host self.host = host
self.port = port self.port = port
self.IP = socket.gethostbyname(socket.gethostname())
self.node_manager = node_manager self.node_manager = node_manager
@ -131,7 +129,7 @@ class ServiceExplorationModule():
if self.node_manager.status == 'none': if self.node_manager.status == 'none':
if data.decode() == '[EXPLORE]': if data.decode() == '[EXPLORE]':
self.sock.sendto(self.IP.encode(), addr) self.sock.sendto(self.host.encode(), addr)
def explore(self): def explore(self):
available_host = [] available_host = []
@ -145,7 +143,7 @@ class ServiceExplorationModule():
while True: while True:
try: try:
data, addr = client_sock.recvfrom(1024) data, addr = client_sock.recvfrom(1024)
if addr[0] != self.IP: if addr[0] != self.host:
available_host.append(addr[0]) available_host.append(addr[0])
except: except:
# if socket timeout # if socket timeout