increase p2p channels for Intel platform (#1448)

Co-authored-by: hujingbo <hujingbo@kuaishou.com>
This commit is contained in:
Hujingbo
2024-12-10 23:33:37 +08:00
zatwierdzone przez GitHub
rodzic 9aa5b9f02e
commit ad4c36dc34
+2 -2
Wyświetl plik
@@ -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;