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

cover more gpu/nic mapping cases (#1037)

Browse Source
This commit is contained in:
Hossein Pourreza
2024-01-10 08:01:37 -08:00
committed by GitHub
parent 5851ae5974
commit 735178c1fe
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/graph/paths.cc
+2 -2
View File
@@ -638,8 +638,8 @@ ncclResult_t ncclTopoComputePaths(struct ncclTopoSystem* system, struct ncclComm
IsArchMatch(system->nodes[GPU].nodes[0].gpu.gcn, "gfx94") &&
((system->nodes[GPU].count == 8 && system->nodes[NET].count == 8 && system->nodes[GPU].count == system->nRanks) ||
(system->nodes[GPU].count != system->nRanks))) {
if (!rcclPathOverride(system, 0x100000))
rcclPathOverride(system, 0x1000);
if (!rcclPathOverride(system, 0x100000) && !rcclPathOverride(system, 0x1000))
rcclPathOverride(system, 0xff00000);
}
#if !defined(TOPO_EXPL)
}