From 01d3b20a665a5a3f4862202d60112cb6ee6a926e Mon Sep 17 00:00:00 2001
From: Wenkai Du <wenkai.du@amd.com>
Date: Thu, 5 Aug 2021 17:45:48 -0700
Subject: [PATCH] Fix incorrect network proxy received bytes reporting

---
 src/transport/net_ib.cc | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/src/transport/net_ib.cc b/src/transport/net_ib.cc
index 5586649143..e04c5e10d8 100644
--- a/src/transport/net_ib.cc
+++ b/src/transport/net_ib.cc
@@ -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--;