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