This commit is contained in:
Saeed Maleki 2021-04-03 03:27:55 +00:00
Родитель d01feb2275
Коммит 90f3c3cf0d
1 изменённых файлов: 1 добавлений и 2 удалений

Просмотреть файл

@ -707,15 +707,14 @@ ncclResult_t scklGetAlgoFromXMLAndSetComm(struct ncclComm* comm) {
}
// setting the summary of the sckl aglorithm
scklChannelInfo* scklChannel = &scklAlgo->scklChannels[sTB->channelId];
sTB->rid = scklChannel->nsendPeers + scklChannel->nrecvPeers;
if (sTB->type == SCKL_SEND){
scklChannel->sendPeers[scklChannel->nsendPeers] = peer;
scklChannel->nchunksForSendPeer[scklChannel->nsendPeers] = ntransfers;
sTB->rid = scklChannel->nsendPeers;
scklChannel->nsendPeers++;
} else if (sTB->type == SCKL_RECV){
scklChannel->recvPeers[scklChannel->nrecvPeers] = peer;
scklChannel->nchunksForRecvPeer[scklChannel->nrecvPeers] = ntransfers;
sTB->rid = scklChannel->nsendPeers;
scklChannel->nrecvPeers++;
}
scklChannel->nBlocksForChannel = std::max(scklChannel->nBlocksForChannel, sTB->rid+1);