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

fix: exit when receiving STOP signal (worker)

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-05-30 17:19:13 +08:00
parent 0cae46b25b
commit 0f7409a9c6
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/communication.py
View File

@ -89,10 +89,13 @@ class ClusterCommunicationModule():
data = {'host': self.host, '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())
elif command == '[STOP]': elif command == '[STOP]':
print("Receive STOP signal")
self.node_manager.docker_client.swarm.leave() self.node_manager.docker_client.swarm.leave()
data = {'host': self.host} data = {'host': self.host}
self.client_sock.send(f'[STOP_CHECK] {json.dumps(data)}'.encode()) self.client_sock.send(f'[STOP_CHECK] {json.dumps(data)}'.encode())
self.node_manager.exit()
return True return True
def request(self, host): # master side def request(self, host): # master side