Fix incorrect network proxy received bytes reporting

[ROCm/rccl commit: 01d3b20a66]
Этот коммит содержится в:
Wenkai Du committed 2021-08-05 17:45:48 -07:00
1 родитель 6e997354e8
Коммит 79bc1e3fde
1 file changed
+1 -4
+1 -4
Просмотреть файл
@@ -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--;