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

Tune number of channels for gfx90a (#349)

Browse Source 
[ROCm/rccl commit: 415c7cd3d1]
This commit is contained in:
Wenkai Du authored and GitHub committed 2021-04-19 15:27:01 -07:00
1 parent 951d89b12f
commit e28bac31aa
2 files changed
+3 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/src/graph/paths.cc
+1 -1
View File
@@ -573,7 +573,7 @@ ncclResult_t ncclTopoComputeP2pChannels(struct ncclComm* comm) {
}
}
if (comm->topo->nodes[GPU].count == comm->topo->nRanks && (comm->topo->type & RCCL_TOPO_4P2H_ROME) && !(comm->topo->type & RCCL_TOPO_GDR_ALL)) {
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->nodes[GPU].nodes[0].gpu.gcn != 910)) {
// Adjust P2P channels on Rome
comm->p2pnChannelsPerPeer = 2;
comm->p2pnChannels = 2;