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

[ROCm/rccl commit: d0d00c33ee]
Cette révision appartient à :
Karthikeyan Arumugam
2026-01-09 03:33:05 -08:00
révisé par GitHub
Parent 07453ebfaf
révision 94499918b3
+6
Voir le fichier
@@ -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