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 
[ROCm/rccl commit: 735178c1fe]
This commit is contained in:
Hossein Pourreza
2024-01-10 08:01:37 -08:00
committed by GitHub
parent 64cf812da0
commit df0dbb887d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/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") && 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 == 8 && system->nodes[NET].count == 8 && system->nodes[GPU].count == system->nRanks) ||
(system->nodes[GPU].count != system->nRanks))) { (system->nodes[GPU].count != system->nRanks))) {
if (!rcclPathOverride(system, 0x100000)) if (!rcclPathOverride(system, 0x100000) && !rcclPathOverride(system, 0x1000))
rcclPathOverride(system, 0x1000); rcclPathOverride(system, 0xff00000);
} }
#if !defined(TOPO_EXPL) #if !defined(TOPO_EXPL)
} }