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

Esse commit está contido em:
Wenkai Du
2021-05-17 10:15:16 -07:00
9 arquivos alterados com 35 adições e 20 exclusões
+5 -4
Ver Arquivo
@@ -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);