Rail optimization for rings (#1140)

- Modifies the ring creation algorithm to be friendlier to rail-optimized topologies (should not affect classic fabric topologies)
This commit is contained in:
gilbertlee-amd
2024-04-15 12:03:57 -06:00
gecommit door GitHub
bovenliggende 3caad91f32
commit 4cb62f999a
11 gewijzigde bestanden met toevoegingen van 296 en 69 verwijderingen
+5 -2
Bestand weergeven
@@ -307,8 +307,11 @@ ncclResult_t ncclTopoXmlLoadSystem(FILE* file, struct ncclXml* xml, struct ncclX
}
const char* name;
NCCLCHECK(xmlGetAttr(head, "name", &name));
if (name != NULL) INFO(NCCL_GRAPH, "Loading topology %s", name);
else INFO(NCCL_GRAPH, "Loading unnamed topology");
if (name != NULL) {
INFO(NCCL_GRAPH, "Loading topology %s", name);
} else {
INFO(NCCL_GRAPH, "Loading unnamed topology");
}
struct xmlHandler handlers[] = { { "cpu", ncclTopoXmlLoadCpu } };
NCCLCHECK(xmlLoadSub(file, xml, head, handlers, 1));