AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

introduce a hw topology aware bintree

Browse Source 
for hayabusa architecture.


[ROCm/rccl commit: e645b02cd8]
This commit is contained in:
Edgar Gabriel
2022-10-03 15:12:57 +00:00
parent dbbdf65020
commit f2736a4fb3
4 changed files with 204 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/include/graph.h
+1
View File
@@ -116,6 +116,7 @@ ncclResult_t ncclTopoPostset(struct ncclComm* comm, int* firstRanks, int* treePa
ncclResult_t ncclTreeBasePostset(struct ncclComm* comm, struct ncclTopoGraph* treeGraph);
ncclResult_t ncclBinaryTreePostset(struct ncclComm* comm, struct ncclTopoGraph* treeGraph);
ncclResult_t ncclBinaryTreeHayabusaPostset(struct ncclComm* comm, struct ncclTopoGraph* treeGraph);
ncclResult_t ncclTopoTuneModel(struct ncclComm* comm, int minCompCap, int maxCompCap, struct ncclTopoGraph* treeGraph, struct ncclTopoGraph* ringGraph, struct ncclTopoGraph* collNetGraph);
#include "info.h"