diff --git a/projects/rccl/src/transport/net.cc b/projects/rccl/src/transport/net.cc index 1ad8de99fa..2c349fae10 100644 --- a/projects/rccl/src/transport/net.cc +++ b/projects/rccl/src/transport/net.cc @@ -776,7 +776,8 @@ static ncclResult_t sendProxyConnect(struct ncclProxyConnection* connection, str NCCLCHECK(ncclNetGetDeviceHandle(resources->netDeviceType, resources->netDeviceVersion, false /*isRecv*/, &resources->netDeviceHandle)); bool rccl_anp = !(strcmp(proxyState->ncclNet->name, RCCL_ANP_PLUGIN_STR)); - if (rcclNetP2pPolicy) { + // Only call rcclNetP2pPolicy for ncclNetIb + if (proxyState->ncclNet == &ncclNetIb) { NCCLCHECK(rcclNetP2pPolicy(req->handle, resources->isP2p)); }