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

[Transport] Fix IntraNet (#1582)

Browse Source 
[ROCm/rccl commit: d88cca3098]
This commit is contained in:
Bertan Dogancay authored and GitHub committed 2025-03-04 13:30:36 -05:00
1 parent 086fa823db
commit d1247bbf2a
2 files changed
+17 -7
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/src/enqueue.cc
+6 -7
View File
@@ -805,6 +805,12 @@ static ncclResult_t scheduleCollTasksToPlan(
}
proxyOp->channelId = c;
proxyOp->opCount = proxyOpId;
proxyOp->connIndex = 0;
if (task->protocol == NCCL_PROTO_SIMPLE && task->algorithm == NCCL_ALGO_RING) {
if (comm->useIntraNet && nBytes > rcclParamIntraNetThreshold()) {
proxyOp->connIndex = NCCL_CONN_IDX_P2P_NET;
}
}
addWorkBatchToPlan(comm, plan, c, workNode->workType, task->devFuncId, plan->workBytes);
NCCLCHECK(addProxyOpIfNeeded(comm, plan, proxyOp));
}
@@ -1992,13 +1998,6 @@ static ncclResult_t calcCollChunking(
}
}
proxyOp->connIndex = 0;
if (info->protocol == NCCL_PROTO_SIMPLE && info->algorithm == NCCL_ALGO_RING) {
if (comm->useIntraNet && nBytes > rcclParamIntraNetThreshold()) {
proxyOp->connIndex = NCCL_CONN_IDX_P2P_NET;
}
}
*outChunkSize = chunkSize;
return ncclSuccess;
}