1 /*
2 * kmp_stub.cpp -- stub versions of user-callable OpenMP RT functions.
3 */
4
5 //===----------------------------------------------------------------------===//
6 //
7 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
8 // See https://llvm.org/LICENSE.txt for license information.
9 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include <errno.h>
14 #include <limits.h>
15 #include <stdlib.h>
16
17 #define __KMP_IMP
18 #include "omp.h" // omp_* declarations, must be included before "kmp.h"
19 #include "kmp.h" // KMP_DEFAULT_STKSIZE
20 #include "kmp_stub.h"
21
22 #if KMP_OS_WINDOWS
23 #include <windows.h>
24 #else
25 #include <sys/time.h>
26 #endif
27
28 // Moved from omp.h
29 #define omp_set_max_active_levels ompc_set_max_active_levels
30 #define omp_set_schedule ompc_set_schedule
31 #define omp_get_ancestor_thread_num ompc_get_ancestor_thread_num
32 #define omp_get_team_size ompc_get_team_size
33
34 #define omp_set_num_threads ompc_set_num_threads
35 #define omp_set_dynamic ompc_set_dynamic
36 #define omp_set_nested ompc_set_nested
37 #define omp_set_affinity_format ompc_set_affinity_format
38 #define omp_get_affinity_format ompc_get_affinity_format
39 #define omp_display_affinity ompc_display_affinity
40 #define omp_capture_affinity ompc_capture_affinity
41 #define kmp_set_stacksize kmpc_set_stacksize
42 #define kmp_set_stacksize_s kmpc_set_stacksize_s
43 #define kmp_set_blocktime kmpc_set_blocktime
44 #define kmp_set_library kmpc_set_library
45 #define kmp_set_defaults kmpc_set_defaults
46 #define kmp_set_disp_num_buffers kmpc_set_disp_num_buffers
47 #define kmp_malloc kmpc_malloc
48 #define kmp_aligned_malloc kmpc_aligned_malloc
49 #define kmp_calloc kmpc_calloc
50 #define kmp_realloc kmpc_realloc
51 #define kmp_free kmpc_free
52
53 #if KMP_OS_WINDOWS
54 static double frequency = 0.0;
55 #endif
56
57 // Helper functions.
__kmps_init()58 static size_t __kmps_init() {
59 static int initialized = 0;
60 static size_t dummy = 0;
61 if (!initialized) {
62 // TODO: Analyze KMP_VERSION environment variable, print
63 // __kmp_version_copyright and __kmp_version_build_time.
64 // WARNING: Do not use "fprintf(stderr, ...)" because it will cause
65 // unresolved "__iob" symbol (see C70080). We need to extract __kmp_printf()
66 // stuff from kmp_runtime.cpp and use it.
67
68 // Trick with dummy variable forces linker to keep __kmp_version_copyright
69 // and __kmp_version_build_time strings in executable file (in case of
70 // static linkage). When KMP_VERSION analysis is implemented, dummy
71 // variable should be deleted, function should return void.
72 dummy = __kmp_version_copyright - __kmp_version_build_time;
73
74 #if KMP_OS_WINDOWS
75 LARGE_INTEGER freq;
76 BOOL status = QueryPerformanceFrequency(&freq);
77 if (status) {
78 frequency = double(freq.QuadPart);
79 }
80 #endif
81
82 initialized = 1;
83 }
84 return dummy;
85 } // __kmps_init
86
87 #define i __kmps_init();
88
89 /* set API functions */
omp_set_num_threads(omp_int_t num_threads)90 void omp_set_num_threads(omp_int_t num_threads) { i; }
omp_set_dynamic(omp_int_t dynamic)91 void omp_set_dynamic(omp_int_t dynamic) {
92 i;
93 __kmps_set_dynamic(dynamic);
94 }
omp_set_nested(omp_int_t nested)95 void omp_set_nested(omp_int_t nested) {
96 i;
97 __kmps_set_nested(nested);
98 }
omp_set_max_active_levels(omp_int_t max_active_levels)99 void omp_set_max_active_levels(omp_int_t max_active_levels) { i; }
omp_set_schedule(omp_sched_t kind,omp_int_t modifier)100 void omp_set_schedule(omp_sched_t kind, omp_int_t modifier) {
101 i;
102 __kmps_set_schedule((kmp_sched_t)kind, modifier);
103 }
omp_get_ancestor_thread_num(omp_int_t level)104 int omp_get_ancestor_thread_num(omp_int_t level) {
105 i;
106 return (level) ? (-1) : (0);
107 }
omp_get_team_size(omp_int_t level)108 int omp_get_team_size(omp_int_t level) {
109 i;
110 return (level) ? (-1) : (1);
111 }
kmpc_set_affinity_mask_proc(int proc,void ** mask)112 int kmpc_set_affinity_mask_proc(int proc, void **mask) {
113 i;
114 return -1;
115 }
kmpc_unset_affinity_mask_proc(int proc,void ** mask)116 int kmpc_unset_affinity_mask_proc(int proc, void **mask) {
117 i;
118 return -1;
119 }
kmpc_get_affinity_mask_proc(int proc,void ** mask)120 int kmpc_get_affinity_mask_proc(int proc, void **mask) {
121 i;
122 return -1;
123 }
124
125 /* kmp API functions */
kmp_set_stacksize(omp_int_t arg)126 void kmp_set_stacksize(omp_int_t arg) {
127 i;
128 __kmps_set_stacksize(arg);
129 }
kmp_set_stacksize_s(size_t arg)130 void kmp_set_stacksize_s(size_t arg) {
131 i;
132 __kmps_set_stacksize(arg);
133 }
kmp_set_blocktime(omp_int_t arg)134 void kmp_set_blocktime(omp_int_t arg) {
135 i;
136 __kmps_set_blocktime(arg);
137 }
kmp_set_library(omp_int_t arg)138 void kmp_set_library(omp_int_t arg) {
139 i;
140 __kmps_set_library(arg);
141 }
kmp_set_defaults(char const * str)142 void kmp_set_defaults(char const *str) { i; }
kmp_set_disp_num_buffers(omp_int_t arg)143 void kmp_set_disp_num_buffers(omp_int_t arg) { i; }
144
145 /* KMP memory management functions. */
kmp_malloc(size_t size)146 void *kmp_malloc(size_t size) {
147 i;
148 void *res;
149 #if KMP_OS_WINDOWS
150 // If successful returns a pointer to the memory block, otherwise returns
151 // NULL.
152 // Sets errno to ENOMEM or EINVAL if memory allocation failed or parameter
153 // validation failed.
154 res = _aligned_malloc(size, 1);
155 #else
156 res = malloc(size);
157 #endif
158 return res;
159 }
kmp_aligned_malloc(size_t sz,size_t a)160 void *kmp_aligned_malloc(size_t sz, size_t a) {
161 i;
162 int err;
163 void *res;
164 #if KMP_OS_WINDOWS
165 res = _aligned_malloc(sz, a);
166 #else
167 if ((err = posix_memalign(&res, a, sz))) {
168 errno = err; // can be EINVAL or ENOMEM
169 res = NULL;
170 }
171 #endif
172 return res;
173 }
kmp_calloc(size_t nelem,size_t elsize)174 void *kmp_calloc(size_t nelem, size_t elsize) {
175 i;
176 void *res;
177 #if KMP_OS_WINDOWS
178 res = _aligned_recalloc(NULL, nelem, elsize, 1);
179 #else
180 res = calloc(nelem, elsize);
181 #endif
182 return res;
183 }
kmp_realloc(void * ptr,size_t size)184 void *kmp_realloc(void *ptr, size_t size) {
185 i;
186 void *res;
187 #if KMP_OS_WINDOWS
188 res = _aligned_realloc(ptr, size, 1);
189 #else
190 res = realloc(ptr, size);
191 #endif
192 return res;
193 }
kmp_free(void * ptr)194 void kmp_free(void *ptr) {
195 i;
196 #if KMP_OS_WINDOWS
197 _aligned_free(ptr);
198 #else
199 free(ptr);
200 #endif
201 }
202
203 static int __kmps_blocktime = INT_MAX;
204
__kmps_set_blocktime(int arg)205 void __kmps_set_blocktime(int arg) {
206 i;
207 __kmps_blocktime = arg;
208 } // __kmps_set_blocktime
209
__kmps_get_blocktime(void)210 int __kmps_get_blocktime(void) {
211 i;
212 return __kmps_blocktime;
213 } // __kmps_get_blocktime
214
215 static int __kmps_dynamic = 0;
216
__kmps_set_dynamic(int arg)217 void __kmps_set_dynamic(int arg) {
218 i;
219 __kmps_dynamic = arg;
220 } // __kmps_set_dynamic
221
__kmps_get_dynamic(void)222 int __kmps_get_dynamic(void) {
223 i;
224 return __kmps_dynamic;
225 } // __kmps_get_dynamic
226
227 static int __kmps_library = 1000;
228
__kmps_set_library(int arg)229 void __kmps_set_library(int arg) {
230 i;
231 __kmps_library = arg;
232 } // __kmps_set_library
233
__kmps_get_library(void)234 int __kmps_get_library(void) {
235 i;
236 return __kmps_library;
237 } // __kmps_get_library
238
239 static int __kmps_nested = 0;
240
__kmps_set_nested(int arg)241 void __kmps_set_nested(int arg) {
242 i;
243 __kmps_nested = arg;
244 } // __kmps_set_nested
245
__kmps_get_nested(void)246 int __kmps_get_nested(void) {
247 i;
248 return __kmps_nested;
249 } // __kmps_get_nested
250
251 static size_t __kmps_stacksize = KMP_DEFAULT_STKSIZE;
252
__kmps_set_stacksize(int arg)253 void __kmps_set_stacksize(int arg) {
254 i;
255 __kmps_stacksize = arg;
256 } // __kmps_set_stacksize
257
__kmps_get_stacksize(void)258 int __kmps_get_stacksize(void) {
259 i;
260 return __kmps_stacksize;
261 } // __kmps_get_stacksize
262
263 static kmp_sched_t __kmps_sched_kind = kmp_sched_default;
264 static int __kmps_sched_modifier = 0;
265
__kmps_set_schedule(kmp_sched_t kind,int modifier)266 void __kmps_set_schedule(kmp_sched_t kind, int modifier) {
267 i;
268 __kmps_sched_kind = kind;
269 __kmps_sched_modifier = modifier;
270 } // __kmps_set_schedule
271
__kmps_get_schedule(kmp_sched_t * kind,int * modifier)272 void __kmps_get_schedule(kmp_sched_t *kind, int *modifier) {
273 i;
274 *kind = __kmps_sched_kind;
275 *modifier = __kmps_sched_modifier;
276 } // __kmps_get_schedule
277
__kmps_get_proc_bind(void)278 kmp_proc_bind_t __kmps_get_proc_bind(void) {
279 i;
280 return proc_bind_false;
281 } // __kmps_get_proc_bind
282
__kmps_get_wtime(void)283 double __kmps_get_wtime(void) {
284 // Elapsed wall clock time (in second) from "sometime in the past".
285 double wtime = 0.0;
286 i;
287 #if KMP_OS_WINDOWS
288 if (frequency > 0.0) {
289 LARGE_INTEGER now;
290 BOOL status = QueryPerformanceCounter(&now);
291 if (status) {
292 wtime = double(now.QuadPart) / frequency;
293 }
294 }
295 #else
296 // gettimeofday() returns seconds and microseconds since the Epoch.
297 struct timeval tval;
298 int rc;
299 rc = gettimeofday(&tval, NULL);
300 if (rc == 0) {
301 wtime = (double)(tval.tv_sec) + 1.0E-06 * (double)(tval.tv_usec);
302 } else {
303 // TODO: Assert or abort here.
304 }
305 #endif
306 return wtime;
307 } // __kmps_get_wtime
308
__kmps_get_wtick(void)309 double __kmps_get_wtick(void) {
310 // Number of seconds between successive clock ticks.
311 double wtick = 0.0;
312 i;
313 #if KMP_OS_WINDOWS
314 {
315 DWORD increment;
316 DWORD adjustment;
317 BOOL disabled;
318 BOOL rc;
319 rc = GetSystemTimeAdjustment(&adjustment, &increment, &disabled);
320 if (rc) {
321 wtick = 1.0E-07 * (double)(disabled ? increment : adjustment);
322 } else {
323 // TODO: Assert or abort here.
324 wtick = 1.0E-03;
325 }
326 }
327 #else
328 // TODO: gettimeofday() returns in microseconds, but what the precision?
329 wtick = 1.0E-06;
330 #endif
331 return wtick;
332 } // __kmps_get_wtick
333
334 /* OpenMP 5.0 Memory Management */
335 #if KMP_OS_WINDOWS
336 omp_allocator_handle_t const omp_null_allocator = 0;
337 omp_allocator_handle_t const omp_default_mem_alloc =
338 (omp_allocator_handle_t const)1;
339 omp_allocator_handle_t const omp_large_cap_mem_alloc =
340 (omp_allocator_handle_t const)2;
341 omp_allocator_handle_t const omp_const_mem_alloc =
342 (omp_allocator_handle_t const)3;
343 omp_allocator_handle_t const omp_high_bw_mem_alloc =
344 (omp_allocator_handle_t const)4;
345 omp_allocator_handle_t const omp_low_lat_mem_alloc =
346 (omp_allocator_handle_t const)5;
347 omp_allocator_handle_t const omp_cgroup_mem_alloc =
348 (omp_allocator_handle_t const)6;
349 omp_allocator_handle_t const omp_pteam_mem_alloc =
350 (omp_allocator_handle_t const)7;
351 omp_allocator_handle_t const omp_thread_mem_alloc =
352 (omp_allocator_handle_t const)8;
353
354 omp_memspace_handle_t const omp_default_mem_space =
355 (omp_memspace_handle_t const)0;
356 omp_memspace_handle_t const omp_large_cap_mem_space =
357 (omp_memspace_handle_t const)1;
358 omp_memspace_handle_t const omp_const_mem_space =
359 (omp_memspace_handle_t const)2;
360 omp_memspace_handle_t const omp_high_bw_mem_space =
361 (omp_memspace_handle_t const)3;
362 omp_memspace_handle_t const omp_low_lat_mem_space =
363 (omp_memspace_handle_t const)4;
364 #endif /* KMP_OS_WINDOWS */
omp_alloc(size_t size,const omp_allocator_handle_t allocator)365 void *omp_alloc(size_t size, const omp_allocator_handle_t allocator) {
366 i;
367 return malloc(size);
368 }
omp_calloc(size_t nmemb,size_t size,const omp_allocator_handle_t allocator)369 void *omp_calloc(size_t nmemb, size_t size,
370 const omp_allocator_handle_t allocator) {
371 i;
372 return calloc(nmemb, size);
373 }
omp_realloc(void * ptr,size_t size,const omp_allocator_handle_t allocator,const omp_allocator_handle_t free_allocator)374 void *omp_realloc(void *ptr, size_t size,
375 const omp_allocator_handle_t allocator,
376 const omp_allocator_handle_t free_allocator) {
377 i;
378 return realloc(ptr, size);
379 }
omp_free(void * ptr,const omp_allocator_handle_t allocator)380 void omp_free(void *ptr, const omp_allocator_handle_t allocator) {
381 i;
382 free(ptr);
383 }
384 /* OpenMP 5.0 Affinity Format */
omp_set_affinity_format(char const * format)385 void omp_set_affinity_format(char const *format) { i; }
omp_get_affinity_format(char * buffer,size_t size)386 size_t omp_get_affinity_format(char *buffer, size_t size) {
387 i;
388 return 0;
389 }
omp_display_affinity(char const * format)390 void omp_display_affinity(char const *format) { i; }
omp_capture_affinity(char * buffer,size_t buf_size,char const * format)391 size_t omp_capture_affinity(char *buffer, size_t buf_size, char const *format) {
392 i;
393 return 0;
394 }
395
396 // end of file //
397