From dbb7500fd1f55021c3f60a079b4258cc65804771 Mon Sep 17 00:00:00 2001 From: Gilbert Lee Date: Fri, 6 Sep 2019 16:35:55 +0000 Subject: [PATCH] RDMA HDP flush fix [ROCm/rccl commit: 86ce0a93b5acdba12adf77c118b04813700cf3f6] --- projects/rccl/src/transport/net.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/rccl/src/transport/net.cc b/projects/rccl/src/transport/net.cc index 7991754f65..c0dbe5769c 100644 --- a/projects/rccl/src/transport/net.cc +++ b/projects/rccl/src/transport/net.cc @@ -333,7 +333,7 @@ ncclResult_t netRecvSetup(struct ncclPeerInfo* myInfo, struct ncclPeerInfo* peer int recvSize = offsetof(struct ncclRecvMem, buff)+buffSize; if (resources->useGdr) { NCCLCHECK(ncclCudaCalloc((char**)(&resources->devRecvMem), recvSize, true)); - CUDACHECK(hipDeviceGetAttribute((int*)&resources->curr_hdp_reg, hipDeviceAttributeHdpMemFlushCntl,peerInfo->cudaDev)); + CUDACHECK(hipDeviceGetAttribute((int*)&resources->curr_hdp_reg, hipDeviceAttributeHdpMemFlushCntl, cudaDev)); } NCCLCHECK(ncclCudaHostAlloc((void**)&resources->hostRecvMem, (void**)&resources->devHostRecvMem, recvSize)); resources->buffSize = buffSize;