AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Treat PIX and PXB as same GDR distance (#1894)

Browse Source 
[ROCm/rccl commit: de9ebd8a8b]
This commit is contained in:
Wenkai Du
2025-09-11 08:44:10 -07:00
committed by GitHub
parent 9fa7a738da
commit 94b53a03ac
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/graph/rome_models.cc
+4
View File
@@ -1791,6 +1791,10 @@ static bool permuteNetIds(int *n, int *g, int s, int last, struct rcclRomeModel*
for (j = 0; j < ref->nGpus; j++) {
// enabling PXN override paths over PHB and SYS
if (topo->gdrLevel[n[i]*ref->nGpus+g[j]] == PATH_PXN) continue;
// treat PIX and PXB as same
if ((ref->gdrLevel[i*ref->nGpus+j] == PATH_PXB && topo->gdrLevel[n[i]*ref->nGpus+g[j]] == PATH_PIX) ||
(ref->gdrLevel[i*ref->nGpus+j] == PATH_PIX && topo->gdrLevel[n[i]*ref->nGpus+g[j]] == PATH_PXB))
continue;
if (ref->gdrLevel[i*ref->nGpus+j] != topo->gdrLevel[n[i]*ref->nGpus+g[j]]) break;
}
if (j < ref->nGpus) break;