From df0dbb887dd0ff64417482ac8a2e983de5467f22 Mon Sep 17 00:00:00 2001 From: Hossein Pourreza Date: Wed, 10 Jan 2024 08:01:37 -0800 Subject: [PATCH] cover more gpu/nic mapping cases (#1037) [ROCm/rccl commit: 735178c1fe5bc2e4ab39a5e894f07066e1713d28] --- projects/rccl/src/graph/paths.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/rccl/src/graph/paths.cc b/projects/rccl/src/graph/paths.cc index 61c771bd0a..41fb1ead74 100644 --- a/projects/rccl/src/graph/paths.cc +++ b/projects/rccl/src/graph/paths.cc @@ -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) }