Fix incorrect next device ID in PCI ordered search

[ROCm/rccl commit: edb49ed2d5]
2020-04-25 01:01:13 +00:00
parent 033f8b7398
commit 7b7f781658
1 changed files with 1 additions and 1 deletions
@@ -345,7 +345,7 @@ ncclResult_t ncclTopoSearchRecGpu(struct ncclTopoSystem* system, struct ncclTopo
    int next[NCCL_TOPO_MAX_NODES];
    int count;
    if (forcedOrder == FORCED_ORDER_PCI) { // Try the PCI order
-      next[0] = (busIdToCudaDev(gpu->id)+1)%system->nodes[GPU].count;
+      next[0] = step+1;
      count = 1;
    } else if (forcedOrder == FORCED_ORDER_REPLAY) { // Try last channel order
      NCCLCHECK(ncclTopoReplayGetGpu(system, graph, step, next));