File tree Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Original file line number Diff line number Diff line change @@ -1061,8 +1061,10 @@ int mca_btl_openib_add_procs(
1061
1061
1062
1062
}
1063
1063
1064
+ opal_mutex_lock (& openib_btl -> ib_lock );
1064
1065
openib_btl -> local_procs += local_procs ;
1065
1066
openib_btl -> device -> mem_reg_max = openib_btl -> device -> mem_reg_max_total / openib_btl -> local_procs ;
1067
+ opal_mutex_unlock (& openib_btl -> ib_lock );
1066
1068
1067
1069
return OPAL_SUCCESS ;
1068
1070
}
@@ -1124,8 +1126,17 @@ struct mca_btl_base_endpoint_t *mca_btl_openib_get_ep (struct mca_btl_base_modul
1124
1126
1125
1127
(void )init_ib_proc_nolock (openib_btl , ib_proc , & endpoint ,
1126
1128
local_port_cnt , btl_rank );
1129
+
1127
1130
exit :
1128
1131
opal_mutex_unlock (& ib_proc -> proc_lock );
1132
+
1133
+ if (is_new && OPAL_PROC_ON_LOCAL_NODE (proc -> proc_flags )) {
1134
+ opal_mutex_lock (& openib_btl -> ib_lock );
1135
+ openib_btl -> local_procs += 1 ;
1136
+ openib_btl -> device -> mem_reg_max = openib_btl -> device -> mem_reg_max_total / openib_btl -> local_procs ;
1137
+ opal_mutex_unlock (& openib_btl -> ib_lock );
1138
+ }
1139
+
1129
1140
return endpoint ;
1130
1141
}
1131
1142
You can’t perform that action at this time.
0 commit comments