Merge pull request #141 from wenkaidu/hdp

Disable HDP flush for RDMA

[ROCm/rccl commit: b98d334114]
This commit is contained in:
Wenkai Du
2019-10-24 16:26:01 -07:00
committad av GitHub
förälder fc33ee4f44 d3f399f619
incheckning 2edd1a1c5d
+2 -2
Visa fil
@@ -547,8 +547,8 @@ ncclResult_t netRecvProxy(struct ncclProxyArgs* args) {
if (resources->useGdr) {
ncclNetFlush(resources->netRecvComm, localBuff+buffSlot*stepSize, size, mhandle);
// Flush local HDP register after local read-back finishes
STORE(resources->curr_hdp_reg, 0x1);
TRACE(NCCL_NET, "Flushing GPU memory via HDP %p", resources->curr_hdp_reg);
//STORE(resources->curr_hdp_reg, 0x1);
//TRACE(NCCL_NET, "Flushing GPU memory via HDP %p", resources->curr_hdp_reg);
}
STORE(&resources->hostRecvMem->tail, args->head);
}