AI/rocm-systems
2
0
Förgrening 0
Kod Ärenden Pull-förfrågningar Actions Packages Projekt Släpp Wiki Aktiviteter

[GRAPH] Reduce NCCL_TOPO_MAX_NODES to 64 (#1153)

Bläddra i källkod 
Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>
This commit is contained in:
Nilesh M Negi
2024-04-27 23:41:11 -05:00
committad av GitHub
förälder cc39e91c6f
incheckning b90436d292
1 ändrade filer med 1 tillägg och 1 borttagningar
Download Patch File Download Diff File Expand all files Collapse all files
src/include/graph.h
+1 -1
Visa fil
@@ -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);