add 256 channels support

[ROCm/rccl commit: 506f16c506]
This commit is contained in:
Nusrat Islam
2024-05-15 16:58:28 -05:00
والد 1934d0f377
کامیت b0b2aa1166
11فایلهای تغییر یافته به همراه95 افزوده شده و 53 حذف شده
+4 -2
مشاهده پرونده
@@ -225,8 +225,10 @@ static void groupCleanup(struct ncclComm** groupCommHeadPtr, struct ncclComm** g
for (int i = 0; i < comm->nRanks; i++) {
comm->tasks.peers[i].sendSeen = false;
comm->tasks.peers[i].recvSeen = false;
comm->connectSend[i] = 0UL;
comm->connectRecv[i] = 0UL;
for (int j = 0; j < 4; j++) {
comm->connectSend[i].masks[j] = 0UL;
comm->connectRecv[i].masks[j] = 0UL;
}
}
comm->unlaunchedPlansHead = nullptr;
// Reclaim abandoned kernel plan memory. Note ncclWork structs were already