Merge remote-tracking branch 'ncclRepo/master' into develop

This commit is contained in:
Aristotelis
2022-06-02 15:27:24 +00:00
12 zmienionych plików z 97 dodań i 44 usunięć
+2 -4
Wyświetl plik
@@ -962,12 +962,10 @@ void* ncclProxyService(void* _args) {
struct pollfd pollfds[NCCL_MAX_LOCAL_RANKS+1];
struct ncclProxyLocalPeer peers[NCCL_MAX_LOCAL_RANKS];
memset(&peers, 0, sizeof(struct ncclProxyLocalPeer)*NCCL_MAX_LOCAL_RANKS);
for (int s=0; s<NCCL_MAX_LOCAL_RANKS; s++) {
peers[s].sock.fd = pollfds[s].fd = -1;
peers[s].sock.abortFlag = NULL;
peers[s].sock.asyncFlag = 0;
pollfds[s].events = POLLHUP|POLLIN;
peers[s].asyncOps.type = 0;
}
pollfds[NCCL_MAX_LOCAL_RANKS].fd = comm->proxyState.listenSock->fd;
pollfds[NCCL_MAX_LOCAL_RANKS].events = POLLIN;
@@ -1068,13 +1066,13 @@ void* ncclProxyService(void* _args) {
ncclResult_t ncclProxyInit(struct ncclComm* comm, struct ncclSocket* sock, union ncclSocketAddress* peerAddresses) {
comm->proxyState.listenSock = sock;
comm->proxyState.peerAddresses = peerAddresses;
ncclSetThreadName(comm->proxyState.thread, "NCCL Service %2d", comm->cudaDev);
return ncclSuccess;
}
ncclResult_t ncclProxyCreate(struct ncclComm* comm) {
// comm->proxyState.thread is pthread_join()'d by commFree() in init.cc
pthread_create(&comm->proxyState.thread, NULL, ncclProxyService, comm);
ncclSetThreadName(comm->proxyState.thread, "NCCL Service %2d", comm->cudaDev);
return ncclSuccess;
}