ports/openmpi
ports/openmpi
1
1
Форкнуть 0
Код Релизы Активность

OSC/UCX: code deduplication

Просмотр исходного кода 
Signed-off-by: Sergey Oblomov <sergeyo@mellanox.com>
Этот коммит содержится в:
Sergey Oblomov 2018-07-19 12:39:15 +03:00
родитель 6f0a7a2005
Коммит fa33e322e7
1 изменённых файлов: 17 добавлений и 16 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

33
ompi/mca/osc/ucx/osc_ucx_component.c
Просмотреть файл

@ -26,6 +26,7 @@ static int component_query(struct ompi_win_t *win, void **base, size_t size, int
static int component_select(struct ompi_win_t *win, void **base, size_t size, int disp_unit, static int component_select(struct ompi_win_t *win, void **base, size_t size, int disp_unit,
struct ompi_communicator_t *comm, struct opal_info_t *info, struct ompi_communicator_t *comm, struct opal_info_t *info,
int flavor, int *model); int flavor, int *model);
static void ompi_osc_ucx_unregister_progress(void);
ompi_osc_ucx_component_t mca_osc_ucx_component = { ompi_osc_ucx_component_t mca_osc_ucx_component = {
{ /* ompi_osc_base_component_t */ { /* ompi_osc_base_component_t */
@ -236,6 +237,20 @@ static inline int mem_map(void **base, size_t size, ucp_mem_h *memh_ptr,
return ret; return ret;
} }
static void ompi_osc_ucx_unregister_progress()
{
int ret;
mca_osc_ucx_component.num_modules--;
OSC_UCX_ASSERT(mca_osc_ucx_component.num_modules >= 0);
if (0 == mca_osc_ucx_component.num_modules) {
ret = opal_progress_unregister(progress_callback);
if (OMPI_SUCCESS != ret) {
OSC_UCX_VERBOSE(1, "opal_progress_unregister failed: %d", ret);
}
}
}
static int component_select(struct ompi_win_t *win, void **base, size_t size, int disp_unit, static int component_select(struct ompi_win_t *win, void **base, size_t size, int disp_unit,
struct ompi_communicator_t *comm, struct opal_info_t *info, struct ompi_communicator_t *comm, struct opal_info_t *info,
int flavor, int *model) { int flavor, int *model) {
@ -643,14 +658,7 @@ static int component_select(struct ompi_win_t *win, void **base, size_t size, in
} }
if (module) { if (module) {
free(module); free(module);
mca_osc_ucx_component.num_modules--; ompi_osc_ucx_unregister_progress();
OSC_UCX_ASSERT(mca_osc_ucx_component.num_modules >= 0);
if (0 == mca_osc_ucx_component.num_modules) {
ret = opal_progress_unregister(progress_callback);
if (OMPI_SUCCESS != ret) {
OSC_UCX_VERBOSE(1, "opal_progress_unregister failed: %d", ret);
}
}
} }
error_nomem: error_nomem:
@ -819,14 +827,7 @@ int ompi_osc_ucx_free(struct ompi_win_t *win) {
ompi_comm_free(&module->comm); ompi_comm_free(&module->comm);
free(module); free(module);
mca_osc_ucx_component.num_modules--; ompi_osc_ucx_unregister_progress();
OSC_UCX_ASSERT(mca_osc_ucx_component.num_modules >= 0);
if (0 == mca_osc_ucx_component.num_modules) {
ret = opal_progress_unregister(progress_callback);
if (OMPI_SUCCESS != ret) {
OSC_UCX_VERBOSE(1, "opal_progress_unregister failed: %d", ret);
}
}
return ret; return ret;
} }