Fix incorrect next device ID in PCI ordered search

Αυτή η υποβολή περιλαμβάνεται σε:
Wenkai Du
2020-04-25 01:01:13 +00:00
γονέας c017f6e900
υποβολή edb49ed2d5
+1 -1
Προβολή Αρχείου
@@ -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));