Merge pull request #141 from wenkaidu/hdp

Disable HDP flush for RDMA

[ROCm/rccl commit: b98d334114]
Этот коммит содержится в:
Wenkai Du
2019-10-24 16:26:01 -07:00
коммит произвёл GitHub
родитель fc33ee4f44 d3f399f619
Коммит 2edd1a1c5d
+2 -2
Просмотреть файл
@@ -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);
}