AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clean up

Browse Source
This commit is contained in:
BertanDogancay
2024-01-31 17:27:15 -08:00
parent da85abab54
commit 00fdb1ef51
3 changed files with 2 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/enqueue.cc
+1 -9
View File
@@ -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;