AI/rocm-systems
2
0
Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

Remove 2H4P condition from P2P channels adjustment (#890)

Bladeren bron
This commit is contained in:
Wenkai Du
2023-09-13 12:54:21 -07:00
gecommit door GitHub
bovenliggende c1bfd5f0d8
commit 16dd05a58a
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files
src/graph/paths.cc
+1 -1
Bestand weergeven
@@ -812,7 +812,7 @@ ncclResult_t ncclTopoComputeP2pChannels(struct ncclComm* comm) {
// Adjust P2P channels on Intel platform
comm->p2pnChannelsPerPeer = 1;
comm->p2pnChannels = 2;
} else if (comm->topo->nodes[GPU].count == comm->topo->nRanks && (comm->topo->type & RCCL_TOPO_4P2H_ROME) && !(comm->topo->type & RCCL_TOPO_GDR_ALL) && !(comm->topo->type & RCCL_TOPO_XGMI_ALL)) {
} else if (comm->topo->nodes[GPU].count == comm->topo->nRanks && !(comm->topo->type & RCCL_TOPO_GDR_ALL) && !(comm->topo->type & RCCL_TOPO_XGMI_ALL)) {
// Adjust P2P channels on Rome
comm->p2pnChannelsPerPeer = 2;
comm->p2pnChannels = 2;