Fix rcclNetP2pPolicy issue (#2072)

* fix rcclNetP2pPolicy issue * change the comment to ncclNetIb [ROCm/rccl commit: b14e32c46e]
2025-11-21 18:28:10 -06:00
parent 3026698c40
commit d6087d0d62
1 changed files with 2 additions and 1 deletions
@@ -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));
  }