1
2
3
4
5
6
7
8
9
10 #include "oshmem_config.h"
11
12 #include "oshmem/constants.h"
13 #include "oshmem/include/shmem.h"
14
15 #include "oshmem/runtime/runtime.h"
16
17 #include "oshmem/mca/spml/spml.h"
18
19
20
21
22
23
24
25 #define DO_SHMEM_TYPE_GET(ctx, type, target, source, nelems, pe) do { \
26 int rc = OSHMEM_SUCCESS; \
27 size_t size = 0; \
28 \
29 RUNTIME_CHECK_INIT(); \
30 RUNTIME_CHECK_PE(pe); \
31 RUNTIME_CHECK_ADDR(source); \
32 \
33 size = nelems * sizeof(type); \
34 rc = MCA_SPML_CALL(get( \
35 ctx, \
36 (void*)source, \
37 size, \
38 (void*)target, \
39 pe)); \
40 RUNTIME_CHECK_RC(rc); \
41 } while (0)
42
43 #define SHMEM_CTX_TYPE_GET(type_name, type) \
44 void shmem_ctx##type_name##_get(shmem_ctx_t ctx, type *target, const type *source, size_t nelems, int pe) \
45 { \
46 DO_SHMEM_TYPE_GET(ctx, type, target, source, nelems, pe); \
47 return ; \
48 }
49
50 #define SHMEM_TYPE_GET(type_name, type) \
51 void shmem##type_name##_get(type *target, const type *source, size_t nelems, int pe) \
52 { \
53 DO_SHMEM_TYPE_GET(oshmem_ctx_default, type, target, source, \
54 nelems, pe); \
55 return ; \
56 }
57
58 #if OSHMEM_PROFILING
59 #include "oshmem/include/pshmem.h"
60 #pragma weak shmem_ctx_char_get = pshmem_ctx_char_get
61 #pragma weak shmem_ctx_short_get = pshmem_ctx_short_get
62 #pragma weak shmem_ctx_int_get = pshmem_ctx_int_get
63 #pragma weak shmem_ctx_long_get = pshmem_ctx_long_get
64 #pragma weak shmem_ctx_float_get = pshmem_ctx_float_get
65 #pragma weak shmem_ctx_double_get = pshmem_ctx_double_get
66 #pragma weak shmem_ctx_longlong_get = pshmem_ctx_longlong_get
67 #pragma weak shmem_ctx_schar_get = pshmem_ctx_schar_get
68 #pragma weak shmem_ctx_uchar_get = pshmem_ctx_uchar_get
69 #pragma weak shmem_ctx_ushort_get = pshmem_ctx_ushort_get
70 #pragma weak shmem_ctx_uint_get = pshmem_ctx_uint_get
71 #pragma weak shmem_ctx_ulong_get = pshmem_ctx_ulong_get
72 #pragma weak shmem_ctx_ulonglong_get = pshmem_ctx_ulonglong_get
73 #pragma weak shmem_ctx_longdouble_get = pshmem_ctx_longdouble_get
74 #pragma weak shmem_ctx_int8_get = pshmem_ctx_int8_get
75 #pragma weak shmem_ctx_int16_get = pshmem_ctx_int16_get
76 #pragma weak shmem_ctx_int32_get = pshmem_ctx_int32_get
77 #pragma weak shmem_ctx_int64_get = pshmem_ctx_int64_get
78 #pragma weak shmem_ctx_uint8_get = pshmem_ctx_uint8_get
79 #pragma weak shmem_ctx_uint16_get = pshmem_ctx_uint16_get
80 #pragma weak shmem_ctx_uint32_get = pshmem_ctx_uint32_get
81 #pragma weak shmem_ctx_uint64_get = pshmem_ctx_uint64_get
82 #pragma weak shmem_ctx_size_get = pshmem_ctx_size_get
83 #pragma weak shmem_ctx_ptrdiff_get = pshmem_ctx_ptrdiff_get
84
85 #pragma weak shmem_char_get = pshmem_char_get
86 #pragma weak shmem_short_get = pshmem_short_get
87 #pragma weak shmem_int_get = pshmem_int_get
88 #pragma weak shmem_long_get = pshmem_long_get
89 #pragma weak shmem_float_get = pshmem_float_get
90 #pragma weak shmem_double_get = pshmem_double_get
91 #pragma weak shmem_longlong_get = pshmem_longlong_get
92 #pragma weak shmem_schar_get = pshmem_schar_get
93 #pragma weak shmem_uchar_get = pshmem_uchar_get
94 #pragma weak shmem_ushort_get = pshmem_ushort_get
95 #pragma weak shmem_uint_get = pshmem_uint_get
96 #pragma weak shmem_ulong_get = pshmem_ulong_get
97 #pragma weak shmem_ulonglong_get = pshmem_ulonglong_get
98 #pragma weak shmem_longdouble_get = pshmem_longdouble_get
99 #pragma weak shmem_int8_get = pshmem_int8_get
100 #pragma weak shmem_int16_get = pshmem_int16_get
101 #pragma weak shmem_int32_get = pshmem_int32_get
102 #pragma weak shmem_int64_get = pshmem_int64_get
103 #pragma weak shmem_uint8_get = pshmem_uint8_get
104 #pragma weak shmem_uint16_get = pshmem_uint16_get
105 #pragma weak shmem_uint32_get = pshmem_uint32_get
106 #pragma weak shmem_uint64_get = pshmem_uint64_get
107 #pragma weak shmem_size_get = pshmem_size_get
108 #pragma weak shmem_ptrdiff_get = pshmem_ptrdiff_get
109
110 #pragma weak shmem_ctx_getmem = pshmem_ctx_getmem
111 #pragma weak shmem_ctx_get8 = pshmem_ctx_get8
112 #pragma weak shmem_ctx_get16 = pshmem_ctx_get16
113 #pragma weak shmem_ctx_get32 = pshmem_ctx_get32
114 #pragma weak shmem_ctx_get64 = pshmem_ctx_get64
115 #pragma weak shmem_ctx_get128 = pshmem_ctx_get128
116
117 #pragma weak shmem_getmem = pshmem_getmem
118 #pragma weak shmem_get8 = pshmem_get8
119 #pragma weak shmem_get16 = pshmem_get16
120 #pragma weak shmem_get32 = pshmem_get32
121 #pragma weak shmem_get64 = pshmem_get64
122 #pragma weak shmem_get128 = pshmem_get128
123 #include "oshmem/shmem/c/profile/defines.h"
124 #endif
125
126 SHMEM_CTX_TYPE_GET(_char, char)
127 SHMEM_CTX_TYPE_GET(_short, short)
128 SHMEM_CTX_TYPE_GET(_int, int)
129 SHMEM_CTX_TYPE_GET(_long, long)
130 SHMEM_CTX_TYPE_GET(_longlong, long long)
131 SHMEM_CTX_TYPE_GET(_schar, signed char)
132 SHMEM_CTX_TYPE_GET(_uchar, unsigned char)
133 SHMEM_CTX_TYPE_GET(_ushort, unsigned short)
134 SHMEM_CTX_TYPE_GET(_uint, unsigned int)
135 SHMEM_CTX_TYPE_GET(_ulong, unsigned long)
136 SHMEM_CTX_TYPE_GET(_ulonglong, unsigned long long)
137 SHMEM_CTX_TYPE_GET(_float, float)
138 SHMEM_CTX_TYPE_GET(_double, double)
139 SHMEM_CTX_TYPE_GET(_longdouble, long double)
140 SHMEM_CTX_TYPE_GET(_int8, int8_t)
141 SHMEM_CTX_TYPE_GET(_int16, int16_t)
142 SHMEM_CTX_TYPE_GET(_int32, int32_t)
143 SHMEM_CTX_TYPE_GET(_int64, int64_t)
144 SHMEM_CTX_TYPE_GET(_uint8, uint8_t)
145 SHMEM_CTX_TYPE_GET(_uint16, uint16_t)
146 SHMEM_CTX_TYPE_GET(_uint32, uint32_t)
147 SHMEM_CTX_TYPE_GET(_uint64, uint64_t)
148 SHMEM_CTX_TYPE_GET(_size, size_t)
149 SHMEM_CTX_TYPE_GET(_ptrdiff, ptrdiff_t)
150
151 SHMEM_TYPE_GET(_char, char)
152 SHMEM_TYPE_GET(_short, short)
153 SHMEM_TYPE_GET(_int, int)
154 SHMEM_TYPE_GET(_long, long)
155 SHMEM_TYPE_GET(_longlong, long long)
156 SHMEM_TYPE_GET(_schar, signed char)
157 SHMEM_TYPE_GET(_uchar, unsigned char)
158 SHMEM_TYPE_GET(_ushort, unsigned short)
159 SHMEM_TYPE_GET(_uint, unsigned int)
160 SHMEM_TYPE_GET(_ulong, unsigned long)
161 SHMEM_TYPE_GET(_ulonglong, unsigned long long)
162 SHMEM_TYPE_GET(_float, float)
163 SHMEM_TYPE_GET(_double, double)
164 SHMEM_TYPE_GET(_longdouble, long double)
165 SHMEM_TYPE_GET(_int8, int8_t)
166 SHMEM_TYPE_GET(_int16, int16_t)
167 SHMEM_TYPE_GET(_int32, int32_t)
168 SHMEM_TYPE_GET(_int64, int64_t)
169 SHMEM_TYPE_GET(_uint8, uint8_t)
170 SHMEM_TYPE_GET(_uint16, uint16_t)
171 SHMEM_TYPE_GET(_uint32, uint32_t)
172 SHMEM_TYPE_GET(_uint64, uint64_t)
173 SHMEM_TYPE_GET(_size, size_t)
174 SHMEM_TYPE_GET(_ptrdiff, ptrdiff_t)
175
176 #define DO_SHMEM_GETMEM(ctx, target, source, element_size, nelems, pe) do { \
177 int rc = OSHMEM_SUCCESS; \
178 size_t size = 0; \
179 \
180 RUNTIME_CHECK_INIT(); \
181 RUNTIME_CHECK_PE(pe); \
182 RUNTIME_CHECK_ADDR(source); \
183 \
184 size = nelems * element_size; \
185 rc = MCA_SPML_CALL(get( \
186 ctx, \
187 (void*)source, \
188 size, \
189 (void*)target, \
190 pe)); \
191 RUNTIME_CHECK_RC(rc); \
192 } while (0)
193
194 #define SHMEM_CTX_TYPE_GETMEM(name, element_size, prefix) \
195 void prefix##_ctx##name(shmem_ctx_t ctx, void *target, const void *source, size_t nelems, int pe) \
196 { \
197 DO_SHMEM_GETMEM(ctx, target, source, element_size, nelems, pe); \
198 return ; \
199 }
200
201 #define SHMEM_TYPE_GETMEM(name, element_size, prefix) \
202 void prefix##name(void *target, const void *source, size_t nelems, int pe) \
203 { \
204 DO_SHMEM_GETMEM(oshmem_ctx_default, target, source, \
205 element_size, nelems, pe); \
206 return ; \
207 }
208
209 SHMEM_CTX_TYPE_GETMEM(_getmem, 1, shmem)
210 SHMEM_CTX_TYPE_GETMEM(_get8, 1, shmem)
211 SHMEM_CTX_TYPE_GETMEM(_get16, 2, shmem)
212 SHMEM_CTX_TYPE_GETMEM(_get32, 4, shmem)
213 SHMEM_CTX_TYPE_GETMEM(_get64, 8, shmem)
214 SHMEM_CTX_TYPE_GETMEM(_get128, 16, shmem)
215 SHMEM_TYPE_GETMEM(_getmem, 1, shmem)
216 SHMEM_TYPE_GETMEM(_get8, 1, shmem)
217 SHMEM_TYPE_GETMEM(_get16, 2, shmem)
218 SHMEM_TYPE_GETMEM(_get32, 4, shmem)
219 SHMEM_TYPE_GETMEM(_get64, 8, shmem)
220 SHMEM_TYPE_GETMEM(_get128, 16, shmem)
221