Add check for P2pPolicy for rocm-ib (#2122)

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