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

increase p2p channels for Intel platform (#1448)

Browse Source 
Co-authored-by: hujingbo <hujingbo@kuaishou.com>

[ROCm/rccl commit: ad4c36dc34]
This commit is contained in:
Hujingbo
2024-12-10 23:33:37 +08:00
committed by GitHub
parent d7498b88a5
commit bea86f6248
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
@@ -899,8 +899,8 @@ ncclResult_t ncclTopoComputeP2pChannels(struct ncclComm* comm) {
NCCLCHECK(ncclTopoCpuType(comm->topo, &arch, &vendor, &model));
if (arch == NCCL_TOPO_CPU_ARCH_X86 && vendor == NCCL_TOPO_CPU_VENDOR_INTEL && !(comm->topo->type & RCCL_TOPO_XGMI_ALL)) {
// Adjust P2P channels on Intel platform
comm->p2pnChannelsPerPeer = 1;
comm->p2pnChannels = 2;
comm->p2pnChannelsPerPeer = 8;
comm->p2pnChannels = 8;
} 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)) {
// Adjust P2P channels on Rome
comm->p2pnChannelsPerPeer = 2;