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

Fix P2P scheduling (#690)

Browse Source
This commit is contained in:
Wenkai Du
2023-02-21 07:49:54 -08:00
committed by GitHub
parent a640c6983f
commit 86e7b71234
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/enqueue.cc
-4
View File
@@ -645,8 +645,6 @@ static ncclResult_t scheduleP2pTasksToPlan(
ncclIntruQueueDequeue(&peers[recvPeer].recvQueue);
tasks->nTasksP2p -= 1;
}
} else {
NCCLCHECK(addP2pToPlan(comm, plan, nWorkBudget, /*isSendNotRecv=*/false, recvPeer, 0, nullptr, 0, recvIdx));
}
if (sendChunkBytes != 0) {
if (sendChunkBytes == -1) sendChunkBytes = 0;
@@ -660,8 +658,6 @@ static ncclResult_t scheduleP2pTasksToPlan(
ncclIntruQueueDequeue(&peers[sendPeer].sendQueue);
tasks->nTasksP2p -= 1;
}
} else {
NCCLCHECK(addP2pToPlan(comm, plan, nWorkBudget, /*isSendNotRecv=*/true, sendPeer, 0, nullptr, 0, sendIdx));
}
} while (sendBytes != 0 || recvBytes != 0);
}