Fix P2P scheduling (#690)

[ROCm/rccl commit: 86e7b71234]
This commit is contained in:
Wenkai Du
2023-02-21 07:49:54 -08:00
zatwierdzone przez GitHub
rodzic 7d860d3642
commit 1f42d485f8
-4
Wyświetl plik
@@ -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);
}