From 0d02cd4d8bc1719cbb6b652b7dcd86e66269c249 Mon Sep 17 00:00:00 2001
From: Artem Polyakov <artpol84@gmail.com>
Date: Fri, 22 Dec 2017 13:57:23 +0700
Subject: [PATCH] fix prev #2

---
 src/uct/ib/base/ib_umr.c | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/uct/ib/base/ib_umr.c b/src/uct/ib/base/ib_umr.c
index 9b1a049f19da..1a71b4b34e60 100644
--- a/src/uct/ib/base/ib_umr.c
+++ b/src/uct/ib/base/ib_umr.c
@@ -472,6 +472,11 @@ ucs_status_t uct_ib_umr_reg_nc(uct_md_h uct_md, const uct_iov_t *iov,
     int i;
     size_t length;
 
+    length = 0;
+    for(i=0;i<iovcnt;i++) {
+        length += iov[i].length;
+    }
+
     uct_ib_md_t *md = ucs_derived_of(uct_md, uct_ib_md_t);
     if (ucs_unlikely(md->umr.qp == NULL)) {
         return UCS_ERR_UNSUPPORTED;
@@ -484,6 +489,8 @@ ucs_status_t uct_ib_umr_reg_nc(uct_md_h uct_md, const uct_iov_t *iov,
             return status;
         }
 
+        umr->mr->addr = iov[0].buffer;
+        umr->mr->length = length;
         memh->mr        = umr->mr;
         memh->umr       = umr;
         memh->lkey      = umr->mr->lkey;
@@ -494,10 +501,6 @@ ucs_status_t uct_ib_umr_reg_nc(uct_md_h uct_md, const uct_iov_t *iov,
     }
 
     umr = memh->umr;
-    length = 0;
-    for(i=0;i<iovcnt;i++) {
-        length += iov[i].length;
-    }
     umr->mr->addr = iov[0].buffer;
     umr->mr->length = length;
     *wr_p = &memh->umr->wr;