AI/rocm-systems
2
0
Bifurcation 0
Code Tickets Demandes d'ajout Actions Paquets Projets Publications Wiki Activité

Merge pull request #194 from wenkaidu/search

Parcourir la source 
Fix incorrect next device ID in PCI ordered search

[ROCm/rccl commit: cf5070f6c0]
Cette révision appartient à :
Wenkai Du
2020-04-27 09:54:09 -07:00
révisé par GitHub
Parent b9acac2db6 7b7f781658
révision 5f57e6b466
1 fichiers modifiés avec 1 ajouts et 1 suppressions
Télécharger le Fichier Patch Télécharger le Fichier des Différences Développer tous les fichiers Réduire tous les fichiers
projects/rccl/src/graph/search.cc
+1 -1
Voir le fichier
@@ -345,7 +345,7 @@ ncclResult_t ncclTopoSearchRecGpu(struct ncclTopoSystem* system, struct ncclTopo
int next[NCCL_TOPO_MAX_NODES]; int next[NCCL_TOPO_MAX_NODES];
int count; int count;
if (forcedOrder == FORCED_ORDER_PCI) { // Try the PCI order 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; count = 1;
} else if (forcedOrder == FORCED_ORDER_REPLAY) { // Try last channel order } else if (forcedOrder == FORCED_ORDER_REPLAY) { // Try last channel order
NCCLCHECK(ncclTopoReplayGetGpu(system, graph, step, next)); NCCLCHECK(ncclTopoReplayGetGpu(system, graph, step, next));