Shutdown socket before close in ncclSocketClose()

[ROCm/rccl commit: 367e9b61c3]
Ta zmiana jest zawarta w:
%! (template.HTML=Kaiming Ouyang)złożyło 2023-04-13 09:11:52 -07:00
1 rodzic 8dcf8e8720
złożenie e35b05a872
Plik 1 zmieniony
+4 -1
+4 -1
Wyświetl plik
@@ -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;
}