[GRAPH] Reduce NCCL_TOPO_MAX_NODES to 64 (#1153)

Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>
Этот коммит содержится в:
Nilesh M Negi
2024-04-27 23:41:11 -05:00
коммит произвёл GitHub
родитель cc39e91c6f
Коммит b90436d292
+1 -1
Просмотреть файл
@@ -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);