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