AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'nccl/master' into 2.9.8

Browse Source
This commit is contained in:
Wenkai Du
2021-05-17 10:15:16 -07:00
parent 0b2bfdd6d8 ca8485b0d0
commit 87727383fe
9 changed files with 35 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/bootstrap.cc
+5 -4
View File
@@ -551,10 +551,11 @@ ncclResult_t bootstrapClose(void* commState) {
ncclResult_t bootstrapAbort(void* commState) {
struct extState* state = (struct extState*)commState;
close(state->extListenFd);
close(state->extRingSendFd);
close(state->extRingRecvFd);
state->allocState->stop = 2;
if (commState == NULL) return ncclSuccess;
if (state->extListenFd) close(state->extListenFd);
if (state->extRingSendFd) close(state->extRingSendFd);
if (state->extRingRecvFd) close(state->extRingRecvFd);
if (state->allocState) state->allocState->stop = 2;
free(state->peerCommAddresses);
free(state->peerAllocAddresses);
free(state);