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);