From 928414ac06d8d9080e05fa2ba972c44d8911a557 Mon Sep 17 00:00:00 2001 From: Arm Patinyasakdikul Date: Wed, 23 Oct 2024 11:15:11 -0500 Subject: [PATCH] Increased maximum number of XML nodes to support CPX mode. (#1386) [ROCm/rccl commit: 29f87c7191326d88f48c0e7500def65173e2bdab] --- projects/rccl/src/graph/topo.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/rccl/src/graph/topo.h b/projects/rccl/src/graph/topo.h index b3597bc39c..a409fc24ba 100644 --- a/projects/rccl/src/graph/topo.h +++ b/projects/rccl/src/graph/topo.h @@ -200,8 +200,8 @@ ncclResult_t ncclTopoPrintPaths(struct ncclTopoSystem* system); ncclResult_t ncclTopoLoadSystem(const char* xmlTopoFile, struct ncclTopoSystem* system); ncclResult_t ncclTopoGetIntermediateRank(struct ncclTopoSystem* system, int rank, int64_t netId, int* intermediateRank); -#define NCCL_TOPO_XML_MAX_NODES 256 -#define NCCL_GRAPH_XML_MAX_NODES 4096 +#define NCCL_TOPO_XML_MAX_NODES 8192 +#define NCCL_GRAPH_XML_MAX_NODES 8192 ncclResult_t ncclTopoGetSystemFromXml(struct ncclXml* xml, struct ncclTopoSystem** topoSystem, uint64_t localHostHash); ncclResult_t ncclTopoGetGraphFromXml(struct ncclXmlNode *xmlGraphs, struct ncclTopoSystem* system, struct ncclTopoGraph* graph, int* nChannels); ncclResult_t ncclTopoGetXmlFromGraphs(int ngraphs, struct ncclTopoGraph** graphs, struct ncclTopoSystem* system, struct ncclXml *xml);