From b14e32c46e7afd66dfd717fa8f020df254e40634 Mon Sep 17 00:00:00 2001
From: AbandiGa <abandiga@gmail.com>
Date: Fri, 21 Nov 2025 18:28:10 -0600
Subject: [PATCH] Fix rcclNetP2pPolicy issue (#2072)

* fix rcclNetP2pPolicy issue

* change the comment to ncclNetIb
---
 src/transport/net.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/transport/net.cc b/src/transport/net.cc
index 1ad8de99fa..2c349fae10 100644
--- a/src/transport/net.cc
+++ b/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));
   }