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
|
/*
* Copyright (c) 2016-2017 Inria. All rights reserved.
* Copyright (c) 2017-2019 Research Organization for Information Science
* and Technology (RIST). All rights reserved.
* Copyright (c) 2017 Amazon.com, Inc. or its affiliates. All Rights
* reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#ifndef MCA_OSC_MONITORING_MODULE_H
#define MCA_OSC_MONITORING_MODULE_H
#include "ompi/info/info.h"
#include "ompi/win/win.h"
#include "ompi/mca/osc/osc.h"
/* Define once and for all the module_template variable name */
#define OMPI_OSC_MONITORING_MODULE_VARIABLE(template) \
ompi_osc_monitoring_module_## template ##_template
/* Define once and for all the
* ompi_osc_monitoring_## template ##_set_template function name
*/
#define OSC_MONITORING_SET_TEMPLATE_FCT_NAME(template) \
ompi_osc_monitoring_## template ##_set_template
/* Define the ompi_osc_monitoring_module_## template ##_template
* variable
*/
#define OMPI_OSC_MONITORING_MODULE_GENERATE(template) \
/* Define the ompi_osc_monitoring_module_## template ##_template */ \
static ompi_osc_base_module_t OMPI_OSC_MONITORING_MODULE_VARIABLE(template);
#define OSC_MONITORING_GENERATE_TEMPLATE_MODULE(template) \
\
static int ompi_osc_monitoring_## template ##_free(ompi_win_t *win) \
{ \
return OMPI_OSC_MONITORING_MODULE_VARIABLE(template).osc_free(win); \
}
#define MCA_OSC_MONITORING_MODULE_TEMPLATE_GENERATE(template) \
/* Generate template specific module initialization function: \
* ompi_osc_monitoring_## template ##_set_template(ompi_osc_base_module_t*module) \
*/ \
static inline ompi_osc_base_module_t * \
OSC_MONITORING_SET_TEMPLATE_FCT_NAME(template) (ompi_osc_base_module_t*module) \
{ \
/* Define the ompi_osc_monitoring_module_## template ##_init_done variable */ \
opal_atomic_int32_t init_done = 0; \
/* Define and set the ompi_osc_monitoring_## template \
* ##_template variable. The functions recorded here are \
* linked to the original functions of the original \
* {template} module that was replaced. \
*/ \
static const ompi_osc_base_module_t module_specific_interception_layer = { \
.osc_win_attach = ompi_osc_monitoring_## template ##_attach, \
.osc_win_detach = ompi_osc_monitoring_## template ##_detach, \
.osc_free = ompi_osc_monitoring_## template ##_free, \
\
.osc_put = ompi_osc_monitoring_## template ##_put, \
.osc_get = ompi_osc_monitoring_## template ##_get, \
.osc_accumulate = ompi_osc_monitoring_## template ##_accumulate, \
.osc_compare_and_swap = ompi_osc_monitoring_## template ##_compare_and_swap, \
.osc_fetch_and_op = ompi_osc_monitoring_## template ##_fetch_and_op, \
.osc_get_accumulate = ompi_osc_monitoring_## template ##_get_accumulate, \
\
.osc_rput = ompi_osc_monitoring_## template ##_rput, \
.osc_rget = ompi_osc_monitoring_## template ##_rget, \
.osc_raccumulate = ompi_osc_monitoring_## template ##_raccumulate, \
.osc_rget_accumulate = ompi_osc_monitoring_## template ##_rget_accumulate, \
\
.osc_fence = ompi_osc_monitoring_## template ##_fence, \
\
.osc_start = ompi_osc_monitoring_## template ##_start, \
.osc_complete = ompi_osc_monitoring_## template ##_complete, \
.osc_post = ompi_osc_monitoring_## template ##_post, \
.osc_wait = ompi_osc_monitoring_## template ##_wait, \
.osc_test = ompi_osc_monitoring_## template ##_test, \
\
.osc_lock = ompi_osc_monitoring_## template ##_lock, \
.osc_unlock = ompi_osc_monitoring_## template ##_unlock, \
.osc_lock_all = ompi_osc_monitoring_## template ##_lock_all, \
.osc_unlock_all = ompi_osc_monitoring_## template ##_unlock_all, \
\
.osc_sync = ompi_osc_monitoring_## template ##_sync, \
.osc_flush = ompi_osc_monitoring_## template ##_flush, \
.osc_flush_all = ompi_osc_monitoring_## template ##_flush_all, \
.osc_flush_local = ompi_osc_monitoring_## template ##_flush_local, \
.osc_flush_local_all = ompi_osc_monitoring_## template ##_flush_local_all, \
}; \
if ( 1 == opal_atomic_add_fetch_32(&init_done, 1) ) { \
/* Saves the original module functions in \
* ompi_osc_monitoring_module_## template ##_template \
*/ \
memcpy(&OMPI_OSC_MONITORING_MODULE_VARIABLE(template), \
module, sizeof(ompi_osc_base_module_t)); \
} \
/* Replace the original functions with our generated ones */ \
memcpy(module, &module_specific_interception_layer, \
sizeof(ompi_osc_base_module_t)); \
return module; \
}
#endif /* MCA_OSC_MONITORING_MODULE_H */
|
