osc/ucx: Detach dynamic window memory when freeing a window

Signed-off-by: Hessam Mirsadeghi <hmirsadeghi@nvidia.com>
open-mpi · May 11, 2021 · dd7a882 · dd7a882
1 parent 8e00f1d
commit dd7a882
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/ompi/mca/osc/ucx/osc_ucx_component.c b/ompi/mca/osc/ucx/osc_ucx_component.c
@@ -671,6 +671,7 @@ int ompi_osc_ucx_win_detach(struct ompi_win_t *win, const void *base) {
 int ompi_osc_ucx_free(struct ompi_win_t *win) {
     ompi_osc_ucx_module_t *module = (ompi_osc_ucx_module_t*) win->w_osc_module;
     int ret;
+    uint64_t i;
 
     assert(module->lock_count == 0);
     assert(opal_list_is_empty(&module->pending_posts) == true);
@@ -687,6 +688,13 @@ int ompi_osc_ucx_free(struct ompi_win_t *win) {
         return ret;
     }
 
+   /* MPI_Win_free should detach any memory attached to dynamic windows */
+    for (i = 0; i < module->state.dynamic_win_count; i++) {
+        assert(module->local_dynamic_win_info[i].refcnt == 1);
+        opal_common_ucx_wpmem_free(module->local_dynamic_win_info[i].mem);
+    }
+    module->state.dynamic_win_count = 0;
+
     free(module->addrs);
     free(module->state_addrs);