[ROCm/rccl commit: 00fdb1ef51]
Этот коммит содержится в:
BertanDogancay
2024-01-31 17:27:15 -08:00
родитель 5ce968f570
Коммит ef57c82b60
3 изменённых файлов: 2 добавлений и 13 удалений
+1 -9
Просмотреть файл
@@ -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;