Fix incorrect network proxy received bytes reporting

[ROCm/rccl commit: 01d3b20a66]
This commit is contained in:
Wenkai Du
2021-08-05 17:45:48 -07:00
förälder 6e997354e8
incheckning 79bc1e3fde
+1 -4
Visa fil
@@ -901,10 +901,7 @@ ncclResult_t ncclIbTest(void* request, int* done, int* size) {
doneReq->size = wc->byte_len;
#if USE_RDMA_WRITE
} else if (wc->opcode == IBV_WC_RECV_RDMA_WITH_IMM) {
if (doneReq->size == -1)
doneReq->size = wc->imm_data;
else
doneReq->size += wc->imm_data;
doneReq->size = wc->imm_data;
#endif
}
doneReq->events--;