AI/rocm-systems
2
0
Forka 0
Codice Problemi Pull Request Actions Pacchetti Progetti Rilasci Wiki Attività

Increase max channles to 64 (#993)

Sfoglia il codice sorgente
This commit is contained in:
Wenkai Du
2023-12-01 16:01:11 -08:00
committato da GitHub
parent 0b53f79196
commit 4ba65d1d6a
4 ha cambiato i file con 4 aggiunte e 11 eliminazioni
Scarica il file Patch Scarica il file Diff Expand all files Collapse all files
src/graph/paths.cc
-7
Vedi File
@@ -771,13 +771,6 @@ ncclResult_t ncclTopoTrimSystem(struct ncclTopoSystem* system, struct ncclComm*
INFO(NCCL_GRAPH, "GDR is available on all GPUs");
}
// Special handling of gfx94x
if (rcclParamEnableIntranet() == 1 || (rcclParamEnableIntranet() == -2 &&
IsArchMatch(system->nodes[GPU].nodes[0].gpu.gcn, "gfx94") &&
system->nodes[GPU].count == 8 && system->nodes[NET].count == 8)) {
remove = 0;
system->type |= RCCL_TOPO_FORCE_INTRA;
}
comm->localRanks = system->nodes[GPU].count;
if (system->nodes[GPU].count == comm->nRanks && remove) {
for (int n=system->nodes[NET].count-1; n>=0; n--)