From b99b89e7a2ddede520209e9b7aa8b7a57e3137cf Mon Sep 17 00:00:00 2001 From: Nilesh M Negi Date: Sat, 27 Apr 2024 23:41:11 -0500 Subject: [PATCH] [GRAPH] Reduce NCCL_TOPO_MAX_NODES to 64 (#1153) Signed-off-by: nileshnegi [ROCm/rccl commit: b90436d292d7838cf8689367f8d80589af361cae] --- projects/rccl/src/include/graph.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/rccl/src/include/graph.h b/projects/rccl/src/include/graph.h index ce92bd667b..8c892fcc15 100644 --- a/projects/rccl/src/include/graph.h +++ b/projects/rccl/src/include/graph.h @@ -66,7 +66,7 @@ ncclResult_t ncclTopoGetLocalNet(struct ncclTopoSystem* system, int rank, int ch ncclResult_t ncclTopoGetLocalGpu(struct ncclTopoSystem* system, int net, int* gpuIndex); ncclResult_t getLocalNetCountByBw(struct ncclTopoSystem* system, int gpu, int *count); -#define NCCL_TOPO_MAX_NODES 256 +#define NCCL_TOPO_MAX_NODES 64 // Init search. Needs to be done before calling ncclTopoCompute ncclResult_t ncclTopoSearchInit(struct ncclTopoSystem* system);