Shutdown socket before close in ncclSocketClose()

[ROCm/rccl commit: 367e9b61c3]
This commit is contained in:
Kaiming Ouyang committed 2023-04-13 09:11:52 -07:00
1 szülő 8dcf8e8720
commit e35b05a872
1 file changed
+4 -1
+4 -1
Fájl megtekintése
@@ -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;
}