Fix min_nchannels bug for gfx94* nranks=4 (#1202)

Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>

[ROCm/rccl commit: d9661c17e6]
Šī revīzija ir iekļauta:
Nilesh M Negi
2024-06-07 14:31:28 -05:00
revīziju iesūtīja GitHub
vecāks efe851cfff
revīzija 9b88e59ea4
+1 -4
Parādīt failu
@@ -1396,10 +1396,7 @@ static ncclResult_t initTransportsRank(struct ncclComm* comm, struct ncclComm* p
allGather3Data[rank].nc = 16;
else if (nranks == 4)
// NCCL_MIN_NCHANNELS=24
allGather3Data[rank].nc = 6;
else if (nranks == 8)
// NCCL_MIN_NCHANNELS=56
allGather3Data[rank].nc = 2;
allGather3Data[rank].nc = 4;
}
allGather3Data[rank].pivotA2AEnabled = comm->topo->pivotA2AEnabled && rcclParamPivotAlltoallEnable();