Fix Rome PCIe 2 node topology generation (#310)

[ROCm/rccl commit: 373a108516]
Этот коммит содержится в:
Wenkai Du authored and GitHub committed 2020-12-15 17:16:17 -08:00
1 родитель c570f09681
Коммит 4ea285c527
5 files changed
+56 -3

No files matched your search

+1 -1
Просмотреть файл
@@ -426,7 +426,7 @@ ncclResult_t ncclTopoSearchRecNet(struct ncclTopoSystem* system, struct ncclTopo
for (int i = 0; i<system->nodes[GPU].count; i++)
if (paths[i].count < paths[f].count) f = i;
int t = 1 << 10;
NCCLCHECK(ncclTopoSearchTryGpu(system, graph, saveGraph, 0, backToNet, backToFirstRank, FORCED_ORDER_PCI, &t, NET, n, f));
NCCLCHECK(ncclTopoSearchTryGpu(system, graph, saveGraph, 0, backToNet, backToFirstRank, (f == 0) ? FORCED_ORDER_PCI : 0, &t, NET, n, f));
// [RCCL] Event if forced order PCI is found, continue the search instead of ending early
// if (t == -1) *time = -1;
// [/RCCL]