From d0d00c33ee70e808eb693816095f5234b7797faf Mon Sep 17 00:00:00 2001
From: Karthikeyan Arumugam <karthik@pensando.io>
Date: Fri, 9 Jan 2026 03:33:05 -0800
Subject: [PATCH] Add check for P2pPolicy for rocm-ib (#2122)

---
 src/transport/net.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/transport/net.cc b/src/transport/net.cc
index 9087604d88..caeadda061 100644
--- a/src/transport/net.cc
+++ b/src/transport/net.cc
@@ -779,6 +779,12 @@ static ncclResult_t sendProxyConnect(struct ncclProxyConnection* connection, str
   if (proxyState->ncclNet == &ncclNetIb) {
     NCCLCHECK(rcclNetP2pPolicy(req->handle, resources->isP2p));
   }
+
+#if defined(__HIP_PLATFORM_AMD__) || defined(__HIPCC__)
+  if (proxyState->ncclNet == &rocmNetIb) {
+    NCCLCHECK(rcclRocmNetP2pPolicy(req->handle, resources->isP2p));
+  }
+#endif
   
   if (resources->shared) {
     // Shared buffers