introduce a hw topology aware bintree

for hayabusa architecture.
2022-10-03 15:12:57 +00:00
parent ef71550738
commit e645b02cd8
4 changed files with 204 additions and 2 deletions
@@ -1042,7 +1042,11 @@ static ncclResult_t initTransportsRank(struct ncclComm* comm, ncclUniqueId* comm

  if (comm->topo->pivotA2ANumBiRings == 3) {
    NCCLCHECK(ncclTreeBasePostset(comm, &treeGraph));
-    NCCLCHECK(ncclBinaryTreePostset(comm, &treeGraph));
+    if (comm->virtualId == -1) {
+      NCCLCHECK(ncclBinaryTreeHayabusaPostset(comm, &treeGraph));
+    } else {
+      NCCLCHECK(ncclBinaryTreePostset(comm, &treeGraph));
+    }
  }

  free(allTopoRanks);