yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Shutdown socket before close in ncclSocketClose()

Browse Source
This commit is contained in:
Kaiming Ouyang 2023-04-12 08:51:09 -07:00
parent 5d3ab08b69
commit 367e9b61c3
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/misc/socket.cc
View File

@ -818,7 +818,10 @@ ncclResult_t ncclSocketTryRecv(struct ncclSocket* sock, void* ptr, int size, int
ncclResult_t ncclSocketClose(struct ncclSocket* sock) { ncclResult_t ncclSocketClose(struct ncclSocket* sock) {
if (sock != NULL) { 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->state = ncclSocketStateClosed;
sock->fd = -1; sock->fd = -1;
} }