fix prev #2

artpol84 · Dec 22, 2017 · 0d02cd4 · 0d02cd4
1 parent cde837b
commit 0d02cd4
Showing 1 changed file with 7 additions and 4 deletions.
diff --git 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;