这个提交包含在:
BertanDogancay
2024-01-31 17:27:15 -08:00
父节点 da85abab54
当前提交 00fdb1ef51
修改 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;
-1
查看文件
@@ -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"