1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
/*
* Copyright (c) 2016-2017 Inria. All rights reserved.
* Copyright (c) 2019 Research Organization for Information Science
* and Technology (RIST). All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "ompi_config.h"
#include <string.h>
#include "osc_monitoring.h"
#include "ompi/constants.h"
#include "ompi/communicator/communicator.h"
#include "ompi/info/info.h"
#include "ompi/win/win.h"
#include "ompi/info/info.h"
#include "ompi/mca/osc/osc.h"
#include "ompi/mca/osc/base/base.h"
#include "opal/mca/base/mca_base_component_repository.h"
/**************************************************/
/* Include templated macros and inlined functions */
#include "osc_monitoring_template_gen.h"
/**************************************************/
static int mca_osc_monitoring_component_init(bool enable_progress_threads,
bool enable_mpi_threads)
{
OPAL_MONITORING_PRINT_INFO("osc_component_init");
return mca_common_monitoring_init();
}
static int mca_osc_monitoring_component_finish(void)
{
OPAL_MONITORING_PRINT_INFO("osc_component_finish");
mca_common_monitoring_finalize();
return OMPI_SUCCESS;
}
static int mca_osc_monitoring_component_register(void)
{
return OMPI_SUCCESS;
}
static int mca_osc_monitoring_component_query(struct ompi_win_t *win, void **base, size_t size, int disp_unit,
struct ompi_communicator_t *comm, struct opal_info_t *info,
int flavor)
{
OPAL_MONITORING_PRINT_INFO("osc_component_query");
return mca_osc_monitoring_component.priority;
}
static inline int
ompi_mca_osc_monitoring_set_template(ompi_osc_base_component_t *best_component,
ompi_osc_base_module_t *module)
{
osc_monitoring_components_list_t comp = osc_monitoring_components_list[0];
for (unsigned i = 0; NULL != comp.name; comp = osc_monitoring_components_list[++i]) {
if ( 0 == strcmp(comp.name, best_component->osc_version.mca_component_name) ) {
comp.fct(module);
return OMPI_SUCCESS;
}
}
return OMPI_ERR_NOT_SUPPORTED;
}
static int mca_osc_monitoring_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,
int flavor, int *model)
{
OPAL_MONITORING_PRINT_INFO("osc_component_select");
opal_list_item_t *item;
ompi_osc_base_component_t *best_component = NULL;
int best_priority = -1, priority, ret = OMPI_SUCCESS;
/* Redo the select loop to add our layer in the middle */
for (item = opal_list_get_first(&ompi_osc_base_framework.framework_components) ;
item != opal_list_get_end(&ompi_osc_base_framework.framework_components) ;
item = opal_list_get_next(item)) {
ompi_osc_base_component_t *component = (ompi_osc_base_component_t*)
((mca_base_component_list_item_t*) item)->cli_component;
if( component == (ompi_osc_base_component_t*)(&mca_osc_monitoring_component) )
continue; /* skip self */
priority = component->osc_query(win, base, size, disp_unit, comm, info, flavor);
if (priority < 0) {
continue;
}
if (priority > best_priority) {
best_component = component;
best_priority = priority;
}
}
if (NULL == best_component) return OMPI_ERR_NOT_SUPPORTED;
OPAL_MONITORING_PRINT_INFO("osc: chosen one: %s", best_component->osc_version.mca_component_name);
ret = best_component->osc_select(win, base, size, disp_unit, comm, info, flavor, model);
if( OMPI_SUCCESS == ret ) {
/* Intercept module functions with ours, based on selected component */
ret = ompi_mca_osc_monitoring_set_template(best_component, win->w_osc_module);
if (OMPI_ERR_NOT_SUPPORTED == ret) {
OPAL_MONITORING_PRINT_WARN("osc: monitoring disabled: no module for this component "
"(%s)", best_component->osc_version.mca_component_name);
return OMPI_SUCCESS;
}
}
return ret;
}
ompi_osc_monitoring_component_t mca_osc_monitoring_component = {
.super = {
/* First, the mca_base_component_t struct containing meta
information about the component itself */
.osc_version = {
OMPI_OSC_BASE_VERSION_3_0_0,
.mca_component_name = "monitoring", /* MCA component name */
MCA_MONITORING_MAKE_VERSION,
.mca_register_component_params = mca_osc_monitoring_component_register
},
.osc_data = {
/* The component is checkpoint ready */
MCA_BASE_METADATA_PARAM_CHECKPOINT
},
.osc_init = mca_osc_monitoring_component_init, /* component init */
.osc_finalize = mca_osc_monitoring_component_finish, /* component finalize */
.osc_query = mca_osc_monitoring_component_query,
.osc_select = mca_osc_monitoring_component_select
},
.priority = INT_MAX
};
|
