UCP/MM: Fix review comment

openucx · Jan 24, 2022 · 6254ddf · 6254ddf
1 parent c9037eb
commit 6254ddf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ucp/core/ucp_mm.c b/src/ucp/core/ucp_mm.c
@@ -120,9 +120,6 @@ ucs_status_t ucp_mem_rereg_mds(ucp_context_h context, ucp_md_map_t reg_md_map,
                 continue;
             }
 
-            base_address = address;
-            reg_length   = length;
-
             if (context->config.ext.reg_whole_alloc_bitmap & UCS_BIT(mem_type)) {
                 ucp_memory_detect_internal(context, address, length, &mem_info);
                 base_address = mem_info.base_address;
@@ -137,6 +134,9 @@ ucs_status_t ucp_mem_rereg_mds(ucp_context_h context, ucp_md_map_t reg_md_map,
                 ucs_assertv(end_address >= UCS_PTR_BYTE_OFFSET(address, length),
                             "end_address=%p address+length=%p", end_address,
                             UCS_PTR_BYTE_OFFSET(address, length));
+            } else {
+                base_address = address;
+                reg_length   = length;
             }
 
             /* MD supports registration, register new memh on it */