Shutdown socket before close in ncclSocketClose()

[ROCm/rccl commit: 367e9b61c3]
Este commit está contenido en:
Kaiming Ouyang
2023-04-12 08:51:09 -07:00
padre 8dcf8e8720
commit e35b05a872
+4 -1
Ver fichero
@@ -818,7 +818,10 @@ ncclResult_t ncclSocketTryRecv(struct ncclSocket* sock, void* ptr, int size, int
ncclResult_t ncclSocketClose(struct ncclSocket* sock) {
if (sock != NULL) {
if (sock->fd >= 0) close(sock->fd);
if (sock->fd >= 0) {
shutdown(sock->fd, SHUT_RDWR);
close(sock->fd);
}
sock->state = ncclSocketStateClosed;
sock->fd = -1;
}