[ROCm/rccl commit: 00fdb1ef51]
This commit is contained in:
BertanDogancay
2024-01-31 17:27:15 -08:00
rodzic 5ce968f570
commit ef57c82b60
3 zmienionych plików z 2 dodań i 13 usunięć
+1 -9
Wyświetl plik
@@ -1434,15 +1434,7 @@ static ncclResult_t computeColl(struct ncclInfo* info /* input */, int* workFunc
work->redOpArgIsPtr = info->opFull.scalarArgIsPtr;
work->opCount = info->comm->opCount;
if (info->comm->nRanks == 1) {
// one-rank reduce index
*workFuncIndex = ncclDevFuncId_P2p() + int(info->datatype);
return ncclSuccess;
} else if (info->coll == ncclFuncAllToAllPivot) {
*workFuncIndex = ncclDevFuncId_AllToAllPivot();
} else {
*workFuncIndex = ncclDevFuncId(info->coll, info->opFull.op, info->datatype, info->algorithm, info->protocol);
}
*workFuncIndex = ncclDevFuncId(info->coll, info->opFull.op, info->datatype, info->algorithm, info->protocol);
work->connIndex = 0;
proxyOp->connIndex = 0;
-1
Wyświetl plik
@@ -34,7 +34,6 @@
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <cstdarg>
#include "graph/topo.h"
#include "graph/xml.h"
#include "archinfo.h"