Fix rcclNetP2pPolicy issue (#2072)
* fix rcclNetP2pPolicy issue * change the comment to ncclNetIb
This commit is contained in:
@@ -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));
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user