1 | /* |
2 | * kmp_settings.cpp -- Initialize environment variables |
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 "kmp.h" |
14 | #include "kmp_affinity.h" |
15 | #include "kmp_atomic.h" |
16 | #if KMP_USE_HIER_SCHED |
17 | #include "kmp_dispatch_hier.h" |
18 | #endif |
19 | #include "kmp_environment.h" |
20 | #include "kmp_i18n.h" |
21 | #include "kmp_io.h" |
22 | #include "kmp_itt.h" |
23 | #include "kmp_lock.h" |
24 | #include "kmp_settings.h" |
25 | #include "kmp_str.h" |
26 | #include "kmp_wrapper_getpid.h" |
27 | #include <ctype.h> // toupper() |
28 | #if OMPD_SUPPORT |
29 | #include "ompd-specific.h" |
30 | #endif |
31 | |
32 | static int __kmp_env_toPrint(char const *name, int flag); |
33 | |
34 | bool __kmp_env_format = 0; // 0 - old format; 1 - new format |
35 | |
36 | // ----------------------------------------------------------------------------- |
37 | // Helper string functions. Subject to move to kmp_str. |
38 | |
39 | #ifdef USE_LOAD_BALANCE |
40 | static double __kmp_convert_to_double(char const *s) { |
41 | double result; |
42 | |
43 | if (KMP_SSCANF(s: s, format: "%lf" , &result) < 1) { |
44 | result = 0.0; |
45 | } |
46 | |
47 | return result; |
48 | } |
49 | #endif |
50 | |
51 | #ifdef KMP_DEBUG |
52 | static unsigned int __kmp_readstr_with_sentinel(char *dest, char const *src, |
53 | size_t len, char sentinel) { |
54 | unsigned int i; |
55 | for (i = 0; i < len; i++) { |
56 | if ((*src == '\0') || (*src == sentinel)) { |
57 | break; |
58 | } |
59 | *(dest++) = *(src++); |
60 | } |
61 | *dest = '\0'; |
62 | return i; |
63 | } |
64 | #endif |
65 | |
66 | static int __kmp_match_with_sentinel(char const *a, char const *b, size_t len, |
67 | char sentinel) { |
68 | size_t l = 0; |
69 | |
70 | if (a == NULL) |
71 | a = "" ; |
72 | if (b == NULL) |
73 | b = "" ; |
74 | while (*a && *b && *b != sentinel) { |
75 | char ca = *a, cb = *b; |
76 | |
77 | if (ca >= 'a' && ca <= 'z') |
78 | ca -= 'a' - 'A'; |
79 | if (cb >= 'a' && cb <= 'z') |
80 | cb -= 'a' - 'A'; |
81 | if (ca != cb) |
82 | return FALSE; |
83 | ++l; |
84 | ++a; |
85 | ++b; |
86 | } |
87 | return l >= len; |
88 | } |
89 | |
90 | // Expected usage: |
91 | // token is the token to check for. |
92 | // buf is the string being parsed. |
93 | // *end returns the char after the end of the token. |
94 | // it is not modified unless a match occurs. |
95 | // |
96 | // Example 1: |
97 | // |
98 | // if (__kmp_match_str("token", buf, *end) { |
99 | // <do something> |
100 | // buf = end; |
101 | // } |
102 | // |
103 | // Example 2: |
104 | // |
105 | // if (__kmp_match_str("token", buf, *end) { |
106 | // char *save = **end; |
107 | // **end = sentinel; |
108 | // <use any of the __kmp*_with_sentinel() functions> |
109 | // **end = save; |
110 | // buf = end; |
111 | // } |
112 | |
113 | static int __kmp_match_str(char const *token, char const *buf, |
114 | const char **end) { |
115 | |
116 | KMP_ASSERT(token != NULL); |
117 | KMP_ASSERT(buf != NULL); |
118 | KMP_ASSERT(end != NULL); |
119 | |
120 | while (*token && *buf) { |
121 | char ct = *token, cb = *buf; |
122 | |
123 | if (ct >= 'a' && ct <= 'z') |
124 | ct -= 'a' - 'A'; |
125 | if (cb >= 'a' && cb <= 'z') |
126 | cb -= 'a' - 'A'; |
127 | if (ct != cb) |
128 | return FALSE; |
129 | ++token; |
130 | ++buf; |
131 | } |
132 | if (*token) { |
133 | return FALSE; |
134 | } |
135 | *end = buf; |
136 | return TRUE; |
137 | } |
138 | |
139 | #if KMP_OS_DARWIN |
140 | static size_t __kmp_round4k(size_t size) { |
141 | size_t _4k = 4 * 1024; |
142 | if (size & (_4k - 1)) { |
143 | size &= ~(_4k - 1); |
144 | if (size <= KMP_SIZE_T_MAX - _4k) { |
145 | size += _4k; // Round up if there is no overflow. |
146 | } |
147 | } |
148 | return size; |
149 | } // __kmp_round4k |
150 | #endif |
151 | |
152 | static int __kmp_strcasecmp_with_sentinel(char const *a, char const *b, |
153 | char sentinel) { |
154 | if (a == NULL) |
155 | a = "" ; |
156 | if (b == NULL) |
157 | b = "" ; |
158 | while (*a && *b && *b != sentinel) { |
159 | char ca = *a, cb = *b; |
160 | |
161 | if (ca >= 'a' && ca <= 'z') |
162 | ca -= 'a' - 'A'; |
163 | if (cb >= 'a' && cb <= 'z') |
164 | cb -= 'a' - 'A'; |
165 | if (ca != cb) |
166 | return (int)(unsigned char)*a - (int)(unsigned char)*b; |
167 | ++a; |
168 | ++b; |
169 | } |
170 | return *a ? (*b && *b != sentinel) |
171 | ? (int)(unsigned char)*a - (int)(unsigned char)*b |
172 | : 1 |
173 | : (*b && *b != sentinel) ? -1 |
174 | : 0; |
175 | } |
176 | |
177 | // ============================================================================= |
178 | // Table structures and helper functions. |
179 | |
180 | typedef struct __kmp_setting kmp_setting_t; |
181 | typedef struct __kmp_stg_ss_data kmp_stg_ss_data_t; |
182 | typedef struct __kmp_stg_wp_data kmp_stg_wp_data_t; |
183 | typedef struct __kmp_stg_fr_data kmp_stg_fr_data_t; |
184 | |
185 | typedef void (*kmp_stg_parse_func_t)(char const *name, char const *value, |
186 | void *data); |
187 | typedef void (*kmp_stg_print_func_t)(kmp_str_buf_t *buffer, char const *name, |
188 | void *data); |
189 | |
190 | struct __kmp_setting { |
191 | char const *name; // Name of setting (environment variable). |
192 | kmp_stg_parse_func_t parse; // Parser function. |
193 | kmp_stg_print_func_t print; // Print function. |
194 | void *data; // Data passed to parser and printer. |
195 | int set; // Variable set during this "session" |
196 | // (__kmp_env_initialize() or kmp_set_defaults() call). |
197 | int defined; // Variable set in any "session". |
198 | }; // struct __kmp_setting |
199 | |
200 | struct __kmp_stg_ss_data { |
201 | size_t factor; // Default factor: 1 for KMP_STACKSIZE, 1024 for others. |
202 | kmp_setting_t **rivals; // Array of pointers to rivals (including itself). |
203 | }; // struct __kmp_stg_ss_data |
204 | |
205 | struct __kmp_stg_wp_data { |
206 | int omp; // 0 -- KMP_LIBRARY, 1 -- OMP_WAIT_POLICY. |
207 | kmp_setting_t **rivals; // Array of pointers to rivals (including itself). |
208 | }; // struct __kmp_stg_wp_data |
209 | |
210 | struct __kmp_stg_fr_data { |
211 | int force; // 0 -- KMP_DETERMINISTIC_REDUCTION, 1 -- KMP_FORCE_REDUCTION. |
212 | kmp_setting_t **rivals; // Array of pointers to rivals (including itself). |
213 | }; // struct __kmp_stg_fr_data |
214 | |
215 | static int __kmp_stg_check_rivals( // 0 -- Ok, 1 -- errors found. |
216 | char const *name, // Name of variable. |
217 | char const *value, // Value of the variable. |
218 | kmp_setting_t **rivals // List of rival settings (must include current one). |
219 | ); |
220 | |
221 | // Helper struct that trims heading/trailing white spaces |
222 | struct kmp_trimmed_str_t { |
223 | kmp_str_buf_t buf; |
224 | kmp_trimmed_str_t(const char *str) { |
225 | __kmp_str_buf_init(&buf); |
226 | size_t len = KMP_STRLEN(s: str); |
227 | if (len == 0) |
228 | return; |
229 | const char *begin = str; |
230 | const char *end = str + KMP_STRLEN(s: str) - 1; |
231 | SKIP_WS(begin); |
232 | while (begin < end && *end == ' ') |
233 | end--; |
234 | __kmp_str_buf_cat(buffer: &buf, str: begin, len: end - begin + 1); |
235 | } |
236 | ~kmp_trimmed_str_t() { __kmp_str_buf_free(buffer: &buf); } |
237 | const char *get() { return buf.str; } |
238 | }; |
239 | |
240 | // ----------------------------------------------------------------------------- |
241 | // Helper parse functions. |
242 | |
243 | static void __kmp_stg_parse_bool(char const *name, char const *value, |
244 | int *out) { |
245 | if (__kmp_str_match_true(data: value)) { |
246 | *out = TRUE; |
247 | } else if (__kmp_str_match_false(data: value)) { |
248 | *out = FALSE; |
249 | } else { |
250 | __kmp_msg(kmp_ms_warning, KMP_MSG(BadBoolValue, name, value), |
251 | KMP_HNT(ValidBoolValues), __kmp_msg_null); |
252 | } |
253 | } // __kmp_stg_parse_bool |
254 | |
255 | // placed here in order to use __kmp_round4k static function |
256 | void __kmp_check_stksize(size_t *val) { |
257 | // if system stack size is too big then limit the size for worker threads |
258 | #if KMP_OS_AIX |
259 | if (*val > KMP_DEFAULT_STKSIZE * 2) // Use 2 times, 16 is too large for AIX. |
260 | *val = KMP_DEFAULT_STKSIZE * 2; |
261 | #else |
262 | if (*val > KMP_DEFAULT_STKSIZE * 16) // just a heuristics... |
263 | *val = KMP_DEFAULT_STKSIZE * 16; |
264 | #endif |
265 | if (*val < __kmp_sys_min_stksize) |
266 | *val = __kmp_sys_min_stksize; |
267 | if (*val > KMP_MAX_STKSIZE) |
268 | *val = KMP_MAX_STKSIZE; // dead code currently, but may work in future |
269 | #if KMP_OS_DARWIN |
270 | *val = __kmp_round4k(*val); |
271 | #endif // KMP_OS_DARWIN |
272 | } |
273 | |
274 | static void __kmp_stg_parse_size(char const *name, char const *value, |
275 | size_t size_min, size_t size_max, |
276 | int *is_specified, size_t *out, |
277 | size_t factor) { |
278 | char const *msg = NULL; |
279 | #if KMP_OS_DARWIN |
280 | size_min = __kmp_round4k(size_min); |
281 | size_max = __kmp_round4k(size_max); |
282 | #endif // KMP_OS_DARWIN |
283 | if (value) { |
284 | if (is_specified != NULL) { |
285 | *is_specified = 1; |
286 | } |
287 | __kmp_str_to_size(str: value, out, dfactor: factor, error: &msg); |
288 | if (msg == NULL) { |
289 | if (*out > size_max) { |
290 | *out = size_max; |
291 | msg = KMP_I18N_STR(ValueTooLarge); |
292 | } else if (*out < size_min) { |
293 | *out = size_min; |
294 | msg = KMP_I18N_STR(ValueTooSmall); |
295 | } else { |
296 | #if KMP_OS_DARWIN |
297 | size_t round4k = __kmp_round4k(*out); |
298 | if (*out != round4k) { |
299 | *out = round4k; |
300 | msg = KMP_I18N_STR(NotMultiple4K); |
301 | } |
302 | #endif |
303 | } |
304 | } else { |
305 | // If integer overflow occurred, * out == KMP_SIZE_T_MAX. Cut it to |
306 | // size_max silently. |
307 | if (*out < size_min) { |
308 | *out = size_max; |
309 | } else if (*out > size_max) { |
310 | *out = size_max; |
311 | } |
312 | } |
313 | if (msg != NULL) { |
314 | // Message is not empty. Print warning. |
315 | kmp_str_buf_t buf; |
316 | __kmp_str_buf_init(&buf); |
317 | __kmp_str_buf_print_size(buffer: &buf, size: *out); |
318 | KMP_WARNING(ParseSizeIntWarn, name, value, msg); |
319 | KMP_INFORM(Using_str_Value, name, buf.str); |
320 | __kmp_str_buf_free(buffer: &buf); |
321 | } |
322 | } |
323 | } // __kmp_stg_parse_size |
324 | |
325 | static void __kmp_stg_parse_str(char const *name, char const *value, |
326 | char **out) { |
327 | __kmp_str_free(str: out); |
328 | *out = __kmp_str_format(format: "%s" , value); |
329 | } // __kmp_stg_parse_str |
330 | |
331 | static void __kmp_stg_parse_int( |
332 | char const |
333 | *name, // I: Name of environment variable (used in warning messages). |
334 | char const *value, // I: Value of environment variable to parse. |
335 | int min, // I: Minimum allowed value. |
336 | int max, // I: Maximum allowed value. |
337 | int *out // O: Output (parsed) value. |
338 | ) { |
339 | char const *msg = NULL; |
340 | kmp_uint64 uint = *out; |
341 | __kmp_str_to_uint(str: value, out: &uint, error: &msg); |
342 | if (msg == NULL) { |
343 | if (uint < (unsigned int)min) { |
344 | msg = KMP_I18N_STR(ValueTooSmall); |
345 | uint = min; |
346 | } else if (uint > (unsigned int)max) { |
347 | msg = KMP_I18N_STR(ValueTooLarge); |
348 | uint = max; |
349 | } |
350 | } else { |
351 | // If overflow occurred msg contains error message and uint is very big. Cut |
352 | // tmp it to INT_MAX. |
353 | if (uint < (unsigned int)min) { |
354 | uint = min; |
355 | } else if (uint > (unsigned int)max) { |
356 | uint = max; |
357 | } |
358 | } |
359 | if (msg != NULL) { |
360 | // Message is not empty. Print warning. |
361 | kmp_str_buf_t buf; |
362 | KMP_WARNING(ParseSizeIntWarn, name, value, msg); |
363 | __kmp_str_buf_init(&buf); |
364 | __kmp_str_buf_print(buffer: &buf, format: "%" KMP_UINT64_SPEC "" , uint); |
365 | KMP_INFORM(Using_uint64_Value, name, buf.str); |
366 | __kmp_str_buf_free(buffer: &buf); |
367 | } |
368 | __kmp_type_convert(src: uint, dest: out); |
369 | } // __kmp_stg_parse_int |
370 | |
371 | #if KMP_DEBUG_ADAPTIVE_LOCKS |
372 | static void __kmp_stg_parse_file(char const *name, char const *value, |
373 | const char *suffix, char **out) { |
374 | char buffer[256]; |
375 | char *t; |
376 | int hasSuffix; |
377 | __kmp_str_free(out); |
378 | t = (char *)strrchr(value, '.'); |
379 | hasSuffix = t && __kmp_str_eqf(t, suffix); |
380 | t = __kmp_str_format("%s%s" , value, hasSuffix ? "" : suffix); |
381 | __kmp_expand_file_name(buffer, sizeof(buffer), t); |
382 | __kmp_str_free(&t); |
383 | *out = __kmp_str_format("%s" , buffer); |
384 | } // __kmp_stg_parse_file |
385 | #endif |
386 | |
387 | #ifdef KMP_DEBUG |
388 | static char *par_range_to_print = NULL; |
389 | |
390 | static void __kmp_stg_parse_par_range(char const *name, char const *value, |
391 | int *out_range, char *out_routine, |
392 | char *out_file, int *out_lb, |
393 | int *out_ub) { |
394 | const char *par_range_value; |
395 | size_t len = KMP_STRLEN(s: value) + 1; |
396 | par_range_to_print = (char *)KMP_INTERNAL_MALLOC(len + 1); |
397 | KMP_STRNCPY_S(par_range_to_print, len + 1, value, len + 1); |
398 | __kmp_par_range = +1; |
399 | __kmp_par_range_lb = 0; |
400 | __kmp_par_range_ub = INT_MAX; |
401 | for (;;) { |
402 | unsigned int len; |
403 | if (!value || *value == '\0') { |
404 | break; |
405 | } |
406 | if (!__kmp_strcasecmp_with_sentinel(a: "routine" , b: value, sentinel: '=')) { |
407 | par_range_value = strchr(s: value, c: '=') + 1; |
408 | if (!par_range_value) |
409 | goto par_range_error; |
410 | value = par_range_value; |
411 | len = __kmp_readstr_with_sentinel(dest: out_routine, src: value, |
412 | KMP_PAR_RANGE_ROUTINE_LEN - 1, sentinel: ','); |
413 | if (len == 0) { |
414 | goto par_range_error; |
415 | } |
416 | value = strchr(s: value, c: ','); |
417 | if (value != NULL) { |
418 | value++; |
419 | } |
420 | continue; |
421 | } |
422 | if (!__kmp_strcasecmp_with_sentinel(a: "filename" , b: value, sentinel: '=')) { |
423 | par_range_value = strchr(s: value, c: '=') + 1; |
424 | if (!par_range_value) |
425 | goto par_range_error; |
426 | value = par_range_value; |
427 | len = __kmp_readstr_with_sentinel(dest: out_file, src: value, |
428 | KMP_PAR_RANGE_FILENAME_LEN - 1, sentinel: ','); |
429 | if (len == 0) { |
430 | goto par_range_error; |
431 | } |
432 | value = strchr(s: value, c: ','); |
433 | if (value != NULL) { |
434 | value++; |
435 | } |
436 | continue; |
437 | } |
438 | if ((!__kmp_strcasecmp_with_sentinel(a: "range" , b: value, sentinel: '=')) || |
439 | (!__kmp_strcasecmp_with_sentinel(a: "incl_range" , b: value, sentinel: '='))) { |
440 | par_range_value = strchr(s: value, c: '=') + 1; |
441 | if (!par_range_value) |
442 | goto par_range_error; |
443 | value = par_range_value; |
444 | if (KMP_SSCANF(s: value, format: "%d:%d" , out_lb, out_ub) != 2) { |
445 | goto par_range_error; |
446 | } |
447 | *out_range = +1; |
448 | value = strchr(s: value, c: ','); |
449 | if (value != NULL) { |
450 | value++; |
451 | } |
452 | continue; |
453 | } |
454 | if (!__kmp_strcasecmp_with_sentinel(a: "excl_range" , b: value, sentinel: '=')) { |
455 | par_range_value = strchr(s: value, c: '=') + 1; |
456 | if (!par_range_value) |
457 | goto par_range_error; |
458 | value = par_range_value; |
459 | if (KMP_SSCANF(s: value, format: "%d:%d" , out_lb, out_ub) != 2) { |
460 | goto par_range_error; |
461 | } |
462 | *out_range = -1; |
463 | value = strchr(s: value, c: ','); |
464 | if (value != NULL) { |
465 | value++; |
466 | } |
467 | continue; |
468 | } |
469 | par_range_error: |
470 | KMP_WARNING(ParRangeSyntax, name); |
471 | __kmp_par_range = 0; |
472 | break; |
473 | } |
474 | } // __kmp_stg_parse_par_range |
475 | #endif |
476 | |
477 | int __kmp_initial_threads_capacity(int req_nproc) { |
478 | int nth = 32; |
479 | |
480 | /* MIN( MAX( 32, 4 * $OMP_NUM_THREADS, 4 * omp_get_num_procs() ), |
481 | * __kmp_max_nth) */ |
482 | if (nth < (4 * req_nproc)) |
483 | nth = (4 * req_nproc); |
484 | if (nth < (4 * __kmp_xproc)) |
485 | nth = (4 * __kmp_xproc); |
486 | |
487 | // If hidden helper task is enabled, we initialize the thread capacity with |
488 | // extra __kmp_hidden_helper_threads_num. |
489 | if (__kmp_enable_hidden_helper) { |
490 | nth += __kmp_hidden_helper_threads_num; |
491 | } |
492 | |
493 | if (nth > __kmp_max_nth) |
494 | nth = __kmp_max_nth; |
495 | |
496 | return nth; |
497 | } |
498 | |
499 | int __kmp_default_tp_capacity(int req_nproc, int max_nth, |
500 | int all_threads_specified) { |
501 | int nth = 128; |
502 | |
503 | if (all_threads_specified) |
504 | return max_nth; |
505 | /* MIN( MAX (128, 4 * $OMP_NUM_THREADS, 4 * omp_get_num_procs() ), |
506 | * __kmp_max_nth ) */ |
507 | if (nth < (4 * req_nproc)) |
508 | nth = (4 * req_nproc); |
509 | if (nth < (4 * __kmp_xproc)) |
510 | nth = (4 * __kmp_xproc); |
511 | |
512 | if (nth > __kmp_max_nth) |
513 | nth = __kmp_max_nth; |
514 | |
515 | return nth; |
516 | } |
517 | |
518 | // ----------------------------------------------------------------------------- |
519 | // Helper print functions. |
520 | |
521 | static void __kmp_stg_print_bool(kmp_str_buf_t *buffer, char const *name, |
522 | int value) { |
523 | if (__kmp_env_format) { |
524 | KMP_STR_BUF_PRINT_BOOL; |
525 | } else { |
526 | __kmp_str_buf_print(buffer, format: " %s=%s\n" , name, value ? "true" : "false" ); |
527 | } |
528 | } // __kmp_stg_print_bool |
529 | |
530 | static void __kmp_stg_print_int(kmp_str_buf_t *buffer, char const *name, |
531 | int value) { |
532 | if (__kmp_env_format) { |
533 | KMP_STR_BUF_PRINT_INT; |
534 | } else { |
535 | __kmp_str_buf_print(buffer, format: " %s=%d\n" , name, value); |
536 | } |
537 | } // __kmp_stg_print_int |
538 | |
539 | static void __kmp_stg_print_uint64(kmp_str_buf_t *buffer, char const *name, |
540 | kmp_uint64 value) { |
541 | if (__kmp_env_format) { |
542 | KMP_STR_BUF_PRINT_UINT64; |
543 | } else { |
544 | __kmp_str_buf_print(buffer, format: " %s=%" KMP_UINT64_SPEC "\n" , name, value); |
545 | } |
546 | } // __kmp_stg_print_uint64 |
547 | |
548 | static void __kmp_stg_print_str(kmp_str_buf_t *buffer, char const *name, |
549 | char const *value) { |
550 | if (__kmp_env_format) { |
551 | KMP_STR_BUF_PRINT_STR; |
552 | } else { |
553 | __kmp_str_buf_print(buffer, format: " %s=%s\n" , name, value); |
554 | } |
555 | } // __kmp_stg_print_str |
556 | |
557 | static void __kmp_stg_print_size(kmp_str_buf_t *buffer, char const *name, |
558 | size_t value) { |
559 | if (__kmp_env_format) { |
560 | KMP_STR_BUF_PRINT_NAME_EX(name); |
561 | __kmp_str_buf_print_size(buffer, size: value); |
562 | __kmp_str_buf_print(buffer, format: "'\n" ); |
563 | } else { |
564 | __kmp_str_buf_print(buffer, format: " %s=" , name); |
565 | __kmp_str_buf_print_size(buffer, size: value); |
566 | __kmp_str_buf_print(buffer, format: "\n" ); |
567 | return; |
568 | } |
569 | } // __kmp_stg_print_size |
570 | |
571 | // ============================================================================= |
572 | // Parse and print functions. |
573 | |
574 | // ----------------------------------------------------------------------------- |
575 | // KMP_DEVICE_THREAD_LIMIT, KMP_ALL_THREADS |
576 | |
577 | static void __kmp_stg_parse_device_thread_limit(char const *name, |
578 | char const *value, void *data) { |
579 | kmp_setting_t **rivals = (kmp_setting_t **)data; |
580 | int rc; |
581 | if (strcmp(s1: name, s2: "KMP_ALL_THREADS" ) == 0) { |
582 | KMP_INFORM(EnvVarDeprecated, name, "KMP_DEVICE_THREAD_LIMIT" ); |
583 | } |
584 | rc = __kmp_stg_check_rivals(name, value, rivals); |
585 | if (rc) { |
586 | return; |
587 | } |
588 | if (!__kmp_strcasecmp_with_sentinel(a: "all" , b: value, sentinel: 0)) { |
589 | __kmp_max_nth = __kmp_xproc; |
590 | __kmp_allThreadsSpecified = 1; |
591 | } else { |
592 | __kmp_stg_parse_int(name, value, min: 1, max: __kmp_sys_max_nth, out: &__kmp_max_nth); |
593 | __kmp_allThreadsSpecified = 0; |
594 | } |
595 | K_DIAG(1, ("__kmp_max_nth == %d\n" , __kmp_max_nth)); |
596 | |
597 | } // __kmp_stg_parse_device_thread_limit |
598 | |
599 | static void __kmp_stg_print_device_thread_limit(kmp_str_buf_t *buffer, |
600 | char const *name, void *data) { |
601 | __kmp_stg_print_int(buffer, name, value: __kmp_max_nth); |
602 | } // __kmp_stg_print_device_thread_limit |
603 | |
604 | // ----------------------------------------------------------------------------- |
605 | // OMP_THREAD_LIMIT |
606 | static void __kmp_stg_parse_thread_limit(char const *name, char const *value, |
607 | void *data) { |
608 | __kmp_stg_parse_int(name, value, min: 1, max: __kmp_sys_max_nth, out: &__kmp_cg_max_nth); |
609 | K_DIAG(1, ("__kmp_cg_max_nth == %d\n" , __kmp_cg_max_nth)); |
610 | |
611 | } // __kmp_stg_parse_thread_limit |
612 | |
613 | static void __kmp_stg_print_thread_limit(kmp_str_buf_t *buffer, |
614 | char const *name, void *data) { |
615 | __kmp_stg_print_int(buffer, name, value: __kmp_cg_max_nth); |
616 | } // __kmp_stg_print_thread_limit |
617 | |
618 | // ----------------------------------------------------------------------------- |
619 | // OMP_NUM_TEAMS |
620 | static void __kmp_stg_parse_nteams(char const *name, char const *value, |
621 | void *data) { |
622 | __kmp_stg_parse_int(name, value, min: 1, max: __kmp_sys_max_nth, out: &__kmp_nteams); |
623 | K_DIAG(1, ("__kmp_nteams == %d\n" , __kmp_nteams)); |
624 | } // __kmp_stg_parse_nteams |
625 | |
626 | static void __kmp_stg_print_nteams(kmp_str_buf_t *buffer, char const *name, |
627 | void *data) { |
628 | __kmp_stg_print_int(buffer, name, value: __kmp_nteams); |
629 | } // __kmp_stg_print_nteams |
630 | |
631 | // ----------------------------------------------------------------------------- |
632 | // OMP_TEAMS_THREAD_LIMIT |
633 | static void __kmp_stg_parse_teams_th_limit(char const *name, char const *value, |
634 | void *data) { |
635 | __kmp_stg_parse_int(name, value, min: 1, max: __kmp_sys_max_nth, |
636 | out: &__kmp_teams_thread_limit); |
637 | K_DIAG(1, ("__kmp_teams_thread_limit == %d\n" , __kmp_teams_thread_limit)); |
638 | } // __kmp_stg_parse_teams_th_limit |
639 | |
640 | static void __kmp_stg_print_teams_th_limit(kmp_str_buf_t *buffer, |
641 | char const *name, void *data) { |
642 | __kmp_stg_print_int(buffer, name, value: __kmp_teams_thread_limit); |
643 | } // __kmp_stg_print_teams_th_limit |
644 | |
645 | // ----------------------------------------------------------------------------- |
646 | // KMP_TEAMS_THREAD_LIMIT |
647 | static void __kmp_stg_parse_teams_thread_limit(char const *name, |
648 | char const *value, void *data) { |
649 | __kmp_stg_parse_int(name, value, min: 1, max: __kmp_sys_max_nth, out: &__kmp_teams_max_nth); |
650 | } // __kmp_stg_teams_thread_limit |
651 | |
652 | static void __kmp_stg_print_teams_thread_limit(kmp_str_buf_t *buffer, |
653 | char const *name, void *data) { |
654 | __kmp_stg_print_int(buffer, name, value: __kmp_teams_max_nth); |
655 | } // __kmp_stg_print_teams_thread_limit |
656 | |
657 | // ----------------------------------------------------------------------------- |
658 | // KMP_USE_YIELD |
659 | static void __kmp_stg_parse_use_yield(char const *name, char const *value, |
660 | void *data) { |
661 | __kmp_stg_parse_int(name, value, min: 0, max: 2, out: &__kmp_use_yield); |
662 | __kmp_use_yield_exp_set = 1; |
663 | } // __kmp_stg_parse_use_yield |
664 | |
665 | static void __kmp_stg_print_use_yield(kmp_str_buf_t *buffer, char const *name, |
666 | void *data) { |
667 | __kmp_stg_print_int(buffer, name, value: __kmp_use_yield); |
668 | } // __kmp_stg_print_use_yield |
669 | |
670 | // ----------------------------------------------------------------------------- |
671 | // KMP_BLOCKTIME |
672 | |
673 | static void __kmp_stg_parse_blocktime(char const *name, char const *value, |
674 | void *data) { |
675 | const char *buf = value; |
676 | const char *next; |
677 | const int ms_mult = 1000; |
678 | int multiplier = 1; |
679 | int num; |
680 | |
681 | // Read integer blocktime value |
682 | SKIP_WS(buf); |
683 | if ((*buf >= '0') && (*buf <= '9')) { |
684 | next = buf; |
685 | SKIP_DIGITS(next); |
686 | num = __kmp_basic_str_to_int(str: buf); |
687 | KMP_ASSERT(num >= 0); |
688 | buf = next; |
689 | SKIP_WS(buf); |
690 | } else { |
691 | num = -1; |
692 | } |
693 | |
694 | // Read units: note that __kmp_dflt_blocktime units is now us |
695 | next = buf; |
696 | if (*buf == '\0' || __kmp_match_str(token: "ms" , buf, end: &next)) { |
697 | // units are in ms; convert |
698 | __kmp_dflt_blocktime = ms_mult * num; |
699 | __kmp_blocktime_units = 'm'; |
700 | multiplier = ms_mult; |
701 | } else if (__kmp_match_str(token: "us" , buf, end: &next)) { |
702 | // units are in us |
703 | __kmp_dflt_blocktime = num; |
704 | __kmp_blocktime_units = 'u'; |
705 | } else if (__kmp_match_str(token: "infinite" , buf, end: &next) || |
706 | __kmp_match_str(token: "infinity" , buf, end: &next)) { |
707 | // units are in ms |
708 | __kmp_dflt_blocktime = KMP_MAX_BLOCKTIME; |
709 | __kmp_blocktime_units = 'm'; |
710 | multiplier = ms_mult; |
711 | } else { |
712 | KMP_WARNING(StgInvalidValue, name, value); |
713 | // default units are in ms |
714 | __kmp_dflt_blocktime = ms_mult * num; |
715 | __kmp_blocktime_units = 'm'; |
716 | multiplier = ms_mult; |
717 | } |
718 | |
719 | if (num < 0 && __kmp_dflt_blocktime < 0) { // num out of range |
720 | __kmp_dflt_blocktime = KMP_DEFAULT_BLOCKTIME; // now in us |
721 | __kmp_msg(kmp_ms_warning, KMP_MSG(InvalidValue, name, value), |
722 | __kmp_msg_null); |
723 | // Inform in appropriate units |
724 | KMP_INFORM(Using_int_Value, name, __kmp_dflt_blocktime / multiplier); |
725 | __kmp_env_blocktime = FALSE; // Revert to default as if var not set. |
726 | } else if (num > 0 && __kmp_dflt_blocktime < 0) { // overflow |
727 | __kmp_dflt_blocktime = KMP_MAX_BLOCKTIME; |
728 | __kmp_msg(kmp_ms_warning, KMP_MSG(LargeValue, name, value), __kmp_msg_null); |
729 | KMP_INFORM(MaxValueUsing, name, __kmp_dflt_blocktime / multiplier); |
730 | __kmp_env_blocktime = TRUE; // KMP_BLOCKTIME was specified. |
731 | } else { |
732 | if (__kmp_dflt_blocktime < KMP_MIN_BLOCKTIME) { |
733 | __kmp_dflt_blocktime = KMP_MIN_BLOCKTIME; |
734 | __kmp_msg(kmp_ms_warning, KMP_MSG(SmallValue, name, value), |
735 | __kmp_msg_null); |
736 | KMP_INFORM(MinValueUsing, name, __kmp_dflt_blocktime / multiplier); |
737 | } else if (__kmp_dflt_blocktime > KMP_MAX_BLOCKTIME) { |
738 | __kmp_dflt_blocktime = KMP_MAX_BLOCKTIME; |
739 | __kmp_msg(kmp_ms_warning, KMP_MSG(LargeValue, name, value), |
740 | __kmp_msg_null); |
741 | KMP_INFORM(MaxValueUsing, name, __kmp_dflt_blocktime / multiplier); |
742 | } |
743 | __kmp_env_blocktime = TRUE; // KMP_BLOCKTIME was specified. |
744 | } |
745 | #if KMP_USE_MONITOR |
746 | // calculate number of monitor thread wakeup intervals corresponding to |
747 | // blocktime. |
748 | __kmp_monitor_wakeups = |
749 | KMP_WAKEUPS_FROM_BLOCKTIME(__kmp_dflt_blocktime, __kmp_monitor_wakeups); |
750 | __kmp_bt_intervals = |
751 | KMP_INTERVALS_FROM_BLOCKTIME(__kmp_dflt_blocktime, __kmp_monitor_wakeups); |
752 | #endif |
753 | K_DIAG(1, ("__kmp_env_blocktime == %d\n" , __kmp_env_blocktime)); |
754 | if (__kmp_env_blocktime) { |
755 | K_DIAG(1, ("__kmp_dflt_blocktime == %d\n" , __kmp_dflt_blocktime)); |
756 | } |
757 | } // __kmp_stg_parse_blocktime |
758 | |
759 | static void __kmp_stg_print_blocktime(kmp_str_buf_t *buffer, char const *name, |
760 | void *data) { |
761 | int num = __kmp_dflt_blocktime; |
762 | if (__kmp_blocktime_units == 'm') { |
763 | num = num / 1000; |
764 | } |
765 | if (__kmp_env_format) { |
766 | KMP_STR_BUF_PRINT_NAME_EX(name); |
767 | } else { |
768 | __kmp_str_buf_print(buffer, format: " %s=" , name); |
769 | } |
770 | __kmp_str_buf_print(buffer, format: "%d" , num); |
771 | __kmp_str_buf_print(buffer, format: "%cs\n" , __kmp_blocktime_units); |
772 | } // __kmp_stg_print_blocktime |
773 | |
774 | // ----------------------------------------------------------------------------- |
775 | // KMP_DUPLICATE_LIB_OK |
776 | |
777 | static void __kmp_stg_parse_duplicate_lib_ok(char const *name, |
778 | char const *value, void *data) { |
779 | /* actually this variable is not supported, put here for compatibility with |
780 | earlier builds and for static/dynamic combination */ |
781 | __kmp_stg_parse_bool(name, value, out: &__kmp_duplicate_library_ok); |
782 | } // __kmp_stg_parse_duplicate_lib_ok |
783 | |
784 | static void __kmp_stg_print_duplicate_lib_ok(kmp_str_buf_t *buffer, |
785 | char const *name, void *data) { |
786 | __kmp_stg_print_bool(buffer, name, value: __kmp_duplicate_library_ok); |
787 | } // __kmp_stg_print_duplicate_lib_ok |
788 | |
789 | // ----------------------------------------------------------------------------- |
790 | // KMP_INHERIT_FP_CONTROL |
791 | |
792 | #if KMP_ARCH_X86 || KMP_ARCH_X86_64 |
793 | |
794 | static void __kmp_stg_parse_inherit_fp_control(char const *name, |
795 | char const *value, void *data) { |
796 | __kmp_stg_parse_bool(name, value, out: &__kmp_inherit_fp_control); |
797 | } // __kmp_stg_parse_inherit_fp_control |
798 | |
799 | static void __kmp_stg_print_inherit_fp_control(kmp_str_buf_t *buffer, |
800 | char const *name, void *data) { |
801 | #if KMP_DEBUG |
802 | __kmp_stg_print_bool(buffer, name, value: __kmp_inherit_fp_control); |
803 | #endif /* KMP_DEBUG */ |
804 | } // __kmp_stg_print_inherit_fp_control |
805 | |
806 | #endif /* KMP_ARCH_X86 || KMP_ARCH_X86_64 */ |
807 | |
808 | // Used for OMP_WAIT_POLICY |
809 | static char const *blocktime_str = NULL; |
810 | |
811 | // ----------------------------------------------------------------------------- |
812 | // KMP_LIBRARY, OMP_WAIT_POLICY |
813 | |
814 | static void __kmp_stg_parse_wait_policy(char const *name, char const *value, |
815 | void *data) { |
816 | |
817 | kmp_stg_wp_data_t *wait = (kmp_stg_wp_data_t *)data; |
818 | int rc; |
819 | |
820 | rc = __kmp_stg_check_rivals(name, value, rivals: wait->rivals); |
821 | if (rc) { |
822 | return; |
823 | } |
824 | |
825 | if (wait->omp) { |
826 | if (__kmp_str_match(target: "ACTIVE" , len: 1, data: value)) { |
827 | __kmp_library = library_turnaround; |
828 | if (blocktime_str == NULL) { |
829 | // KMP_BLOCKTIME not specified, so set default to "infinite". |
830 | __kmp_dflt_blocktime = KMP_MAX_BLOCKTIME; |
831 | } |
832 | } else if (__kmp_str_match(target: "PASSIVE" , len: 1, data: value)) { |
833 | __kmp_library = library_throughput; |
834 | __kmp_wpolicy_passive = true; /* allow sleep while active tasking */ |
835 | if (blocktime_str == NULL) { |
836 | // KMP_BLOCKTIME not specified, so set default to 0. |
837 | __kmp_dflt_blocktime = 0; |
838 | } |
839 | } else { |
840 | KMP_WARNING(StgInvalidValue, name, value); |
841 | } |
842 | } else { |
843 | if (__kmp_str_match(target: "serial" , len: 1, data: value)) { /* S */ |
844 | __kmp_library = library_serial; |
845 | } else if (__kmp_str_match(target: "throughput" , len: 2, data: value)) { /* TH */ |
846 | __kmp_library = library_throughput; |
847 | if (blocktime_str == NULL) { |
848 | // KMP_BLOCKTIME not specified, so set default to 0. |
849 | __kmp_dflt_blocktime = 0; |
850 | } |
851 | } else if (__kmp_str_match(target: "turnaround" , len: 2, data: value)) { /* TU */ |
852 | __kmp_library = library_turnaround; |
853 | } else if (__kmp_str_match(target: "dedicated" , len: 1, data: value)) { /* D */ |
854 | __kmp_library = library_turnaround; |
855 | } else if (__kmp_str_match(target: "multiuser" , len: 1, data: value)) { /* M */ |
856 | __kmp_library = library_throughput; |
857 | if (blocktime_str == NULL) { |
858 | // KMP_BLOCKTIME not specified, so set default to 0. |
859 | __kmp_dflt_blocktime = 0; |
860 | } |
861 | } else { |
862 | KMP_WARNING(StgInvalidValue, name, value); |
863 | } |
864 | } |
865 | } // __kmp_stg_parse_wait_policy |
866 | |
867 | static void __kmp_stg_print_wait_policy(kmp_str_buf_t *buffer, char const *name, |
868 | void *data) { |
869 | |
870 | kmp_stg_wp_data_t *wait = (kmp_stg_wp_data_t *)data; |
871 | char const *value = NULL; |
872 | |
873 | if (wait->omp) { |
874 | switch (__kmp_library) { |
875 | case library_turnaround: { |
876 | value = "ACTIVE" ; |
877 | } break; |
878 | case library_throughput: { |
879 | value = "PASSIVE" ; |
880 | } break; |
881 | case library_none: |
882 | case library_serial: { |
883 | value = NULL; |
884 | } break; |
885 | } |
886 | } else { |
887 | switch (__kmp_library) { |
888 | case library_serial: { |
889 | value = "serial" ; |
890 | } break; |
891 | case library_turnaround: { |
892 | value = "turnaround" ; |
893 | } break; |
894 | case library_throughput: { |
895 | value = "throughput" ; |
896 | } break; |
897 | case library_none: { |
898 | value = NULL; |
899 | } break; |
900 | } |
901 | } |
902 | if (value != NULL) { |
903 | __kmp_stg_print_str(buffer, name, value); |
904 | } |
905 | |
906 | } // __kmp_stg_print_wait_policy |
907 | |
908 | #if KMP_USE_MONITOR |
909 | // ----------------------------------------------------------------------------- |
910 | // KMP_MONITOR_STACKSIZE |
911 | |
912 | static void __kmp_stg_parse_monitor_stacksize(char const *name, |
913 | char const *value, void *data) { |
914 | __kmp_stg_parse_size(name, value, __kmp_sys_min_stksize, KMP_MAX_STKSIZE, |
915 | NULL, &__kmp_monitor_stksize, 1); |
916 | } // __kmp_stg_parse_monitor_stacksize |
917 | |
918 | static void __kmp_stg_print_monitor_stacksize(kmp_str_buf_t *buffer, |
919 | char const *name, void *data) { |
920 | if (__kmp_env_format) { |
921 | if (__kmp_monitor_stksize > 0) |
922 | KMP_STR_BUF_PRINT_NAME_EX(name); |
923 | else |
924 | KMP_STR_BUF_PRINT_NAME; |
925 | } else { |
926 | __kmp_str_buf_print(buffer, " %s" , name); |
927 | } |
928 | if (__kmp_monitor_stksize > 0) { |
929 | __kmp_str_buf_print_size(buffer, __kmp_monitor_stksize); |
930 | } else { |
931 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
932 | } |
933 | if (__kmp_env_format && __kmp_monitor_stksize) { |
934 | __kmp_str_buf_print(buffer, "'\n" ); |
935 | } |
936 | } // __kmp_stg_print_monitor_stacksize |
937 | #endif // KMP_USE_MONITOR |
938 | |
939 | // ----------------------------------------------------------------------------- |
940 | // KMP_SETTINGS |
941 | |
942 | static void __kmp_stg_parse_settings(char const *name, char const *value, |
943 | void *data) { |
944 | __kmp_stg_parse_bool(name, value, out: &__kmp_settings); |
945 | } // __kmp_stg_parse_settings |
946 | |
947 | static void __kmp_stg_print_settings(kmp_str_buf_t *buffer, char const *name, |
948 | void *data) { |
949 | __kmp_stg_print_bool(buffer, name, value: __kmp_settings); |
950 | } // __kmp_stg_print_settings |
951 | |
952 | // ----------------------------------------------------------------------------- |
953 | // KMP_STACKPAD |
954 | |
955 | static void __kmp_stg_parse_stackpad(char const *name, char const *value, |
956 | void *data) { |
957 | __kmp_stg_parse_int(name, // Env var name |
958 | value, // Env var value |
959 | KMP_MIN_STKPADDING, // Min value |
960 | KMP_MAX_STKPADDING, // Max value |
961 | out: &__kmp_stkpadding // Var to initialize |
962 | ); |
963 | } // __kmp_stg_parse_stackpad |
964 | |
965 | static void __kmp_stg_print_stackpad(kmp_str_buf_t *buffer, char const *name, |
966 | void *data) { |
967 | __kmp_stg_print_int(buffer, name, value: __kmp_stkpadding); |
968 | } // __kmp_stg_print_stackpad |
969 | |
970 | // ----------------------------------------------------------------------------- |
971 | // KMP_STACKOFFSET |
972 | |
973 | static void __kmp_stg_parse_stackoffset(char const *name, char const *value, |
974 | void *data) { |
975 | __kmp_stg_parse_size(name, // Env var name |
976 | value, // Env var value |
977 | KMP_MIN_STKOFFSET, // Min value |
978 | KMP_MAX_STKOFFSET, // Max value |
979 | NULL, // |
980 | out: &__kmp_stkoffset, // Var to initialize |
981 | factor: 1); |
982 | } // __kmp_stg_parse_stackoffset |
983 | |
984 | static void __kmp_stg_print_stackoffset(kmp_str_buf_t *buffer, char const *name, |
985 | void *data) { |
986 | __kmp_stg_print_size(buffer, name, value: __kmp_stkoffset); |
987 | } // __kmp_stg_print_stackoffset |
988 | |
989 | // ----------------------------------------------------------------------------- |
990 | // KMP_STACKSIZE, OMP_STACKSIZE, GOMP_STACKSIZE |
991 | |
992 | static void __kmp_stg_parse_stacksize(char const *name, char const *value, |
993 | void *data) { |
994 | |
995 | kmp_stg_ss_data_t *stacksize = (kmp_stg_ss_data_t *)data; |
996 | int rc; |
997 | |
998 | rc = __kmp_stg_check_rivals(name, value, rivals: stacksize->rivals); |
999 | if (rc) { |
1000 | return; |
1001 | } |
1002 | __kmp_stg_parse_size(name, // Env var name |
1003 | value, // Env var value |
1004 | size_min: __kmp_sys_min_stksize, // Min value |
1005 | KMP_MAX_STKSIZE, // Max value |
1006 | is_specified: &__kmp_env_stksize, // |
1007 | out: &__kmp_stksize, // Var to initialize |
1008 | factor: stacksize->factor); |
1009 | |
1010 | } // __kmp_stg_parse_stacksize |
1011 | |
1012 | // This function is called for printing both KMP_STACKSIZE (factor is 1) and |
1013 | // OMP_STACKSIZE (factor is 1024). Currently it is not possible to print |
1014 | // OMP_STACKSIZE value in bytes. We can consider adding this possibility by a |
1015 | // customer request in future. |
1016 | static void __kmp_stg_print_stacksize(kmp_str_buf_t *buffer, char const *name, |
1017 | void *data) { |
1018 | kmp_stg_ss_data_t *stacksize = (kmp_stg_ss_data_t *)data; |
1019 | if (__kmp_env_format) { |
1020 | KMP_STR_BUF_PRINT_NAME_EX(name); |
1021 | __kmp_str_buf_print_size(buffer, size: (__kmp_stksize % 1024) |
1022 | ? __kmp_stksize / stacksize->factor |
1023 | : __kmp_stksize); |
1024 | __kmp_str_buf_print(buffer, format: "'\n" ); |
1025 | } else { |
1026 | __kmp_str_buf_print(buffer, format: " %s=" , name); |
1027 | __kmp_str_buf_print_size(buffer, size: (__kmp_stksize % 1024) |
1028 | ? __kmp_stksize / stacksize->factor |
1029 | : __kmp_stksize); |
1030 | __kmp_str_buf_print(buffer, format: "\n" ); |
1031 | } |
1032 | } // __kmp_stg_print_stacksize |
1033 | |
1034 | // ----------------------------------------------------------------------------- |
1035 | // KMP_VERSION |
1036 | |
1037 | static void __kmp_stg_parse_version(char const *name, char const *value, |
1038 | void *data) { |
1039 | __kmp_stg_parse_bool(name, value, out: &__kmp_version); |
1040 | } // __kmp_stg_parse_version |
1041 | |
1042 | static void __kmp_stg_print_version(kmp_str_buf_t *buffer, char const *name, |
1043 | void *data) { |
1044 | __kmp_stg_print_bool(buffer, name, value: __kmp_version); |
1045 | } // __kmp_stg_print_version |
1046 | |
1047 | // ----------------------------------------------------------------------------- |
1048 | // KMP_WARNINGS |
1049 | |
1050 | static void __kmp_stg_parse_warnings(char const *name, char const *value, |
1051 | void *data) { |
1052 | __kmp_stg_parse_bool(name, value, out: &__kmp_generate_warnings); |
1053 | if (__kmp_generate_warnings != kmp_warnings_off) { |
1054 | // AC: only 0/1 values documented, so reset to explicit to distinguish from |
1055 | // default setting |
1056 | __kmp_generate_warnings = kmp_warnings_explicit; |
1057 | } |
1058 | } // __kmp_stg_parse_warnings |
1059 | |
1060 | static void __kmp_stg_print_warnings(kmp_str_buf_t *buffer, char const *name, |
1061 | void *data) { |
1062 | // AC: TODO: change to print_int? (needs documentation change) |
1063 | __kmp_stg_print_bool(buffer, name, value: __kmp_generate_warnings); |
1064 | } // __kmp_stg_print_warnings |
1065 | |
1066 | // ----------------------------------------------------------------------------- |
1067 | // KMP_NESTING_MODE |
1068 | |
1069 | static void __kmp_stg_parse_nesting_mode(char const *name, char const *value, |
1070 | void *data) { |
1071 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX, out: &__kmp_nesting_mode); |
1072 | #if KMP_AFFINITY_SUPPORTED && KMP_USE_HWLOC |
1073 | if (__kmp_nesting_mode > 0) |
1074 | __kmp_affinity_top_method = affinity_top_method_hwloc; |
1075 | #endif |
1076 | } // __kmp_stg_parse_nesting_mode |
1077 | |
1078 | static void __kmp_stg_print_nesting_mode(kmp_str_buf_t *buffer, |
1079 | char const *name, void *data) { |
1080 | if (__kmp_env_format) { |
1081 | KMP_STR_BUF_PRINT_NAME; |
1082 | } else { |
1083 | __kmp_str_buf_print(buffer, format: " %s" , name); |
1084 | } |
1085 | __kmp_str_buf_print(buffer, format: "=%d\n" , __kmp_nesting_mode); |
1086 | } // __kmp_stg_print_nesting_mode |
1087 | |
1088 | // ----------------------------------------------------------------------------- |
1089 | // OMP_NESTED, OMP_NUM_THREADS |
1090 | |
1091 | static void __kmp_stg_parse_nested(char const *name, char const *value, |
1092 | void *data) { |
1093 | int nested; |
1094 | KMP_INFORM(EnvVarDeprecated, name, "OMP_MAX_ACTIVE_LEVELS" ); |
1095 | __kmp_stg_parse_bool(name, value, out: &nested); |
1096 | if (nested) { |
1097 | if (!__kmp_dflt_max_active_levels_set) |
1098 | __kmp_dflt_max_active_levels = KMP_MAX_ACTIVE_LEVELS_LIMIT; |
1099 | } else { // nesting explicitly turned off |
1100 | __kmp_dflt_max_active_levels = 1; |
1101 | __kmp_dflt_max_active_levels_set = true; |
1102 | } |
1103 | } // __kmp_stg_parse_nested |
1104 | |
1105 | static void __kmp_stg_print_nested(kmp_str_buf_t *buffer, char const *name, |
1106 | void *data) { |
1107 | if (__kmp_env_format) { |
1108 | KMP_STR_BUF_PRINT_NAME; |
1109 | } else { |
1110 | __kmp_str_buf_print(buffer, format: " %s" , name); |
1111 | } |
1112 | __kmp_str_buf_print(buffer, format: ": deprecated; max-active-levels-var=%d\n" , |
1113 | __kmp_dflt_max_active_levels); |
1114 | } // __kmp_stg_print_nested |
1115 | |
1116 | static void __kmp_parse_nested_num_threads(const char *var, const char *env, |
1117 | kmp_nested_nthreads_t *nth_array) { |
1118 | const char *next = env; |
1119 | const char *scan = next; |
1120 | |
1121 | int total = 0; // Count elements that were set. It'll be used as an array size |
1122 | int prev_comma = FALSE; // For correct processing sequential commas |
1123 | |
1124 | // Count the number of values in the env. var string |
1125 | for (;;) { |
1126 | SKIP_WS(next); |
1127 | |
1128 | if (*next == '\0') { |
1129 | break; |
1130 | } |
1131 | // Next character is not an integer or not a comma => end of list |
1132 | if (((*next < '0') || (*next > '9')) && (*next != ',')) { |
1133 | KMP_WARNING(NthSyntaxError, var, env); |
1134 | return; |
1135 | } |
1136 | // The next character is ',' |
1137 | if (*next == ',') { |
1138 | // ',' is the first character |
1139 | if (total == 0 || prev_comma) { |
1140 | total++; |
1141 | } |
1142 | prev_comma = TRUE; |
1143 | next++; // skip ',' |
1144 | SKIP_WS(next); |
1145 | } |
1146 | // Next character is a digit |
1147 | if (*next >= '0' && *next <= '9') { |
1148 | prev_comma = FALSE; |
1149 | SKIP_DIGITS(next); |
1150 | total++; |
1151 | const char *tmp = next; |
1152 | SKIP_WS(tmp); |
1153 | if ((*next == ' ' || *next == '\t') && (*tmp >= '0' && *tmp <= '9')) { |
1154 | KMP_WARNING(NthSpacesNotAllowed, var, env); |
1155 | return; |
1156 | } |
1157 | } |
1158 | } |
1159 | if (!__kmp_dflt_max_active_levels_set && total > 1) |
1160 | __kmp_dflt_max_active_levels = KMP_MAX_ACTIVE_LEVELS_LIMIT; |
1161 | KMP_DEBUG_ASSERT(total > 0); |
1162 | if (total <= 0) { |
1163 | KMP_WARNING(NthSyntaxError, var, env); |
1164 | return; |
1165 | } |
1166 | |
1167 | // Check if the nested nthreads array exists |
1168 | if (!nth_array->nth) { |
1169 | // Allocate an array of double size |
1170 | nth_array->nth = (int *)KMP_INTERNAL_MALLOC(sizeof(int) * total * 2); |
1171 | if (nth_array->nth == NULL) { |
1172 | KMP_FATAL(MemoryAllocFailed); |
1173 | } |
1174 | nth_array->size = total * 2; |
1175 | } else { |
1176 | if (nth_array->size < total) { |
1177 | // Increase the array size |
1178 | do { |
1179 | nth_array->size *= 2; |
1180 | } while (nth_array->size < total); |
1181 | |
1182 | nth_array->nth = (int *)KMP_INTERNAL_REALLOC( |
1183 | nth_array->nth, sizeof(int) * nth_array->size); |
1184 | if (nth_array->nth == NULL) { |
1185 | KMP_FATAL(MemoryAllocFailed); |
1186 | } |
1187 | } |
1188 | } |
1189 | nth_array->used = total; |
1190 | int i = 0; |
1191 | |
1192 | prev_comma = FALSE; |
1193 | total = 0; |
1194 | // Save values in the array |
1195 | for (;;) { |
1196 | SKIP_WS(scan); |
1197 | if (*scan == '\0') { |
1198 | break; |
1199 | } |
1200 | // The next character is ',' |
1201 | if (*scan == ',') { |
1202 | // ',' in the beginning of the list |
1203 | if (total == 0) { |
1204 | // The value is supposed to be equal to __kmp_avail_proc but it is |
1205 | // unknown at the moment. |
1206 | // So let's put a placeholder (#threads = 0) to correct it later. |
1207 | nth_array->nth[i++] = 0; |
1208 | total++; |
1209 | } else if (prev_comma) { |
1210 | // Num threads is inherited from the previous level |
1211 | nth_array->nth[i] = nth_array->nth[i - 1]; |
1212 | i++; |
1213 | total++; |
1214 | } |
1215 | prev_comma = TRUE; |
1216 | scan++; // skip ',' |
1217 | SKIP_WS(scan); |
1218 | } |
1219 | // Next character is a digit |
1220 | if (*scan >= '0' && *scan <= '9') { |
1221 | int num; |
1222 | const char *buf = scan; |
1223 | char const *msg = NULL; |
1224 | prev_comma = FALSE; |
1225 | SKIP_DIGITS(scan); |
1226 | total++; |
1227 | |
1228 | num = __kmp_str_to_int(str: buf, sentinel: *scan); |
1229 | if (num < KMP_MIN_NTH) { |
1230 | msg = KMP_I18N_STR(ValueTooSmall); |
1231 | num = KMP_MIN_NTH; |
1232 | } else if (num > __kmp_sys_max_nth) { |
1233 | msg = KMP_I18N_STR(ValueTooLarge); |
1234 | num = __kmp_sys_max_nth; |
1235 | } |
1236 | if (msg != NULL) { |
1237 | // Message is not empty. Print warning. |
1238 | KMP_WARNING(ParseSizeIntWarn, var, env, msg); |
1239 | KMP_INFORM(Using_int_Value, var, num); |
1240 | } |
1241 | nth_array->nth[i++] = num; |
1242 | } |
1243 | } |
1244 | } |
1245 | |
1246 | static void __kmp_stg_parse_num_threads(char const *name, char const *value, |
1247 | void *data) { |
1248 | // TODO: Remove this option. OMP_NUM_THREADS is a list of positive integers! |
1249 | if (!__kmp_strcasecmp_with_sentinel(a: "all" , b: value, sentinel: 0)) { |
1250 | // The array of 1 element |
1251 | __kmp_nested_nth.nth = (int *)KMP_INTERNAL_MALLOC(sizeof(int)); |
1252 | __kmp_nested_nth.size = __kmp_nested_nth.used = 1; |
1253 | __kmp_nested_nth.nth[0] = __kmp_dflt_team_nth = __kmp_dflt_team_nth_ub = |
1254 | __kmp_xproc; |
1255 | } else { |
1256 | __kmp_parse_nested_num_threads(var: name, env: value, nth_array: &__kmp_nested_nth); |
1257 | if (__kmp_nested_nth.nth) { |
1258 | __kmp_dflt_team_nth = __kmp_nested_nth.nth[0]; |
1259 | if (__kmp_dflt_team_nth_ub < __kmp_dflt_team_nth) { |
1260 | __kmp_dflt_team_nth_ub = __kmp_dflt_team_nth; |
1261 | } |
1262 | } |
1263 | } |
1264 | K_DIAG(1, ("__kmp_dflt_team_nth == %d\n" , __kmp_dflt_team_nth)); |
1265 | } // __kmp_stg_parse_num_threads |
1266 | |
1267 | #if OMPX_TASKGRAPH |
1268 | static void __kmp_stg_parse_max_tdgs(char const *name, char const *value, |
1269 | void *data) { |
1270 | __kmp_stg_parse_int(name, value, 0, INT_MAX, &__kmp_max_tdgs); |
1271 | } // __kmp_stg_parse_max_tdgs |
1272 | |
1273 | static void __kmp_std_print_max_tdgs(kmp_str_buf_t *buffer, char const *name, |
1274 | void *data) { |
1275 | __kmp_stg_print_int(buffer, name, __kmp_max_tdgs); |
1276 | } // __kmp_std_print_max_tdgs |
1277 | |
1278 | static void __kmp_stg_parse_tdg_dot(char const *name, char const *value, |
1279 | void *data) { |
1280 | __kmp_stg_parse_bool(name, value, &__kmp_tdg_dot); |
1281 | } // __kmp_stg_parse_tdg_dot |
1282 | |
1283 | static void __kmp_stg_print_tdg_dot(kmp_str_buf_t *buffer, char const *name, |
1284 | void *data) { |
1285 | __kmp_stg_print_bool(buffer, name, __kmp_tdg_dot); |
1286 | } // __kmp_stg_print_tdg_dot |
1287 | #endif |
1288 | |
1289 | static void __kmp_stg_parse_num_hidden_helper_threads(char const *name, |
1290 | char const *value, |
1291 | void *data) { |
1292 | __kmp_stg_parse_int(name, value, min: 0, max: 16, out: &__kmp_hidden_helper_threads_num); |
1293 | // If the number of hidden helper threads is zero, we disable hidden helper |
1294 | // task |
1295 | if (__kmp_hidden_helper_threads_num == 0) { |
1296 | __kmp_enable_hidden_helper = FALSE; |
1297 | } else { |
1298 | // Since the main thread of hidden helper team does not participate |
1299 | // in tasks execution let's increment the number of threads by one |
1300 | // so that requested number of threads do actual job. |
1301 | __kmp_hidden_helper_threads_num++; |
1302 | } |
1303 | } // __kmp_stg_parse_num_hidden_helper_threads |
1304 | |
1305 | static void __kmp_stg_print_num_hidden_helper_threads(kmp_str_buf_t *buffer, |
1306 | char const *name, |
1307 | void *data) { |
1308 | if (__kmp_hidden_helper_threads_num == 0) { |
1309 | __kmp_stg_print_int(buffer, name, value: __kmp_hidden_helper_threads_num); |
1310 | } else { |
1311 | KMP_DEBUG_ASSERT(__kmp_hidden_helper_threads_num > 1); |
1312 | // Let's exclude the main thread of hidden helper team and print |
1313 | // number of worker threads those do actual job. |
1314 | __kmp_stg_print_int(buffer, name, value: __kmp_hidden_helper_threads_num - 1); |
1315 | } |
1316 | } // __kmp_stg_print_num_hidden_helper_threads |
1317 | |
1318 | static void __kmp_stg_parse_use_hidden_helper(char const *name, |
1319 | char const *value, void *data) { |
1320 | __kmp_stg_parse_bool(name, value, out: &__kmp_enable_hidden_helper); |
1321 | #if !KMP_OS_LINUX |
1322 | __kmp_enable_hidden_helper = FALSE; |
1323 | K_DIAG(1, |
1324 | ("__kmp_stg_parse_use_hidden_helper: Disable hidden helper task on " |
1325 | "non-Linux platform although it is enabled by user explicitly.\n" )); |
1326 | #endif |
1327 | } // __kmp_stg_parse_use_hidden_helper |
1328 | |
1329 | static void __kmp_stg_print_use_hidden_helper(kmp_str_buf_t *buffer, |
1330 | char const *name, void *data) { |
1331 | __kmp_stg_print_bool(buffer, name, value: __kmp_enable_hidden_helper); |
1332 | } // __kmp_stg_print_use_hidden_helper |
1333 | |
1334 | static void __kmp_stg_print_num_threads(kmp_str_buf_t *buffer, char const *name, |
1335 | void *data) { |
1336 | if (__kmp_env_format) { |
1337 | KMP_STR_BUF_PRINT_NAME; |
1338 | } else { |
1339 | __kmp_str_buf_print(buffer, format: " %s" , name); |
1340 | } |
1341 | if (__kmp_nested_nth.used) { |
1342 | kmp_str_buf_t buf; |
1343 | __kmp_str_buf_init(&buf); |
1344 | for (int i = 0; i < __kmp_nested_nth.used; i++) { |
1345 | __kmp_str_buf_print(buffer: &buf, format: "%d" , __kmp_nested_nth.nth[i]); |
1346 | if (i < __kmp_nested_nth.used - 1) { |
1347 | __kmp_str_buf_print(buffer: &buf, format: "," ); |
1348 | } |
1349 | } |
1350 | __kmp_str_buf_print(buffer, format: "='%s'\n" , buf.str); |
1351 | __kmp_str_buf_free(buffer: &buf); |
1352 | } else { |
1353 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
1354 | } |
1355 | } // __kmp_stg_print_num_threads |
1356 | |
1357 | // ----------------------------------------------------------------------------- |
1358 | // OpenMP 3.0: KMP_TASKING, OMP_MAX_ACTIVE_LEVELS, |
1359 | |
1360 | static void __kmp_stg_parse_tasking(char const *name, char const *value, |
1361 | void *data) { |
1362 | __kmp_stg_parse_int(name, value, min: 0, max: (int)tskm_max, |
1363 | out: (int *)&__kmp_tasking_mode); |
1364 | } // __kmp_stg_parse_tasking |
1365 | |
1366 | static void __kmp_stg_print_tasking(kmp_str_buf_t *buffer, char const *name, |
1367 | void *data) { |
1368 | __kmp_stg_print_int(buffer, name, value: __kmp_tasking_mode); |
1369 | } // __kmp_stg_print_tasking |
1370 | |
1371 | static void __kmp_stg_parse_task_stealing(char const *name, char const *value, |
1372 | void *data) { |
1373 | __kmp_stg_parse_int(name, value, min: 0, max: 1, |
1374 | out: (int *)&__kmp_task_stealing_constraint); |
1375 | } // __kmp_stg_parse_task_stealing |
1376 | |
1377 | static void __kmp_stg_print_task_stealing(kmp_str_buf_t *buffer, |
1378 | char const *name, void *data) { |
1379 | __kmp_stg_print_int(buffer, name, value: __kmp_task_stealing_constraint); |
1380 | } // __kmp_stg_print_task_stealing |
1381 | |
1382 | static void __kmp_stg_parse_max_active_levels(char const *name, |
1383 | char const *value, void *data) { |
1384 | kmp_uint64 tmp_dflt = 0; |
1385 | char const *msg = NULL; |
1386 | if (!__kmp_dflt_max_active_levels_set) { |
1387 | // Don't overwrite __kmp_dflt_max_active_levels if we get an invalid setting |
1388 | __kmp_str_to_uint(str: value, out: &tmp_dflt, error: &msg); |
1389 | if (msg != NULL) { // invalid setting; print warning and ignore |
1390 | KMP_WARNING(ParseSizeIntWarn, name, value, msg); |
1391 | } else if (tmp_dflt > KMP_MAX_ACTIVE_LEVELS_LIMIT) { |
1392 | // invalid setting; print warning and ignore |
1393 | msg = KMP_I18N_STR(ValueTooLarge); |
1394 | KMP_WARNING(ParseSizeIntWarn, name, value, msg); |
1395 | } else { // valid setting |
1396 | __kmp_type_convert(src: tmp_dflt, dest: &(__kmp_dflt_max_active_levels)); |
1397 | __kmp_dflt_max_active_levels_set = true; |
1398 | } |
1399 | } |
1400 | } // __kmp_stg_parse_max_active_levels |
1401 | |
1402 | static void __kmp_stg_print_max_active_levels(kmp_str_buf_t *buffer, |
1403 | char const *name, void *data) { |
1404 | __kmp_stg_print_int(buffer, name, value: __kmp_dflt_max_active_levels); |
1405 | } // __kmp_stg_print_max_active_levels |
1406 | |
1407 | // ----------------------------------------------------------------------------- |
1408 | // OpenMP 4.0: OMP_DEFAULT_DEVICE |
1409 | static void __kmp_stg_parse_default_device(char const *name, char const *value, |
1410 | void *data) { |
1411 | __kmp_stg_parse_int(name, value, min: 0, KMP_MAX_DEFAULT_DEVICE_LIMIT, |
1412 | out: &__kmp_default_device); |
1413 | } // __kmp_stg_parse_default_device |
1414 | |
1415 | static void __kmp_stg_print_default_device(kmp_str_buf_t *buffer, |
1416 | char const *name, void *data) { |
1417 | __kmp_stg_print_int(buffer, name, value: __kmp_default_device); |
1418 | } // __kmp_stg_print_default_device |
1419 | |
1420 | // ----------------------------------------------------------------------------- |
1421 | // OpenMP 5.0: OMP_TARGET_OFFLOAD |
1422 | static void __kmp_stg_parse_target_offload(char const *name, char const *value, |
1423 | void *data) { |
1424 | kmp_trimmed_str_t value_str(value); |
1425 | const char *scan = value_str.get(); |
1426 | __kmp_target_offload = tgt_default; |
1427 | |
1428 | if (*scan == '\0') |
1429 | return; |
1430 | |
1431 | if (!__kmp_strcasecmp_with_sentinel(a: "mandatory" , b: scan, sentinel: 0)) { |
1432 | __kmp_target_offload = tgt_mandatory; |
1433 | } else if (!__kmp_strcasecmp_with_sentinel(a: "disabled" , b: scan, sentinel: 0)) { |
1434 | __kmp_target_offload = tgt_disabled; |
1435 | } else if (!__kmp_strcasecmp_with_sentinel(a: "default" , b: scan, sentinel: 0)) { |
1436 | __kmp_target_offload = tgt_default; |
1437 | } else { |
1438 | KMP_WARNING(SyntaxErrorUsing, name, "DEFAULT" ); |
1439 | } |
1440 | } // __kmp_stg_parse_target_offload |
1441 | |
1442 | static void __kmp_stg_print_target_offload(kmp_str_buf_t *buffer, |
1443 | char const *name, void *data) { |
1444 | const char *value = NULL; |
1445 | if (__kmp_target_offload == tgt_default) |
1446 | value = "DEFAULT" ; |
1447 | else if (__kmp_target_offload == tgt_mandatory) |
1448 | value = "MANDATORY" ; |
1449 | else if (__kmp_target_offload == tgt_disabled) |
1450 | value = "DISABLED" ; |
1451 | KMP_DEBUG_ASSERT(value); |
1452 | if (__kmp_env_format) { |
1453 | KMP_STR_BUF_PRINT_NAME; |
1454 | } else { |
1455 | __kmp_str_buf_print(buffer, format: " %s" , name); |
1456 | } |
1457 | __kmp_str_buf_print(buffer, format: "=%s\n" , value); |
1458 | } // __kmp_stg_print_target_offload |
1459 | |
1460 | // ----------------------------------------------------------------------------- |
1461 | // OpenMP 4.5: OMP_MAX_TASK_PRIORITY |
1462 | static void __kmp_stg_parse_max_task_priority(char const *name, |
1463 | char const *value, void *data) { |
1464 | __kmp_stg_parse_int(name, value, min: 0, KMP_MAX_TASK_PRIORITY_LIMIT, |
1465 | out: &__kmp_max_task_priority); |
1466 | } // __kmp_stg_parse_max_task_priority |
1467 | |
1468 | static void __kmp_stg_print_max_task_priority(kmp_str_buf_t *buffer, |
1469 | char const *name, void *data) { |
1470 | __kmp_stg_print_int(buffer, name, value: __kmp_max_task_priority); |
1471 | } // __kmp_stg_print_max_task_priority |
1472 | |
1473 | // KMP_TASKLOOP_MIN_TASKS |
1474 | // taskloop threshold to switch from recursive to linear tasks creation |
1475 | static void __kmp_stg_parse_taskloop_min_tasks(char const *name, |
1476 | char const *value, void *data) { |
1477 | int tmp = 0; |
1478 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX, out: &tmp); |
1479 | __kmp_taskloop_min_tasks = tmp; |
1480 | } // __kmp_stg_parse_taskloop_min_tasks |
1481 | |
1482 | static void __kmp_stg_print_taskloop_min_tasks(kmp_str_buf_t *buffer, |
1483 | char const *name, void *data) { |
1484 | __kmp_stg_print_uint64(buffer, name, value: __kmp_taskloop_min_tasks); |
1485 | } // __kmp_stg_print_taskloop_min_tasks |
1486 | |
1487 | // ----------------------------------------------------------------------------- |
1488 | // KMP_DISP_NUM_BUFFERS |
1489 | static void __kmp_stg_parse_disp_buffers(char const *name, char const *value, |
1490 | void *data) { |
1491 | if (TCR_4(__kmp_init_serial)) { |
1492 | KMP_WARNING(EnvSerialWarn, name); |
1493 | return; |
1494 | } // read value before serial initialization only |
1495 | __kmp_stg_parse_int(name, value, KMP_MIN_DISP_NUM_BUFF, KMP_MAX_DISP_NUM_BUFF, |
1496 | out: &__kmp_dispatch_num_buffers); |
1497 | } // __kmp_stg_parse_disp_buffers |
1498 | |
1499 | static void __kmp_stg_print_disp_buffers(kmp_str_buf_t *buffer, |
1500 | char const *name, void *data) { |
1501 | __kmp_stg_print_int(buffer, name, value: __kmp_dispatch_num_buffers); |
1502 | } // __kmp_stg_print_disp_buffers |
1503 | |
1504 | #if KMP_NESTED_HOT_TEAMS |
1505 | // ----------------------------------------------------------------------------- |
1506 | // KMP_HOT_TEAMS_MAX_LEVEL, KMP_HOT_TEAMS_MODE |
1507 | |
1508 | static void __kmp_stg_parse_hot_teams_level(char const *name, char const *value, |
1509 | void *data) { |
1510 | if (TCR_4(__kmp_init_parallel)) { |
1511 | KMP_WARNING(EnvParallelWarn, name); |
1512 | return; |
1513 | } // read value before first parallel only |
1514 | __kmp_stg_parse_int(name, value, min: 0, KMP_MAX_ACTIVE_LEVELS_LIMIT, |
1515 | out: &__kmp_hot_teams_max_level); |
1516 | } // __kmp_stg_parse_hot_teams_level |
1517 | |
1518 | static void __kmp_stg_print_hot_teams_level(kmp_str_buf_t *buffer, |
1519 | char const *name, void *data) { |
1520 | __kmp_stg_print_int(buffer, name, value: __kmp_hot_teams_max_level); |
1521 | } // __kmp_stg_print_hot_teams_level |
1522 | |
1523 | static void __kmp_stg_parse_hot_teams_mode(char const *name, char const *value, |
1524 | void *data) { |
1525 | if (TCR_4(__kmp_init_parallel)) { |
1526 | KMP_WARNING(EnvParallelWarn, name); |
1527 | return; |
1528 | } // read value before first parallel only |
1529 | __kmp_stg_parse_int(name, value, min: 0, KMP_MAX_ACTIVE_LEVELS_LIMIT, |
1530 | out: &__kmp_hot_teams_mode); |
1531 | } // __kmp_stg_parse_hot_teams_mode |
1532 | |
1533 | static void __kmp_stg_print_hot_teams_mode(kmp_str_buf_t *buffer, |
1534 | char const *name, void *data) { |
1535 | __kmp_stg_print_int(buffer, name, value: __kmp_hot_teams_mode); |
1536 | } // __kmp_stg_print_hot_teams_mode |
1537 | |
1538 | #endif // KMP_NESTED_HOT_TEAMS |
1539 | |
1540 | // ----------------------------------------------------------------------------- |
1541 | // KMP_HANDLE_SIGNALS |
1542 | |
1543 | #if KMP_HANDLE_SIGNALS |
1544 | |
1545 | static void __kmp_stg_parse_handle_signals(char const *name, char const *value, |
1546 | void *data) { |
1547 | __kmp_stg_parse_bool(name, value, out: &__kmp_handle_signals); |
1548 | } // __kmp_stg_parse_handle_signals |
1549 | |
1550 | static void __kmp_stg_print_handle_signals(kmp_str_buf_t *buffer, |
1551 | char const *name, void *data) { |
1552 | __kmp_stg_print_bool(buffer, name, value: __kmp_handle_signals); |
1553 | } // __kmp_stg_print_handle_signals |
1554 | |
1555 | #endif // KMP_HANDLE_SIGNALS |
1556 | |
1557 | // ----------------------------------------------------------------------------- |
1558 | // KMP_X_DEBUG, KMP_DEBUG, KMP_DEBUG_BUF_*, KMP_DIAG |
1559 | |
1560 | #ifdef KMP_DEBUG |
1561 | |
1562 | #define KMP_STG_X_DEBUG(x) \ |
1563 | static void __kmp_stg_parse_##x##_debug(char const *name, char const *value, \ |
1564 | void *data) { \ |
1565 | __kmp_stg_parse_int(name, value, 0, INT_MAX, &kmp_##x##_debug); \ |
1566 | } /* __kmp_stg_parse_x_debug */ \ |
1567 | static void __kmp_stg_print_##x##_debug(kmp_str_buf_t *buffer, \ |
1568 | char const *name, void *data) { \ |
1569 | __kmp_stg_print_int(buffer, name, kmp_##x##_debug); \ |
1570 | } /* __kmp_stg_print_x_debug */ |
1571 | |
1572 | KMP_STG_X_DEBUG(a) |
1573 | KMP_STG_X_DEBUG(b) |
1574 | KMP_STG_X_DEBUG(c) |
1575 | KMP_STG_X_DEBUG(d) |
1576 | KMP_STG_X_DEBUG(e) |
1577 | KMP_STG_X_DEBUG(f) |
1578 | |
1579 | #undef KMP_STG_X_DEBUG |
1580 | |
1581 | static void __kmp_stg_parse_debug(char const *name, char const *value, |
1582 | void *data) { |
1583 | int debug = 0; |
1584 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX, out: &debug); |
1585 | if (kmp_a_debug < debug) { |
1586 | kmp_a_debug = debug; |
1587 | } |
1588 | if (kmp_b_debug < debug) { |
1589 | kmp_b_debug = debug; |
1590 | } |
1591 | if (kmp_c_debug < debug) { |
1592 | kmp_c_debug = debug; |
1593 | } |
1594 | if (kmp_d_debug < debug) { |
1595 | kmp_d_debug = debug; |
1596 | } |
1597 | if (kmp_e_debug < debug) { |
1598 | kmp_e_debug = debug; |
1599 | } |
1600 | if (kmp_f_debug < debug) { |
1601 | kmp_f_debug = debug; |
1602 | } |
1603 | } // __kmp_stg_parse_debug |
1604 | |
1605 | static void __kmp_stg_parse_debug_buf(char const *name, char const *value, |
1606 | void *data) { |
1607 | __kmp_stg_parse_bool(name, value, out: &__kmp_debug_buf); |
1608 | // !!! TODO: Move buffer initialization of this file! It may works |
1609 | // incorrectly if KMP_DEBUG_BUF is parsed before KMP_DEBUG_BUF_LINES or |
1610 | // KMP_DEBUG_BUF_CHARS. |
1611 | if (__kmp_debug_buf) { |
1612 | int i; |
1613 | int elements = __kmp_debug_buf_lines * __kmp_debug_buf_chars; |
1614 | |
1615 | /* allocate and initialize all entries in debug buffer to empty */ |
1616 | __kmp_debug_buffer = (char *)__kmp_page_allocate(elements * sizeof(char)); |
1617 | for (i = 0; i < elements; i += __kmp_debug_buf_chars) |
1618 | __kmp_debug_buffer[i] = '\0'; |
1619 | |
1620 | __kmp_debug_count = 0; |
1621 | } |
1622 | K_DIAG(1, ("__kmp_debug_buf = %d\n" , __kmp_debug_buf)); |
1623 | } // __kmp_stg_parse_debug_buf |
1624 | |
1625 | static void __kmp_stg_print_debug_buf(kmp_str_buf_t *buffer, char const *name, |
1626 | void *data) { |
1627 | __kmp_stg_print_bool(buffer, name, value: __kmp_debug_buf); |
1628 | } // __kmp_stg_print_debug_buf |
1629 | |
1630 | static void __kmp_stg_parse_debug_buf_atomic(char const *name, |
1631 | char const *value, void *data) { |
1632 | __kmp_stg_parse_bool(name, value, out: &__kmp_debug_buf_atomic); |
1633 | } // __kmp_stg_parse_debug_buf_atomic |
1634 | |
1635 | static void __kmp_stg_print_debug_buf_atomic(kmp_str_buf_t *buffer, |
1636 | char const *name, void *data) { |
1637 | __kmp_stg_print_bool(buffer, name, value: __kmp_debug_buf_atomic); |
1638 | } // __kmp_stg_print_debug_buf_atomic |
1639 | |
1640 | static void __kmp_stg_parse_debug_buf_chars(char const *name, char const *value, |
1641 | void *data) { |
1642 | __kmp_stg_parse_int(name, value, KMP_DEBUG_BUF_CHARS_MIN, INT_MAX, |
1643 | out: &__kmp_debug_buf_chars); |
1644 | } // __kmp_stg_debug_parse_buf_chars |
1645 | |
1646 | static void __kmp_stg_print_debug_buf_chars(kmp_str_buf_t *buffer, |
1647 | char const *name, void *data) { |
1648 | __kmp_stg_print_int(buffer, name, value: __kmp_debug_buf_chars); |
1649 | } // __kmp_stg_print_debug_buf_chars |
1650 | |
1651 | static void __kmp_stg_parse_debug_buf_lines(char const *name, char const *value, |
1652 | void *data) { |
1653 | __kmp_stg_parse_int(name, value, KMP_DEBUG_BUF_LINES_MIN, INT_MAX, |
1654 | out: &__kmp_debug_buf_lines); |
1655 | } // __kmp_stg_parse_debug_buf_lines |
1656 | |
1657 | static void __kmp_stg_print_debug_buf_lines(kmp_str_buf_t *buffer, |
1658 | char const *name, void *data) { |
1659 | __kmp_stg_print_int(buffer, name, value: __kmp_debug_buf_lines); |
1660 | } // __kmp_stg_print_debug_buf_lines |
1661 | |
1662 | static void __kmp_stg_parse_diag(char const *name, char const *value, |
1663 | void *data) { |
1664 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX, out: &kmp_diag); |
1665 | } // __kmp_stg_parse_diag |
1666 | |
1667 | static void __kmp_stg_print_diag(kmp_str_buf_t *buffer, char const *name, |
1668 | void *data) { |
1669 | __kmp_stg_print_int(buffer, name, value: kmp_diag); |
1670 | } // __kmp_stg_print_diag |
1671 | |
1672 | #endif // KMP_DEBUG |
1673 | |
1674 | // ----------------------------------------------------------------------------- |
1675 | // KMP_ALIGN_ALLOC |
1676 | |
1677 | static void __kmp_stg_parse_align_alloc(char const *name, char const *value, |
1678 | void *data) { |
1679 | __kmp_stg_parse_size(name, value, CACHE_LINE, INT_MAX, NULL, |
1680 | out: &__kmp_align_alloc, factor: 1); |
1681 | } // __kmp_stg_parse_align_alloc |
1682 | |
1683 | static void __kmp_stg_print_align_alloc(kmp_str_buf_t *buffer, char const *name, |
1684 | void *data) { |
1685 | __kmp_stg_print_size(buffer, name, value: __kmp_align_alloc); |
1686 | } // __kmp_stg_print_align_alloc |
1687 | |
1688 | // ----------------------------------------------------------------------------- |
1689 | // KMP_PLAIN_BARRIER, KMP_FORKJOIN_BARRIER, KMP_REDUCTION_BARRIER |
1690 | |
1691 | // TODO: Remove __kmp_barrier_branch_bit_env_name varibale, remove loops from |
1692 | // parse and print functions, pass required info through data argument. |
1693 | |
1694 | static void __kmp_stg_parse_barrier_branch_bit(char const *name, |
1695 | char const *value, void *data) { |
1696 | const char *var; |
1697 | |
1698 | /* ---------- Barrier branch bit control ------------ */ |
1699 | for (int i = bs_plain_barrier; i < bs_last_barrier; i++) { |
1700 | var = __kmp_barrier_branch_bit_env_name[i]; |
1701 | if ((strcmp(s1: var, s2: name) == 0) && (value != 0)) { |
1702 | char *comma; |
1703 | |
1704 | comma = CCAST(char *, strchr(value, ',')); |
1705 | __kmp_barrier_gather_branch_bits[i] = |
1706 | (kmp_uint32)__kmp_str_to_int(str: value, sentinel: ','); |
1707 | /* is there a specified release parameter? */ |
1708 | if (comma == NULL) { |
1709 | __kmp_barrier_release_branch_bits[i] = __kmp_barrier_release_bb_dflt; |
1710 | } else { |
1711 | __kmp_barrier_release_branch_bits[i] = |
1712 | (kmp_uint32)__kmp_str_to_int(str: comma + 1, sentinel: 0); |
1713 | |
1714 | if (__kmp_barrier_release_branch_bits[i] > KMP_MAX_BRANCH_BITS) { |
1715 | __kmp_msg(kmp_ms_warning, |
1716 | KMP_MSG(BarrReleaseValueInvalid, name, comma + 1), |
1717 | __kmp_msg_null); |
1718 | __kmp_barrier_release_branch_bits[i] = __kmp_barrier_release_bb_dflt; |
1719 | } |
1720 | } |
1721 | if (__kmp_barrier_gather_branch_bits[i] > KMP_MAX_BRANCH_BITS) { |
1722 | KMP_WARNING(BarrGatherValueInvalid, name, value); |
1723 | KMP_INFORM(Using_uint_Value, name, __kmp_barrier_gather_bb_dflt); |
1724 | __kmp_barrier_gather_branch_bits[i] = __kmp_barrier_gather_bb_dflt; |
1725 | } |
1726 | } |
1727 | K_DIAG(1, ("%s == %d,%d\n" , __kmp_barrier_branch_bit_env_name[i], |
1728 | __kmp_barrier_gather_branch_bits[i], |
1729 | __kmp_barrier_release_branch_bits[i])) |
1730 | } |
1731 | } // __kmp_stg_parse_barrier_branch_bit |
1732 | |
1733 | static void __kmp_stg_print_barrier_branch_bit(kmp_str_buf_t *buffer, |
1734 | char const *name, void *data) { |
1735 | const char *var; |
1736 | for (int i = bs_plain_barrier; i < bs_last_barrier; i++) { |
1737 | var = __kmp_barrier_branch_bit_env_name[i]; |
1738 | if (strcmp(s1: var, s2: name) == 0) { |
1739 | if (__kmp_env_format) { |
1740 | KMP_STR_BUF_PRINT_NAME_EX(__kmp_barrier_branch_bit_env_name[i]); |
1741 | } else { |
1742 | __kmp_str_buf_print(buffer, format: " %s='" , |
1743 | __kmp_barrier_branch_bit_env_name[i]); |
1744 | } |
1745 | __kmp_str_buf_print(buffer, format: "%d,%d'\n" , |
1746 | __kmp_barrier_gather_branch_bits[i], |
1747 | __kmp_barrier_release_branch_bits[i]); |
1748 | } |
1749 | } |
1750 | } // __kmp_stg_print_barrier_branch_bit |
1751 | |
1752 | // ---------------------------------------------------------------------------- |
1753 | // KMP_PLAIN_BARRIER_PATTERN, KMP_FORKJOIN_BARRIER_PATTERN, |
1754 | // KMP_REDUCTION_BARRIER_PATTERN |
1755 | |
1756 | // TODO: Remove __kmp_barrier_pattern_name variable, remove loops from parse and |
1757 | // print functions, pass required data to functions through data argument. |
1758 | |
1759 | static void __kmp_stg_parse_barrier_pattern(char const *name, char const *value, |
1760 | void *data) { |
1761 | const char *var; |
1762 | /* ---------- Barrier method control ------------ */ |
1763 | |
1764 | static int dist_req = 0, non_dist_req = 0; |
1765 | static bool warn = 1; |
1766 | for (int i = bs_plain_barrier; i < bs_last_barrier; i++) { |
1767 | var = __kmp_barrier_pattern_env_name[i]; |
1768 | |
1769 | if ((strcmp(s1: var, s2: name) == 0) && (value != 0)) { |
1770 | int j; |
1771 | char *comma = CCAST(char *, strchr(value, ',')); |
1772 | |
1773 | /* handle first parameter: gather pattern */ |
1774 | for (j = bp_linear_bar; j < bp_last_bar; j++) { |
1775 | if (__kmp_match_with_sentinel(a: __kmp_barrier_pattern_name[j], b: value, len: 1, |
1776 | sentinel: ',')) { |
1777 | if (j == bp_dist_bar) { |
1778 | dist_req++; |
1779 | } else { |
1780 | non_dist_req++; |
1781 | } |
1782 | __kmp_barrier_gather_pattern[i] = (kmp_bar_pat_e)j; |
1783 | break; |
1784 | } |
1785 | } |
1786 | if (j == bp_last_bar) { |
1787 | KMP_WARNING(BarrGatherValueInvalid, name, value); |
1788 | KMP_INFORM(Using_str_Value, name, |
1789 | __kmp_barrier_pattern_name[bp_linear_bar]); |
1790 | } |
1791 | |
1792 | /* handle second parameter: release pattern */ |
1793 | if (comma != NULL) { |
1794 | for (j = bp_linear_bar; j < bp_last_bar; j++) { |
1795 | if (__kmp_str_match(target: __kmp_barrier_pattern_name[j], len: 1, data: comma + 1)) { |
1796 | if (j == bp_dist_bar) { |
1797 | dist_req++; |
1798 | } else { |
1799 | non_dist_req++; |
1800 | } |
1801 | __kmp_barrier_release_pattern[i] = (kmp_bar_pat_e)j; |
1802 | break; |
1803 | } |
1804 | } |
1805 | if (j == bp_last_bar) { |
1806 | __kmp_msg(kmp_ms_warning, |
1807 | KMP_MSG(BarrReleaseValueInvalid, name, comma + 1), |
1808 | __kmp_msg_null); |
1809 | KMP_INFORM(Using_str_Value, name, |
1810 | __kmp_barrier_pattern_name[bp_linear_bar]); |
1811 | } |
1812 | } |
1813 | } |
1814 | } |
1815 | if (dist_req != 0) { |
1816 | // set all barriers to dist |
1817 | if ((non_dist_req != 0) && warn) { |
1818 | KMP_INFORM(BarrierPatternOverride, name, |
1819 | __kmp_barrier_pattern_name[bp_dist_bar]); |
1820 | warn = 0; |
1821 | } |
1822 | for (int i = bs_plain_barrier; i < bs_last_barrier; i++) { |
1823 | if (__kmp_barrier_release_pattern[i] != bp_dist_bar) |
1824 | __kmp_barrier_release_pattern[i] = bp_dist_bar; |
1825 | if (__kmp_barrier_gather_pattern[i] != bp_dist_bar) |
1826 | __kmp_barrier_gather_pattern[i] = bp_dist_bar; |
1827 | } |
1828 | } |
1829 | } // __kmp_stg_parse_barrier_pattern |
1830 | |
1831 | static void __kmp_stg_print_barrier_pattern(kmp_str_buf_t *buffer, |
1832 | char const *name, void *data) { |
1833 | const char *var; |
1834 | for (int i = bs_plain_barrier; i < bs_last_barrier; i++) { |
1835 | var = __kmp_barrier_pattern_env_name[i]; |
1836 | if (strcmp(s1: var, s2: name) == 0) { |
1837 | int j = __kmp_barrier_gather_pattern[i]; |
1838 | int k = __kmp_barrier_release_pattern[i]; |
1839 | if (__kmp_env_format) { |
1840 | KMP_STR_BUF_PRINT_NAME_EX(__kmp_barrier_pattern_env_name[i]); |
1841 | } else { |
1842 | __kmp_str_buf_print(buffer, format: " %s='" , |
1843 | __kmp_barrier_pattern_env_name[i]); |
1844 | } |
1845 | KMP_DEBUG_ASSERT(j < bp_last_bar && k < bp_last_bar); |
1846 | __kmp_str_buf_print(buffer, format: "%s,%s'\n" , __kmp_barrier_pattern_name[j], |
1847 | __kmp_barrier_pattern_name[k]); |
1848 | } |
1849 | } |
1850 | } // __kmp_stg_print_barrier_pattern |
1851 | |
1852 | // ----------------------------------------------------------------------------- |
1853 | // KMP_ABORT_DELAY |
1854 | |
1855 | static void __kmp_stg_parse_abort_delay(char const *name, char const *value, |
1856 | void *data) { |
1857 | // Units of KMP_DELAY_ABORT are seconds, units of __kmp_abort_delay is |
1858 | // milliseconds. |
1859 | int delay = __kmp_abort_delay / 1000; |
1860 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX / 1000, out: &delay); |
1861 | __kmp_abort_delay = delay * 1000; |
1862 | } // __kmp_stg_parse_abort_delay |
1863 | |
1864 | static void __kmp_stg_print_abort_delay(kmp_str_buf_t *buffer, char const *name, |
1865 | void *data) { |
1866 | __kmp_stg_print_int(buffer, name, value: __kmp_abort_delay); |
1867 | } // __kmp_stg_print_abort_delay |
1868 | |
1869 | // ----------------------------------------------------------------------------- |
1870 | // KMP_CPUINFO_FILE |
1871 | |
1872 | static void __kmp_stg_parse_cpuinfo_file(char const *name, char const *value, |
1873 | void *data) { |
1874 | #if KMP_AFFINITY_SUPPORTED |
1875 | __kmp_stg_parse_str(name, value, out: &__kmp_cpuinfo_file); |
1876 | K_DIAG(1, ("__kmp_cpuinfo_file == %s\n" , __kmp_cpuinfo_file)); |
1877 | #endif |
1878 | } //__kmp_stg_parse_cpuinfo_file |
1879 | |
1880 | static void __kmp_stg_print_cpuinfo_file(kmp_str_buf_t *buffer, |
1881 | char const *name, void *data) { |
1882 | #if KMP_AFFINITY_SUPPORTED |
1883 | if (__kmp_env_format) { |
1884 | KMP_STR_BUF_PRINT_NAME; |
1885 | } else { |
1886 | __kmp_str_buf_print(buffer, format: " %s" , name); |
1887 | } |
1888 | if (__kmp_cpuinfo_file) { |
1889 | __kmp_str_buf_print(buffer, format: "='%s'\n" , __kmp_cpuinfo_file); |
1890 | } else { |
1891 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
1892 | } |
1893 | #endif |
1894 | } //__kmp_stg_print_cpuinfo_file |
1895 | |
1896 | // ----------------------------------------------------------------------------- |
1897 | // KMP_FORCE_REDUCTION, KMP_DETERMINISTIC_REDUCTION |
1898 | |
1899 | static void __kmp_stg_parse_force_reduction(char const *name, char const *value, |
1900 | void *data) { |
1901 | kmp_stg_fr_data_t *reduction = (kmp_stg_fr_data_t *)data; |
1902 | int rc; |
1903 | |
1904 | rc = __kmp_stg_check_rivals(name, value, rivals: reduction->rivals); |
1905 | if (rc) { |
1906 | return; |
1907 | } |
1908 | if (reduction->force) { |
1909 | if (value != 0) { |
1910 | if (__kmp_str_match(target: "critical" , len: 0, data: value)) |
1911 | __kmp_force_reduction_method = critical_reduce_block; |
1912 | else if (__kmp_str_match(target: "atomic" , len: 0, data: value)) |
1913 | __kmp_force_reduction_method = atomic_reduce_block; |
1914 | else if (__kmp_str_match(target: "tree" , len: 0, data: value)) |
1915 | __kmp_force_reduction_method = tree_reduce_block; |
1916 | else { |
1917 | KMP_FATAL(UnknownForceReduction, name, value); |
1918 | } |
1919 | } |
1920 | } else { |
1921 | __kmp_stg_parse_bool(name, value, out: &__kmp_determ_red); |
1922 | if (__kmp_determ_red) { |
1923 | __kmp_force_reduction_method = tree_reduce_block; |
1924 | } else { |
1925 | __kmp_force_reduction_method = reduction_method_not_defined; |
1926 | } |
1927 | } |
1928 | K_DIAG(1, ("__kmp_force_reduction_method == %d\n" , |
1929 | __kmp_force_reduction_method)); |
1930 | } // __kmp_stg_parse_force_reduction |
1931 | |
1932 | static void __kmp_stg_print_force_reduction(kmp_str_buf_t *buffer, |
1933 | char const *name, void *data) { |
1934 | |
1935 | kmp_stg_fr_data_t *reduction = (kmp_stg_fr_data_t *)data; |
1936 | if (reduction->force) { |
1937 | if (__kmp_force_reduction_method == critical_reduce_block) { |
1938 | __kmp_stg_print_str(buffer, name, value: "critical" ); |
1939 | } else if (__kmp_force_reduction_method == atomic_reduce_block) { |
1940 | __kmp_stg_print_str(buffer, name, value: "atomic" ); |
1941 | } else if (__kmp_force_reduction_method == tree_reduce_block) { |
1942 | __kmp_stg_print_str(buffer, name, value: "tree" ); |
1943 | } else { |
1944 | if (__kmp_env_format) { |
1945 | KMP_STR_BUF_PRINT_NAME; |
1946 | } else { |
1947 | __kmp_str_buf_print(buffer, format: " %s" , name); |
1948 | } |
1949 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
1950 | } |
1951 | } else { |
1952 | __kmp_stg_print_bool(buffer, name, value: __kmp_determ_red); |
1953 | } |
1954 | |
1955 | } // __kmp_stg_print_force_reduction |
1956 | |
1957 | // ----------------------------------------------------------------------------- |
1958 | // KMP_STORAGE_MAP |
1959 | |
1960 | static void __kmp_stg_parse_storage_map(char const *name, char const *value, |
1961 | void *data) { |
1962 | if (__kmp_str_match(target: "verbose" , len: 1, data: value)) { |
1963 | __kmp_storage_map = TRUE; |
1964 | __kmp_storage_map_verbose = TRUE; |
1965 | __kmp_storage_map_verbose_specified = TRUE; |
1966 | |
1967 | } else { |
1968 | __kmp_storage_map_verbose = FALSE; |
1969 | __kmp_stg_parse_bool(name, value, out: &__kmp_storage_map); // !!! |
1970 | } |
1971 | } // __kmp_stg_parse_storage_map |
1972 | |
1973 | static void __kmp_stg_print_storage_map(kmp_str_buf_t *buffer, char const *name, |
1974 | void *data) { |
1975 | if (__kmp_storage_map_verbose || __kmp_storage_map_verbose_specified) { |
1976 | __kmp_stg_print_str(buffer, name, value: "verbose" ); |
1977 | } else { |
1978 | __kmp_stg_print_bool(buffer, name, value: __kmp_storage_map); |
1979 | } |
1980 | } // __kmp_stg_print_storage_map |
1981 | |
1982 | // ----------------------------------------------------------------------------- |
1983 | // KMP_ALL_THREADPRIVATE |
1984 | |
1985 | static void __kmp_stg_parse_all_threadprivate(char const *name, |
1986 | char const *value, void *data) { |
1987 | __kmp_stg_parse_int(name, value, |
1988 | min: __kmp_allThreadsSpecified ? __kmp_max_nth : 1, |
1989 | max: __kmp_max_nth, out: &__kmp_tp_capacity); |
1990 | } // __kmp_stg_parse_all_threadprivate |
1991 | |
1992 | static void __kmp_stg_print_all_threadprivate(kmp_str_buf_t *buffer, |
1993 | char const *name, void *data) { |
1994 | __kmp_stg_print_int(buffer, name, value: __kmp_tp_capacity); |
1995 | } |
1996 | |
1997 | // ----------------------------------------------------------------------------- |
1998 | // KMP_FOREIGN_THREADS_THREADPRIVATE |
1999 | |
2000 | static void __kmp_stg_parse_foreign_threads_threadprivate(char const *name, |
2001 | char const *value, |
2002 | void *data) { |
2003 | __kmp_stg_parse_bool(name, value, out: &__kmp_foreign_tp); |
2004 | } // __kmp_stg_parse_foreign_threads_threadprivate |
2005 | |
2006 | static void __kmp_stg_print_foreign_threads_threadprivate(kmp_str_buf_t *buffer, |
2007 | char const *name, |
2008 | void *data) { |
2009 | __kmp_stg_print_bool(buffer, name, value: __kmp_foreign_tp); |
2010 | } // __kmp_stg_print_foreign_threads_threadprivate |
2011 | |
2012 | // ----------------------------------------------------------------------------- |
2013 | // KMP_AFFINITY, GOMP_CPU_AFFINITY, KMP_TOPOLOGY_METHOD |
2014 | |
2015 | static inline const char * |
2016 | __kmp_hw_get_core_type_keyword(kmp_hw_core_type_t type) { |
2017 | switch (type) { |
2018 | case KMP_HW_CORE_TYPE_UNKNOWN: |
2019 | case KMP_HW_MAX_NUM_CORE_TYPES: |
2020 | return "unknown" ; |
2021 | #if KMP_ARCH_X86 || KMP_ARCH_X86_64 |
2022 | case KMP_HW_CORE_TYPE_ATOM: |
2023 | return "intel_atom" ; |
2024 | case KMP_HW_CORE_TYPE_CORE: |
2025 | return "intel_core" ; |
2026 | #endif |
2027 | } |
2028 | KMP_ASSERT2(false, "Unhandled kmp_hw_core_type_t enumeration" ); |
2029 | KMP_BUILTIN_UNREACHABLE; |
2030 | } |
2031 | |
2032 | #if KMP_AFFINITY_SUPPORTED |
2033 | // Parse the proc id list. Return TRUE if successful, FALSE otherwise. |
2034 | static int __kmp_parse_affinity_proc_id_list(const char *var, const char *env, |
2035 | const char **nextEnv, |
2036 | char **proclist) { |
2037 | const char *scan = env; |
2038 | const char *next = scan; |
2039 | int empty = TRUE; |
2040 | |
2041 | *proclist = NULL; |
2042 | |
2043 | for (;;) { |
2044 | int start, end, stride; |
2045 | |
2046 | SKIP_WS(scan); |
2047 | next = scan; |
2048 | if (*next == '\0') { |
2049 | break; |
2050 | } |
2051 | |
2052 | if (*next == '{') { |
2053 | int num; |
2054 | next++; // skip '{' |
2055 | SKIP_WS(next); |
2056 | scan = next; |
2057 | |
2058 | // Read the first integer in the set. |
2059 | if ((*next < '0') || (*next > '9')) { |
2060 | KMP_WARNING(AffSyntaxError, var); |
2061 | return FALSE; |
2062 | } |
2063 | SKIP_DIGITS(next); |
2064 | num = __kmp_str_to_int(str: scan, sentinel: *next); |
2065 | KMP_ASSERT(num >= 0); |
2066 | |
2067 | for (;;) { |
2068 | // Check for end of set. |
2069 | SKIP_WS(next); |
2070 | if (*next == '}') { |
2071 | next++; // skip '}' |
2072 | break; |
2073 | } |
2074 | |
2075 | // Skip optional comma. |
2076 | if (*next == ',') { |
2077 | next++; |
2078 | } |
2079 | SKIP_WS(next); |
2080 | |
2081 | // Read the next integer in the set. |
2082 | scan = next; |
2083 | if ((*next < '0') || (*next > '9')) { |
2084 | KMP_WARNING(AffSyntaxError, var); |
2085 | return FALSE; |
2086 | } |
2087 | |
2088 | SKIP_DIGITS(next); |
2089 | num = __kmp_str_to_int(str: scan, sentinel: *next); |
2090 | KMP_ASSERT(num >= 0); |
2091 | } |
2092 | empty = FALSE; |
2093 | |
2094 | SKIP_WS(next); |
2095 | if (*next == ',') { |
2096 | next++; |
2097 | } |
2098 | scan = next; |
2099 | continue; |
2100 | } |
2101 | |
2102 | // Next character is not an integer => end of list |
2103 | if ((*next < '0') || (*next > '9')) { |
2104 | if (empty) { |
2105 | KMP_WARNING(AffSyntaxError, var); |
2106 | return FALSE; |
2107 | } |
2108 | break; |
2109 | } |
2110 | |
2111 | // Read the first integer. |
2112 | SKIP_DIGITS(next); |
2113 | start = __kmp_str_to_int(str: scan, sentinel: *next); |
2114 | KMP_ASSERT(start >= 0); |
2115 | SKIP_WS(next); |
2116 | |
2117 | // If this isn't a range, then go on. |
2118 | if (*next != '-') { |
2119 | empty = FALSE; |
2120 | |
2121 | // Skip optional comma. |
2122 | if (*next == ',') { |
2123 | next++; |
2124 | } |
2125 | scan = next; |
2126 | continue; |
2127 | } |
2128 | |
2129 | // This is a range. Skip over the '-' and read in the 2nd int. |
2130 | next++; // skip '-' |
2131 | SKIP_WS(next); |
2132 | scan = next; |
2133 | if ((*next < '0') || (*next > '9')) { |
2134 | KMP_WARNING(AffSyntaxError, var); |
2135 | return FALSE; |
2136 | } |
2137 | SKIP_DIGITS(next); |
2138 | end = __kmp_str_to_int(str: scan, sentinel: *next); |
2139 | KMP_ASSERT(end >= 0); |
2140 | |
2141 | // Check for a stride parameter |
2142 | stride = 1; |
2143 | SKIP_WS(next); |
2144 | if (*next == ':') { |
2145 | // A stride is specified. Skip over the ':" and read the 3rd int. |
2146 | int sign = +1; |
2147 | next++; // skip ':' |
2148 | SKIP_WS(next); |
2149 | scan = next; |
2150 | if (*next == '-') { |
2151 | sign = -1; |
2152 | next++; |
2153 | SKIP_WS(next); |
2154 | scan = next; |
2155 | } |
2156 | if ((*next < '0') || (*next > '9')) { |
2157 | KMP_WARNING(AffSyntaxError, var); |
2158 | return FALSE; |
2159 | } |
2160 | SKIP_DIGITS(next); |
2161 | stride = __kmp_str_to_int(str: scan, sentinel: *next); |
2162 | KMP_ASSERT(stride >= 0); |
2163 | stride *= sign; |
2164 | } |
2165 | |
2166 | // Do some range checks. |
2167 | if (stride == 0) { |
2168 | KMP_WARNING(AffZeroStride, var); |
2169 | return FALSE; |
2170 | } |
2171 | if (stride > 0) { |
2172 | if (start > end) { |
2173 | KMP_WARNING(AffStartGreaterEnd, var, start, end); |
2174 | return FALSE; |
2175 | } |
2176 | } else { |
2177 | if (start < end) { |
2178 | KMP_WARNING(AffStrideLessZero, var, start, end); |
2179 | return FALSE; |
2180 | } |
2181 | } |
2182 | if ((end - start) / stride > 65536) { |
2183 | KMP_WARNING(AffRangeTooBig, var, end, start, stride); |
2184 | return FALSE; |
2185 | } |
2186 | |
2187 | empty = FALSE; |
2188 | |
2189 | // Skip optional comma. |
2190 | SKIP_WS(next); |
2191 | if (*next == ',') { |
2192 | next++; |
2193 | } |
2194 | scan = next; |
2195 | } |
2196 | |
2197 | *nextEnv = next; |
2198 | |
2199 | { |
2200 | ptrdiff_t len = next - env; |
2201 | char *retlist = (char *)__kmp_allocate((len + 1) * sizeof(char)); |
2202 | KMP_MEMCPY_S(retlist, (len + 1) * sizeof(char), env, len * sizeof(char)); |
2203 | retlist[len] = '\0'; |
2204 | *proclist = retlist; |
2205 | } |
2206 | return TRUE; |
2207 | } |
2208 | |
2209 | // If KMP_AFFINITY is specified without a type, then |
2210 | // __kmp_affinity_notype should point to its setting. |
2211 | static kmp_setting_t *__kmp_affinity_notype = NULL; |
2212 | |
2213 | static void __kmp_parse_affinity_env(char const *name, char const *value, |
2214 | kmp_affinity_t *out_affinity) { |
2215 | char *buffer = NULL; // Copy of env var value. |
2216 | char *buf = NULL; // Buffer for strtok_r() function. |
2217 | char *next = NULL; // end of token / start of next. |
2218 | const char *start; // start of current token (for err msgs) |
2219 | int count = 0; // Counter of parsed integer numbers. |
2220 | int number[2]; // Parsed numbers. |
2221 | |
2222 | // Guards. |
2223 | int type = 0; |
2224 | int proclist = 0; |
2225 | int verbose = 0; |
2226 | int warnings = 0; |
2227 | int respect = 0; |
2228 | int gran = 0; |
2229 | int dups = 0; |
2230 | int reset = 0; |
2231 | bool set = false; |
2232 | |
2233 | KMP_ASSERT(value != NULL); |
2234 | |
2235 | if (TCR_4(__kmp_init_middle)) { |
2236 | KMP_WARNING(EnvMiddleWarn, name); |
2237 | __kmp_env_toPrint(name, flag: 0); |
2238 | return; |
2239 | } |
2240 | __kmp_env_toPrint(name, flag: 1); |
2241 | |
2242 | buffer = |
2243 | __kmp_str_format(format: "%s" , value); // Copy env var to keep original intact. |
2244 | buf = buffer; |
2245 | SKIP_WS(buf); |
2246 | |
2247 | // Helper macros. |
2248 | |
2249 | // If we see a parse error, emit a warning and scan to the next ",". |
2250 | // |
2251 | // FIXME - there's got to be a better way to print an error |
2252 | // message, hopefully without overwriting peices of buf. |
2253 | #define EMIT_WARN(skip, errlist) \ |
2254 | { \ |
2255 | char ch; \ |
2256 | if (skip) { \ |
2257 | SKIP_TO(next, ','); \ |
2258 | } \ |
2259 | ch = *next; \ |
2260 | *next = '\0'; \ |
2261 | KMP_WARNING errlist; \ |
2262 | *next = ch; \ |
2263 | if (skip) { \ |
2264 | if (ch == ',') \ |
2265 | next++; \ |
2266 | } \ |
2267 | buf = next; \ |
2268 | } |
2269 | |
2270 | #define _set_param(_guard, _var, _val) \ |
2271 | { \ |
2272 | if (_guard == 0) { \ |
2273 | _var = _val; \ |
2274 | } else { \ |
2275 | EMIT_WARN(FALSE, (AffParamDefined, name, start)); \ |
2276 | } \ |
2277 | ++_guard; \ |
2278 | } |
2279 | |
2280 | #define set_type(val) _set_param(type, out_affinity->type, val) |
2281 | #define set_verbose(val) _set_param(verbose, out_affinity->flags.verbose, val) |
2282 | #define set_warnings(val) \ |
2283 | _set_param(warnings, out_affinity->flags.warnings, val) |
2284 | #define set_respect(val) _set_param(respect, out_affinity->flags.respect, val) |
2285 | #define set_dups(val) _set_param(dups, out_affinity->flags.dups, val) |
2286 | #define set_proclist(val) _set_param(proclist, out_affinity->proclist, val) |
2287 | #define set_reset(val) _set_param(reset, out_affinity->flags.reset, val) |
2288 | |
2289 | #define set_gran(val, levels) \ |
2290 | { \ |
2291 | if (gran == 0) { \ |
2292 | out_affinity->gran = val; \ |
2293 | out_affinity->gran_levels = levels; \ |
2294 | } else { \ |
2295 | EMIT_WARN(FALSE, (AffParamDefined, name, start)); \ |
2296 | } \ |
2297 | ++gran; \ |
2298 | } |
2299 | |
2300 | KMP_DEBUG_ASSERT((__kmp_nested_proc_bind.bind_types != NULL) && |
2301 | (__kmp_nested_proc_bind.used > 0)); |
2302 | |
2303 | while (*buf != '\0') { |
2304 | start = next = buf; |
2305 | |
2306 | if (__kmp_match_str(token: "none" , buf, CCAST(const char **, &next))) { |
2307 | set_type(affinity_none); |
2308 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
2309 | buf = next; |
2310 | } else if (__kmp_match_str(token: "scatter" , buf, CCAST(const char **, &next))) { |
2311 | set_type(affinity_scatter); |
2312 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2313 | buf = next; |
2314 | } else if (__kmp_match_str(token: "compact" , buf, CCAST(const char **, &next))) { |
2315 | set_type(affinity_compact); |
2316 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2317 | buf = next; |
2318 | } else if (__kmp_match_str(token: "logical" , buf, CCAST(const char **, &next))) { |
2319 | set_type(affinity_logical); |
2320 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2321 | buf = next; |
2322 | } else if (__kmp_match_str(token: "physical" , buf, CCAST(const char **, &next))) { |
2323 | set_type(affinity_physical); |
2324 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2325 | buf = next; |
2326 | } else if (__kmp_match_str(token: "explicit" , buf, CCAST(const char **, &next))) { |
2327 | set_type(affinity_explicit); |
2328 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2329 | buf = next; |
2330 | } else if (__kmp_match_str(token: "balanced" , buf, CCAST(const char **, &next))) { |
2331 | set_type(affinity_balanced); |
2332 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2333 | buf = next; |
2334 | } else if (__kmp_match_str(token: "disabled" , buf, CCAST(const char **, &next))) { |
2335 | set_type(affinity_disabled); |
2336 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
2337 | buf = next; |
2338 | } else if (__kmp_match_str(token: "verbose" , buf, CCAST(const char **, &next))) { |
2339 | set_verbose(TRUE); |
2340 | buf = next; |
2341 | } else if (__kmp_match_str(token: "noverbose" , buf, CCAST(const char **, &next))) { |
2342 | set_verbose(FALSE); |
2343 | buf = next; |
2344 | } else if (__kmp_match_str(token: "warnings" , buf, CCAST(const char **, &next))) { |
2345 | set_warnings(TRUE); |
2346 | buf = next; |
2347 | } else if (__kmp_match_str(token: "nowarnings" , buf, |
2348 | CCAST(const char **, &next))) { |
2349 | set_warnings(FALSE); |
2350 | buf = next; |
2351 | } else if (__kmp_match_str(token: "respect" , buf, CCAST(const char **, &next))) { |
2352 | set_respect(TRUE); |
2353 | buf = next; |
2354 | } else if (__kmp_match_str(token: "norespect" , buf, CCAST(const char **, &next))) { |
2355 | set_respect(FALSE); |
2356 | buf = next; |
2357 | } else if (__kmp_match_str(token: "reset" , buf, CCAST(const char **, &next))) { |
2358 | set_reset(TRUE); |
2359 | buf = next; |
2360 | } else if (__kmp_match_str(token: "noreset" , buf, CCAST(const char **, &next))) { |
2361 | set_reset(FALSE); |
2362 | buf = next; |
2363 | } else if (__kmp_match_str(token: "duplicates" , buf, |
2364 | CCAST(const char **, &next)) || |
2365 | __kmp_match_str(token: "dups" , buf, CCAST(const char **, &next))) { |
2366 | set_dups(TRUE); |
2367 | buf = next; |
2368 | } else if (__kmp_match_str(token: "noduplicates" , buf, |
2369 | CCAST(const char **, &next)) || |
2370 | __kmp_match_str(token: "nodups" , buf, CCAST(const char **, &next))) { |
2371 | set_dups(FALSE); |
2372 | buf = next; |
2373 | } else if (__kmp_match_str(token: "granularity" , buf, |
2374 | CCAST(const char **, &next)) || |
2375 | __kmp_match_str(token: "gran" , buf, CCAST(const char **, &next))) { |
2376 | SKIP_WS(next); |
2377 | if (*next != '=') { |
2378 | EMIT_WARN(TRUE, (AffInvalidParam, name, start)); |
2379 | continue; |
2380 | } |
2381 | next++; // skip '=' |
2382 | SKIP_WS(next); |
2383 | |
2384 | buf = next; |
2385 | |
2386 | // Have to try core_type and core_efficiency matches first since "core" |
2387 | // will register as core granularity with "extra chars" |
2388 | if (__kmp_match_str(token: "core_type" , buf, CCAST(const char **, &next))) { |
2389 | set_gran(KMP_HW_CORE, -1); |
2390 | out_affinity->flags.core_types_gran = 1; |
2391 | buf = next; |
2392 | set = true; |
2393 | } else if (__kmp_match_str(token: "core_efficiency" , buf, |
2394 | CCAST(const char **, &next)) || |
2395 | __kmp_match_str(token: "core_eff" , buf, |
2396 | CCAST(const char **, &next))) { |
2397 | set_gran(KMP_HW_CORE, -1); |
2398 | out_affinity->flags.core_effs_gran = 1; |
2399 | buf = next; |
2400 | set = true; |
2401 | } |
2402 | if (!set) { |
2403 | // Try any hardware topology type for granularity |
2404 | KMP_FOREACH_HW_TYPE(type) { |
2405 | const char *name = __kmp_hw_get_keyword(type); |
2406 | if (__kmp_match_str(token: name, buf, CCAST(const char **, &next))) { |
2407 | set_gran(type, -1); |
2408 | buf = next; |
2409 | set = true; |
2410 | break; |
2411 | } |
2412 | } |
2413 | } |
2414 | if (!set) { |
2415 | // Support older names for different granularity layers |
2416 | if (__kmp_match_str(token: "fine" , buf, CCAST(const char **, &next))) { |
2417 | set_gran(KMP_HW_THREAD, -1); |
2418 | buf = next; |
2419 | set = true; |
2420 | } else if (__kmp_match_str(token: "package" , buf, |
2421 | CCAST(const char **, &next))) { |
2422 | set_gran(KMP_HW_SOCKET, -1); |
2423 | buf = next; |
2424 | set = true; |
2425 | } else if (__kmp_match_str(token: "node" , buf, CCAST(const char **, &next))) { |
2426 | set_gran(KMP_HW_NUMA, -1); |
2427 | buf = next; |
2428 | set = true; |
2429 | #if KMP_GROUP_AFFINITY |
2430 | } else if (__kmp_match_str("group" , buf, CCAST(const char **, &next))) { |
2431 | set_gran(KMP_HW_PROC_GROUP, -1); |
2432 | buf = next; |
2433 | set = true; |
2434 | #endif /* KMP_GROUP AFFINITY */ |
2435 | } else if ((*buf >= '0') && (*buf <= '9')) { |
2436 | int n; |
2437 | next = buf; |
2438 | SKIP_DIGITS(next); |
2439 | n = __kmp_str_to_int(str: buf, sentinel: *next); |
2440 | KMP_ASSERT(n >= 0); |
2441 | buf = next; |
2442 | set_gran(KMP_HW_UNKNOWN, n); |
2443 | set = true; |
2444 | } else { |
2445 | EMIT_WARN(TRUE, (AffInvalidParam, name, start)); |
2446 | continue; |
2447 | } |
2448 | } |
2449 | } else if (__kmp_match_str(token: "proclist" , buf, CCAST(const char **, &next))) { |
2450 | char *temp_proclist; |
2451 | |
2452 | SKIP_WS(next); |
2453 | if (*next != '=') { |
2454 | EMIT_WARN(TRUE, (AffInvalidParam, name, start)); |
2455 | continue; |
2456 | } |
2457 | next++; // skip '=' |
2458 | SKIP_WS(next); |
2459 | if (*next != '[') { |
2460 | EMIT_WARN(TRUE, (AffInvalidParam, name, start)); |
2461 | continue; |
2462 | } |
2463 | next++; // skip '[' |
2464 | buf = next; |
2465 | if (!__kmp_parse_affinity_proc_id_list( |
2466 | var: name, env: buf, CCAST(const char **, &next), proclist: &temp_proclist)) { |
2467 | // warning already emitted. |
2468 | SKIP_TO(next, ']'); |
2469 | if (*next == ']') |
2470 | next++; |
2471 | SKIP_TO(next, ','); |
2472 | if (*next == ',') |
2473 | next++; |
2474 | buf = next; |
2475 | continue; |
2476 | } |
2477 | if (*next != ']') { |
2478 | EMIT_WARN(TRUE, (AffInvalidParam, name, start)); |
2479 | continue; |
2480 | } |
2481 | next++; // skip ']' |
2482 | set_proclist(temp_proclist); |
2483 | } else if ((*buf >= '0') && (*buf <= '9')) { |
2484 | // Parse integer numbers -- permute and offset. |
2485 | int n; |
2486 | next = buf; |
2487 | SKIP_DIGITS(next); |
2488 | n = __kmp_str_to_int(str: buf, sentinel: *next); |
2489 | KMP_ASSERT(n >= 0); |
2490 | buf = next; |
2491 | if (count < 2) { |
2492 | number[count] = n; |
2493 | } else { |
2494 | KMP_WARNING(AffManyParams, name, start); |
2495 | } |
2496 | ++count; |
2497 | } else { |
2498 | EMIT_WARN(TRUE, (AffInvalidParam, name, start)); |
2499 | continue; |
2500 | } |
2501 | |
2502 | SKIP_WS(next); |
2503 | if (*next == ',') { |
2504 | next++; |
2505 | SKIP_WS(next); |
2506 | } else if (*next != '\0') { |
2507 | const char *temp = next; |
2508 | EMIT_WARN(TRUE, (ParseExtraCharsWarn, name, temp)); |
2509 | continue; |
2510 | } |
2511 | buf = next; |
2512 | } // while |
2513 | |
2514 | #undef EMIT_WARN |
2515 | #undef _set_param |
2516 | #undef set_type |
2517 | #undef set_verbose |
2518 | #undef set_warnings |
2519 | #undef set_respect |
2520 | #undef set_granularity |
2521 | #undef set_reset |
2522 | |
2523 | __kmp_str_free(str: &buffer); |
2524 | |
2525 | if (proclist) { |
2526 | if (!type) { |
2527 | KMP_WARNING(AffProcListNoType, name); |
2528 | out_affinity->type = affinity_explicit; |
2529 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2530 | } else if (out_affinity->type != affinity_explicit) { |
2531 | KMP_WARNING(AffProcListNotExplicit, name); |
2532 | KMP_ASSERT(out_affinity->proclist != NULL); |
2533 | KMP_INTERNAL_FREE(out_affinity->proclist); |
2534 | out_affinity->proclist = NULL; |
2535 | } |
2536 | } |
2537 | switch (out_affinity->type) { |
2538 | case affinity_logical: |
2539 | case affinity_physical: { |
2540 | if (count > 0) { |
2541 | out_affinity->offset = number[0]; |
2542 | } |
2543 | if (count > 1) { |
2544 | KMP_WARNING(AffManyParamsForLogic, name, number[1]); |
2545 | } |
2546 | } break; |
2547 | case affinity_balanced: { |
2548 | if (count > 0) { |
2549 | out_affinity->compact = number[0]; |
2550 | } |
2551 | if (count > 1) { |
2552 | out_affinity->offset = number[1]; |
2553 | } |
2554 | |
2555 | if (__kmp_affinity.gran == KMP_HW_UNKNOWN) { |
2556 | int verbose = out_affinity->flags.verbose; |
2557 | int warnings = out_affinity->flags.warnings; |
2558 | #if KMP_MIC_SUPPORTED |
2559 | if (__kmp_mic_type != non_mic) { |
2560 | if (verbose || warnings) { |
2561 | KMP_WARNING(AffGranUsing, out_affinity->env_var, "fine" ); |
2562 | } |
2563 | out_affinity->gran = KMP_HW_THREAD; |
2564 | } else |
2565 | #endif |
2566 | { |
2567 | if (verbose || warnings) { |
2568 | KMP_WARNING(AffGranUsing, out_affinity->env_var, "core" ); |
2569 | } |
2570 | out_affinity->gran = KMP_HW_CORE; |
2571 | } |
2572 | } |
2573 | } break; |
2574 | case affinity_scatter: |
2575 | case affinity_compact: { |
2576 | if (count > 0) { |
2577 | out_affinity->compact = number[0]; |
2578 | } |
2579 | if (count > 1) { |
2580 | out_affinity->offset = number[1]; |
2581 | } |
2582 | } break; |
2583 | case affinity_explicit: { |
2584 | if (out_affinity->proclist == NULL) { |
2585 | KMP_WARNING(AffNoProcList, name); |
2586 | out_affinity->type = affinity_none; |
2587 | } |
2588 | if (count > 0) { |
2589 | KMP_WARNING(AffNoParam, name, "explicit" ); |
2590 | } |
2591 | } break; |
2592 | case affinity_none: { |
2593 | if (count > 0) { |
2594 | KMP_WARNING(AffNoParam, name, "none" ); |
2595 | } |
2596 | } break; |
2597 | case affinity_disabled: { |
2598 | if (count > 0) { |
2599 | KMP_WARNING(AffNoParam, name, "disabled" ); |
2600 | } |
2601 | } break; |
2602 | case affinity_default: { |
2603 | if (count > 0) { |
2604 | KMP_WARNING(AffNoParam, name, "default" ); |
2605 | } |
2606 | } break; |
2607 | default: { |
2608 | KMP_ASSERT(0); |
2609 | } |
2610 | } |
2611 | } // __kmp_parse_affinity_env |
2612 | |
2613 | static void __kmp_stg_parse_affinity(char const *name, char const *value, |
2614 | void *data) { |
2615 | kmp_setting_t **rivals = (kmp_setting_t **)data; |
2616 | int rc; |
2617 | |
2618 | rc = __kmp_stg_check_rivals(name, value, rivals); |
2619 | if (rc) { |
2620 | return; |
2621 | } |
2622 | |
2623 | __kmp_parse_affinity_env(name, value, out_affinity: &__kmp_affinity); |
2624 | |
2625 | } // __kmp_stg_parse_affinity |
2626 | static void __kmp_stg_parse_hh_affinity(char const *name, char const *value, |
2627 | void *data) { |
2628 | __kmp_parse_affinity_env(name, value, out_affinity: &__kmp_hh_affinity); |
2629 | // Warn about unused parts of hidden helper affinity settings if specified. |
2630 | if (__kmp_hh_affinity.flags.reset) { |
2631 | KMP_WARNING(AffInvalidParam, name, "reset" ); |
2632 | } |
2633 | if (__kmp_hh_affinity.flags.respect != affinity_respect_mask_default) { |
2634 | KMP_WARNING(AffInvalidParam, name, "respect" ); |
2635 | } |
2636 | } |
2637 | |
2638 | static void __kmp_print_affinity_env(kmp_str_buf_t *buffer, char const *name, |
2639 | const kmp_affinity_t &affinity) { |
2640 | bool is_hh_affinity = (&affinity == &__kmp_hh_affinity); |
2641 | if (__kmp_env_format) { |
2642 | KMP_STR_BUF_PRINT_NAME_EX(name); |
2643 | } else { |
2644 | __kmp_str_buf_print(buffer, format: " %s='" , name); |
2645 | } |
2646 | if (affinity.flags.verbose) { |
2647 | __kmp_str_buf_print(buffer, format: "%s," , "verbose" ); |
2648 | } else { |
2649 | __kmp_str_buf_print(buffer, format: "%s," , "noverbose" ); |
2650 | } |
2651 | if (affinity.flags.warnings) { |
2652 | __kmp_str_buf_print(buffer, format: "%s," , "warnings" ); |
2653 | } else { |
2654 | __kmp_str_buf_print(buffer, format: "%s," , "nowarnings" ); |
2655 | } |
2656 | if (KMP_AFFINITY_CAPABLE()) { |
2657 | // Hidden helper affinity does not affect global reset |
2658 | // or respect flags. That is still solely controlled by KMP_AFFINITY. |
2659 | if (!is_hh_affinity) { |
2660 | if (affinity.flags.respect) { |
2661 | __kmp_str_buf_print(buffer, format: "%s," , "respect" ); |
2662 | } else { |
2663 | __kmp_str_buf_print(buffer, format: "%s," , "norespect" ); |
2664 | } |
2665 | if (affinity.flags.reset) { |
2666 | __kmp_str_buf_print(buffer, format: "%s," , "reset" ); |
2667 | } else { |
2668 | __kmp_str_buf_print(buffer, format: "%s," , "noreset" ); |
2669 | } |
2670 | } |
2671 | __kmp_str_buf_print(buffer, format: "granularity=" ); |
2672 | if (affinity.flags.core_types_gran) |
2673 | __kmp_str_buf_print(buffer, format: "core_type," ); |
2674 | else if (affinity.flags.core_effs_gran) { |
2675 | __kmp_str_buf_print(buffer, format: "core_eff," ); |
2676 | } else { |
2677 | __kmp_str_buf_print( |
2678 | buffer, format: "%s," , __kmp_hw_get_keyword(type: affinity.gran, /*plural=*/false)); |
2679 | } |
2680 | } |
2681 | if (!KMP_AFFINITY_CAPABLE()) { |
2682 | __kmp_str_buf_print(buffer, format: "%s" , "disabled" ); |
2683 | } else { |
2684 | int compact = affinity.compact; |
2685 | int offset = affinity.offset; |
2686 | switch (affinity.type) { |
2687 | case affinity_none: |
2688 | __kmp_str_buf_print(buffer, format: "%s" , "none" ); |
2689 | break; |
2690 | case affinity_physical: |
2691 | __kmp_str_buf_print(buffer, format: "%s,%d" , "physical" , offset); |
2692 | break; |
2693 | case affinity_logical: |
2694 | __kmp_str_buf_print(buffer, format: "%s,%d" , "logical" , offset); |
2695 | break; |
2696 | case affinity_compact: |
2697 | __kmp_str_buf_print(buffer, format: "%s,%d,%d" , "compact" , compact, offset); |
2698 | break; |
2699 | case affinity_scatter: |
2700 | __kmp_str_buf_print(buffer, format: "%s,%d,%d" , "scatter" , compact, offset); |
2701 | break; |
2702 | case affinity_explicit: |
2703 | __kmp_str_buf_print(buffer, format: "%s=[%s],%s" , "proclist" , affinity.proclist, |
2704 | "explicit" ); |
2705 | break; |
2706 | case affinity_balanced: |
2707 | __kmp_str_buf_print(buffer, format: "%s,%d,%d" , "balanced" , compact, offset); |
2708 | break; |
2709 | case affinity_disabled: |
2710 | __kmp_str_buf_print(buffer, format: "%s" , "disabled" ); |
2711 | break; |
2712 | case affinity_default: |
2713 | __kmp_str_buf_print(buffer, format: "%s" , "default" ); |
2714 | break; |
2715 | default: |
2716 | __kmp_str_buf_print(buffer, format: "%s" , "<unknown>" ); |
2717 | break; |
2718 | } |
2719 | } |
2720 | __kmp_str_buf_print(buffer, format: "'\n" ); |
2721 | } //__kmp_stg_print_affinity |
2722 | |
2723 | static void __kmp_stg_print_affinity(kmp_str_buf_t *buffer, char const *name, |
2724 | void *data) { |
2725 | __kmp_print_affinity_env(buffer, name, affinity: __kmp_affinity); |
2726 | } |
2727 | static void __kmp_stg_print_hh_affinity(kmp_str_buf_t *buffer, char const *name, |
2728 | void *data) { |
2729 | __kmp_print_affinity_env(buffer, name, affinity: __kmp_hh_affinity); |
2730 | } |
2731 | |
2732 | #ifdef KMP_GOMP_COMPAT |
2733 | |
2734 | static void __kmp_stg_parse_gomp_cpu_affinity(char const *name, |
2735 | char const *value, void *data) { |
2736 | const char *next = NULL; |
2737 | char *temp_proclist; |
2738 | kmp_setting_t **rivals = (kmp_setting_t **)data; |
2739 | int rc; |
2740 | |
2741 | rc = __kmp_stg_check_rivals(name, value, rivals); |
2742 | if (rc) { |
2743 | return; |
2744 | } |
2745 | |
2746 | if (TCR_4(__kmp_init_middle)) { |
2747 | KMP_WARNING(EnvMiddleWarn, name); |
2748 | __kmp_env_toPrint(name, flag: 0); |
2749 | return; |
2750 | } |
2751 | |
2752 | __kmp_env_toPrint(name, flag: 1); |
2753 | |
2754 | if (__kmp_parse_affinity_proc_id_list(var: name, env: value, nextEnv: &next, proclist: &temp_proclist)) { |
2755 | SKIP_WS(next); |
2756 | if (*next == '\0') { |
2757 | // GOMP_CPU_AFFINITY => granularity=fine,explicit,proclist=... |
2758 | __kmp_affinity.proclist = temp_proclist; |
2759 | __kmp_affinity.type = affinity_explicit; |
2760 | __kmp_affinity.gran = KMP_HW_THREAD; |
2761 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
2762 | } else { |
2763 | KMP_WARNING(AffSyntaxError, name); |
2764 | if (temp_proclist != NULL) { |
2765 | KMP_INTERNAL_FREE((void *)temp_proclist); |
2766 | } |
2767 | } |
2768 | } else { |
2769 | // Warning already emitted |
2770 | __kmp_affinity.type = affinity_none; |
2771 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
2772 | } |
2773 | } // __kmp_stg_parse_gomp_cpu_affinity |
2774 | |
2775 | #endif /* KMP_GOMP_COMPAT */ |
2776 | |
2777 | /*----------------------------------------------------------------------------- |
2778 | The OMP_PLACES proc id list parser. Here is the grammar: |
2779 | |
2780 | place_list := place |
2781 | place_list := place , place_list |
2782 | place := num |
2783 | place := place : num |
2784 | place := place : num : signed |
2785 | place := { subplacelist } |
2786 | place := ! place // (lowest priority) |
2787 | subplace_list := subplace |
2788 | subplace_list := subplace , subplace_list |
2789 | subplace := num |
2790 | subplace := num : num |
2791 | subplace := num : num : signed |
2792 | signed := num |
2793 | signed := + signed |
2794 | signed := - signed |
2795 | -----------------------------------------------------------------------------*/ |
2796 | |
2797 | // Return TRUE if successful parse, FALSE otherwise |
2798 | static int __kmp_parse_subplace_list(const char *var, const char **scan) { |
2799 | const char *next; |
2800 | |
2801 | for (;;) { |
2802 | int start, count, stride; |
2803 | |
2804 | // |
2805 | // Read in the starting proc id |
2806 | // |
2807 | SKIP_WS(*scan); |
2808 | if ((**scan < '0') || (**scan > '9')) { |
2809 | return FALSE; |
2810 | } |
2811 | next = *scan; |
2812 | SKIP_DIGITS(next); |
2813 | start = __kmp_str_to_int(str: *scan, sentinel: *next); |
2814 | KMP_ASSERT(start >= 0); |
2815 | *scan = next; |
2816 | |
2817 | // valid follow sets are ',' ':' and '}' |
2818 | SKIP_WS(*scan); |
2819 | if (**scan == '}') { |
2820 | break; |
2821 | } |
2822 | if (**scan == ',') { |
2823 | (*scan)++; // skip ',' |
2824 | continue; |
2825 | } |
2826 | if (**scan != ':') { |
2827 | return FALSE; |
2828 | } |
2829 | (*scan)++; // skip ':' |
2830 | |
2831 | // Read count parameter |
2832 | SKIP_WS(*scan); |
2833 | if ((**scan < '0') || (**scan > '9')) { |
2834 | return FALSE; |
2835 | } |
2836 | next = *scan; |
2837 | SKIP_DIGITS(next); |
2838 | count = __kmp_str_to_int(str: *scan, sentinel: *next); |
2839 | KMP_ASSERT(count >= 0); |
2840 | *scan = next; |
2841 | |
2842 | // valid follow sets are ',' ':' and '}' |
2843 | SKIP_WS(*scan); |
2844 | if (**scan == '}') { |
2845 | break; |
2846 | } |
2847 | if (**scan == ',') { |
2848 | (*scan)++; // skip ',' |
2849 | continue; |
2850 | } |
2851 | if (**scan != ':') { |
2852 | return FALSE; |
2853 | } |
2854 | (*scan)++; // skip ':' |
2855 | |
2856 | // Read stride parameter |
2857 | int sign = +1; |
2858 | for (;;) { |
2859 | SKIP_WS(*scan); |
2860 | if (**scan == '+') { |
2861 | (*scan)++; // skip '+' |
2862 | continue; |
2863 | } |
2864 | if (**scan == '-') { |
2865 | sign *= -1; |
2866 | (*scan)++; // skip '-' |
2867 | continue; |
2868 | } |
2869 | break; |
2870 | } |
2871 | SKIP_WS(*scan); |
2872 | if ((**scan < '0') || (**scan > '9')) { |
2873 | return FALSE; |
2874 | } |
2875 | next = *scan; |
2876 | SKIP_DIGITS(next); |
2877 | stride = __kmp_str_to_int(str: *scan, sentinel: *next); |
2878 | KMP_ASSERT(stride >= 0); |
2879 | *scan = next; |
2880 | stride *= sign; |
2881 | |
2882 | // valid follow sets are ',' and '}' |
2883 | SKIP_WS(*scan); |
2884 | if (**scan == '}') { |
2885 | break; |
2886 | } |
2887 | if (**scan == ',') { |
2888 | (*scan)++; // skip ',' |
2889 | continue; |
2890 | } |
2891 | return FALSE; |
2892 | } |
2893 | return TRUE; |
2894 | } |
2895 | |
2896 | // Return TRUE if successful parse, FALSE otherwise |
2897 | static int __kmp_parse_place(const char *var, const char **scan) { |
2898 | const char *next; |
2899 | |
2900 | // valid follow sets are '{' '!' and num |
2901 | SKIP_WS(*scan); |
2902 | if (**scan == '{') { |
2903 | (*scan)++; // skip '{' |
2904 | if (!__kmp_parse_subplace_list(var, scan)) { |
2905 | return FALSE; |
2906 | } |
2907 | if (**scan != '}') { |
2908 | return FALSE; |
2909 | } |
2910 | (*scan)++; // skip '}' |
2911 | } else if (**scan == '!') { |
2912 | (*scan)++; // skip '!' |
2913 | return __kmp_parse_place(var, scan); //'!' has lower precedence than ':' |
2914 | } else if ((**scan >= '0') && (**scan <= '9')) { |
2915 | next = *scan; |
2916 | SKIP_DIGITS(next); |
2917 | int proc = __kmp_str_to_int(str: *scan, sentinel: *next); |
2918 | KMP_ASSERT(proc >= 0); |
2919 | *scan = next; |
2920 | } else { |
2921 | return FALSE; |
2922 | } |
2923 | return TRUE; |
2924 | } |
2925 | |
2926 | // Return TRUE if successful parse, FALSE otherwise |
2927 | static int __kmp_parse_place_list(const char *var, const char *env, |
2928 | char **place_list) { |
2929 | const char *scan = env; |
2930 | const char *next = scan; |
2931 | |
2932 | for (;;) { |
2933 | int count, stride; |
2934 | |
2935 | if (!__kmp_parse_place(var, scan: &scan)) { |
2936 | return FALSE; |
2937 | } |
2938 | |
2939 | // valid follow sets are ',' ':' and EOL |
2940 | SKIP_WS(scan); |
2941 | if (*scan == '\0') { |
2942 | break; |
2943 | } |
2944 | if (*scan == ',') { |
2945 | scan++; // skip ',' |
2946 | continue; |
2947 | } |
2948 | if (*scan != ':') { |
2949 | return FALSE; |
2950 | } |
2951 | scan++; // skip ':' |
2952 | |
2953 | // Read count parameter |
2954 | SKIP_WS(scan); |
2955 | if ((*scan < '0') || (*scan > '9')) { |
2956 | return FALSE; |
2957 | } |
2958 | next = scan; |
2959 | SKIP_DIGITS(next); |
2960 | count = __kmp_str_to_int(str: scan, sentinel: *next); |
2961 | KMP_ASSERT(count >= 0); |
2962 | scan = next; |
2963 | |
2964 | // valid follow sets are ',' ':' and EOL |
2965 | SKIP_WS(scan); |
2966 | if (*scan == '\0') { |
2967 | break; |
2968 | } |
2969 | if (*scan == ',') { |
2970 | scan++; // skip ',' |
2971 | continue; |
2972 | } |
2973 | if (*scan != ':') { |
2974 | return FALSE; |
2975 | } |
2976 | scan++; // skip ':' |
2977 | |
2978 | // Read stride parameter |
2979 | int sign = +1; |
2980 | for (;;) { |
2981 | SKIP_WS(scan); |
2982 | if (*scan == '+') { |
2983 | scan++; // skip '+' |
2984 | continue; |
2985 | } |
2986 | if (*scan == '-') { |
2987 | sign *= -1; |
2988 | scan++; // skip '-' |
2989 | continue; |
2990 | } |
2991 | break; |
2992 | } |
2993 | SKIP_WS(scan); |
2994 | if ((*scan < '0') || (*scan > '9')) { |
2995 | return FALSE; |
2996 | } |
2997 | next = scan; |
2998 | SKIP_DIGITS(next); |
2999 | stride = __kmp_str_to_int(str: scan, sentinel: *next); |
3000 | KMP_ASSERT(stride >= 0); |
3001 | scan = next; |
3002 | stride *= sign; |
3003 | |
3004 | // valid follow sets are ',' and EOL |
3005 | SKIP_WS(scan); |
3006 | if (*scan == '\0') { |
3007 | break; |
3008 | } |
3009 | if (*scan == ',') { |
3010 | scan++; // skip ',' |
3011 | continue; |
3012 | } |
3013 | |
3014 | return FALSE; |
3015 | } |
3016 | |
3017 | { |
3018 | ptrdiff_t len = scan - env; |
3019 | char *retlist = (char *)__kmp_allocate((len + 1) * sizeof(char)); |
3020 | KMP_MEMCPY_S(retlist, (len + 1) * sizeof(char), env, len * sizeof(char)); |
3021 | retlist[len] = '\0'; |
3022 | *place_list = retlist; |
3023 | } |
3024 | return TRUE; |
3025 | } |
3026 | |
3027 | static inline void __kmp_places_set(enum affinity_type type, kmp_hw_t kind) { |
3028 | __kmp_affinity.type = type; |
3029 | __kmp_affinity.gran = kind; |
3030 | __kmp_affinity.flags.dups = FALSE; |
3031 | __kmp_affinity.flags.omp_places = TRUE; |
3032 | } |
3033 | |
3034 | static void __kmp_places_syntax_error_fallback(char const *name, |
3035 | kmp_hw_t kind) { |
3036 | const char *str = __kmp_hw_get_catalog_string(type: kind, /*plural=*/true); |
3037 | KMP_WARNING(SyntaxErrorUsing, name, str); |
3038 | __kmp_places_set(type: affinity_compact, kind); |
3039 | if (__kmp_nested_proc_bind.bind_types[0] == proc_bind_default) |
3040 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_true; |
3041 | } |
3042 | |
3043 | static void __kmp_stg_parse_places(char const *name, char const *value, |
3044 | void *data) { |
3045 | struct kmp_place_t { |
3046 | const char *name; |
3047 | kmp_hw_t type; |
3048 | }; |
3049 | int count; |
3050 | bool set = false; |
3051 | const char *scan = value; |
3052 | const char *next = scan; |
3053 | kmp_place_t std_places[] = {{.name: "threads" , .type: KMP_HW_THREAD}, |
3054 | {.name: "cores" , .type: KMP_HW_CORE}, |
3055 | {.name: "numa_domains" , .type: KMP_HW_NUMA}, |
3056 | {.name: "ll_caches" , .type: KMP_HW_LLC}, |
3057 | {.name: "sockets" , .type: KMP_HW_SOCKET}}; |
3058 | kmp_setting_t **rivals = (kmp_setting_t **)data; |
3059 | int rc; |
3060 | |
3061 | rc = __kmp_stg_check_rivals(name, value, rivals); |
3062 | if (rc) { |
3063 | return; |
3064 | } |
3065 | |
3066 | // Standard choices |
3067 | for (size_t i = 0; i < sizeof(std_places) / sizeof(std_places[0]); ++i) { |
3068 | const kmp_place_t &place = std_places[i]; |
3069 | if (__kmp_match_str(token: place.name, buf: scan, end: &next)) { |
3070 | scan = next; |
3071 | __kmp_places_set(type: affinity_compact, kind: place.type); |
3072 | set = true; |
3073 | // Parse core attribute if it exists |
3074 | if (KMP_HW_MAX_NUM_CORE_TYPES > 1) { |
3075 | SKIP_WS(scan); |
3076 | if (*scan == ':') { |
3077 | if (place.type != KMP_HW_CORE) { |
3078 | __kmp_places_syntax_error_fallback(name, kind: place.type); |
3079 | return; |
3080 | } |
3081 | scan++; // skip ':' |
3082 | SKIP_WS(scan); |
3083 | #if KMP_ARCH_X86 || KMP_ARCH_X86_64 |
3084 | if (__kmp_match_str(token: "intel_core" , buf: scan, end: &next)) { |
3085 | __kmp_affinity.core_attr_gran.core_type = KMP_HW_CORE_TYPE_CORE; |
3086 | __kmp_affinity.core_attr_gran.valid = 1; |
3087 | scan = next; |
3088 | } else if (__kmp_match_str(token: "intel_atom" , buf: scan, end: &next)) { |
3089 | __kmp_affinity.core_attr_gran.core_type = KMP_HW_CORE_TYPE_ATOM; |
3090 | __kmp_affinity.core_attr_gran.valid = 1; |
3091 | scan = next; |
3092 | } else |
3093 | #endif |
3094 | if (__kmp_match_str(token: "eff" , buf: scan, end: &next)) { |
3095 | int eff; |
3096 | if (!isdigit(*next)) { |
3097 | __kmp_places_syntax_error_fallback(name, kind: place.type); |
3098 | return; |
3099 | } |
3100 | scan = next; |
3101 | SKIP_DIGITS(next); |
3102 | eff = __kmp_str_to_int(str: scan, sentinel: *next); |
3103 | if (eff < 0) { |
3104 | __kmp_places_syntax_error_fallback(name, kind: place.type); |
3105 | return; |
3106 | } |
3107 | if (eff >= KMP_HW_MAX_NUM_CORE_EFFS) |
3108 | eff = KMP_HW_MAX_NUM_CORE_EFFS - 1; |
3109 | __kmp_affinity.core_attr_gran.core_eff = eff; |
3110 | __kmp_affinity.core_attr_gran.valid = 1; |
3111 | scan = next; |
3112 | } |
3113 | if (!__kmp_affinity.core_attr_gran.valid) { |
3114 | __kmp_places_syntax_error_fallback(name, kind: place.type); |
3115 | return; |
3116 | } |
3117 | } |
3118 | } |
3119 | break; |
3120 | } |
3121 | } |
3122 | // Implementation choices for OMP_PLACES based on internal types |
3123 | if (!set) { |
3124 | KMP_FOREACH_HW_TYPE(type) { |
3125 | const char *name = __kmp_hw_get_keyword(type, plural: true); |
3126 | if (__kmp_match_str(token: "unknowns" , buf: scan, end: &next)) |
3127 | continue; |
3128 | if (__kmp_match_str(token: name, buf: scan, end: &next)) { |
3129 | scan = next; |
3130 | __kmp_places_set(type: affinity_compact, kind: type); |
3131 | set = true; |
3132 | break; |
3133 | } |
3134 | } |
3135 | } |
3136 | // Implementation choices for OMP_PLACES based on core attributes |
3137 | if (!set) { |
3138 | if (__kmp_match_str(token: "core_types" , buf: scan, end: &next)) { |
3139 | scan = next; |
3140 | if (*scan != '\0') { |
3141 | KMP_WARNING(ParseExtraCharsWarn, name, scan); |
3142 | } |
3143 | __kmp_places_set(type: affinity_compact, kind: KMP_HW_CORE); |
3144 | __kmp_affinity.flags.core_types_gran = 1; |
3145 | set = true; |
3146 | } else if (__kmp_match_str(token: "core_effs" , buf: scan, end: &next) || |
3147 | __kmp_match_str(token: "core_efficiencies" , buf: scan, end: &next)) { |
3148 | scan = next; |
3149 | if (*scan != '\0') { |
3150 | KMP_WARNING(ParseExtraCharsWarn, name, scan); |
3151 | } |
3152 | __kmp_places_set(type: affinity_compact, kind: KMP_HW_CORE); |
3153 | __kmp_affinity.flags.core_effs_gran = 1; |
3154 | set = true; |
3155 | } |
3156 | } |
3157 | // Explicit place list |
3158 | if (!set) { |
3159 | if (__kmp_affinity.proclist != NULL) { |
3160 | KMP_INTERNAL_FREE((void *)__kmp_affinity.proclist); |
3161 | __kmp_affinity.proclist = NULL; |
3162 | } |
3163 | if (__kmp_parse_place_list(var: name, env: value, place_list: &__kmp_affinity.proclist)) { |
3164 | __kmp_places_set(type: affinity_explicit, kind: KMP_HW_THREAD); |
3165 | } else { |
3166 | // Syntax error fallback |
3167 | __kmp_places_syntax_error_fallback(name, kind: KMP_HW_CORE); |
3168 | } |
3169 | if (__kmp_nested_proc_bind.bind_types[0] == proc_bind_default) { |
3170 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_true; |
3171 | } |
3172 | return; |
3173 | } |
3174 | |
3175 | kmp_hw_t gran = __kmp_affinity.gran; |
3176 | if (__kmp_affinity.gran != KMP_HW_UNKNOWN) { |
3177 | gran = __kmp_affinity.gran; |
3178 | } else { |
3179 | gran = KMP_HW_CORE; |
3180 | } |
3181 | |
3182 | if (__kmp_nested_proc_bind.bind_types[0] == proc_bind_default) { |
3183 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_true; |
3184 | } |
3185 | |
3186 | SKIP_WS(scan); |
3187 | if (*scan == '\0') { |
3188 | return; |
3189 | } |
3190 | |
3191 | // Parse option count parameter in parentheses |
3192 | if (*scan != '(') { |
3193 | __kmp_places_syntax_error_fallback(name, kind: gran); |
3194 | return; |
3195 | } |
3196 | scan++; // skip '(' |
3197 | |
3198 | SKIP_WS(scan); |
3199 | next = scan; |
3200 | SKIP_DIGITS(next); |
3201 | count = __kmp_str_to_int(str: scan, sentinel: *next); |
3202 | KMP_ASSERT(count >= 0); |
3203 | scan = next; |
3204 | |
3205 | SKIP_WS(scan); |
3206 | if (*scan != ')') { |
3207 | __kmp_places_syntax_error_fallback(name, kind: gran); |
3208 | return; |
3209 | } |
3210 | scan++; // skip ')' |
3211 | |
3212 | SKIP_WS(scan); |
3213 | if (*scan != '\0') { |
3214 | KMP_WARNING(ParseExtraCharsWarn, name, scan); |
3215 | } |
3216 | __kmp_affinity_num_places = count; |
3217 | } |
3218 | |
3219 | static void __kmp_stg_print_places(kmp_str_buf_t *buffer, char const *name, |
3220 | void *data) { |
3221 | enum affinity_type type = __kmp_affinity.type; |
3222 | const char *proclist = __kmp_affinity.proclist; |
3223 | kmp_hw_t gran = __kmp_affinity.gran; |
3224 | |
3225 | if (__kmp_env_format) { |
3226 | KMP_STR_BUF_PRINT_NAME; |
3227 | } else { |
3228 | __kmp_str_buf_print(buffer, format: " %s" , name); |
3229 | } |
3230 | if ((__kmp_nested_proc_bind.used == 0) || |
3231 | (__kmp_nested_proc_bind.bind_types == NULL) || |
3232 | (__kmp_nested_proc_bind.bind_types[0] == proc_bind_false)) { |
3233 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
3234 | } else if (type == affinity_explicit) { |
3235 | if (proclist != NULL) { |
3236 | __kmp_str_buf_print(buffer, format: "='%s'\n" , proclist); |
3237 | } else { |
3238 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
3239 | } |
3240 | } else if (type == affinity_compact) { |
3241 | int num; |
3242 | if (__kmp_affinity.num_masks > 0) { |
3243 | num = __kmp_affinity.num_masks; |
3244 | } else if (__kmp_affinity_num_places > 0) { |
3245 | num = __kmp_affinity_num_places; |
3246 | } else { |
3247 | num = 0; |
3248 | } |
3249 | if (gran != KMP_HW_UNKNOWN) { |
3250 | // If core_types or core_effs, just print and return |
3251 | if (__kmp_affinity.flags.core_types_gran) { |
3252 | __kmp_str_buf_print(buffer, format: "='%s'\n" , "core_types" ); |
3253 | return; |
3254 | } |
3255 | if (__kmp_affinity.flags.core_effs_gran) { |
3256 | __kmp_str_buf_print(buffer, format: "='%s'\n" , "core_effs" ); |
3257 | return; |
3258 | } |
3259 | |
3260 | // threads, cores, sockets, cores:<attribute>, etc. |
3261 | const char *name = __kmp_hw_get_keyword(type: gran, plural: true); |
3262 | __kmp_str_buf_print(buffer, format: "='%s" , name); |
3263 | |
3264 | // Add core attributes if it exists |
3265 | if (__kmp_affinity.core_attr_gran.valid) { |
3266 | kmp_hw_core_type_t ct = |
3267 | (kmp_hw_core_type_t)__kmp_affinity.core_attr_gran.core_type; |
3268 | int eff = __kmp_affinity.core_attr_gran.core_eff; |
3269 | if (ct != KMP_HW_CORE_TYPE_UNKNOWN) { |
3270 | const char *ct_name = __kmp_hw_get_core_type_keyword(type: ct); |
3271 | __kmp_str_buf_print(buffer, format: ":%s" , name, ct_name); |
3272 | } else if (eff >= 0 && eff < KMP_HW_MAX_NUM_CORE_EFFS) { |
3273 | __kmp_str_buf_print(buffer, format: ":eff%d" , name, eff); |
3274 | } |
3275 | } |
3276 | |
3277 | // Add the '(#)' part if it exists |
3278 | if (num > 0) |
3279 | __kmp_str_buf_print(buffer, format: "(%d)" , num); |
3280 | __kmp_str_buf_print(buffer, format: "'\n" ); |
3281 | } else { |
3282 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
3283 | } |
3284 | } else { |
3285 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
3286 | } |
3287 | } |
3288 | |
3289 | static void __kmp_stg_parse_topology_method(char const *name, char const *value, |
3290 | void *data) { |
3291 | if (__kmp_str_match(target: "all" , len: 1, data: value)) { |
3292 | __kmp_affinity_top_method = affinity_top_method_all; |
3293 | } |
3294 | #if KMP_USE_HWLOC |
3295 | else if (__kmp_str_match("hwloc" , 1, value)) { |
3296 | __kmp_affinity_top_method = affinity_top_method_hwloc; |
3297 | } |
3298 | #endif |
3299 | #if KMP_ARCH_X86 || KMP_ARCH_X86_64 |
3300 | else if (__kmp_str_match(target: "cpuid_leaf31" , len: 12, data: value) || |
3301 | __kmp_str_match(target: "cpuid 1f" , len: 8, data: value) || |
3302 | __kmp_str_match(target: "cpuid 31" , len: 8, data: value) || |
3303 | __kmp_str_match(target: "cpuid1f" , len: 7, data: value) || |
3304 | __kmp_str_match(target: "cpuid31" , len: 7, data: value) || |
3305 | __kmp_str_match(target: "leaf 1f" , len: 7, data: value) || |
3306 | __kmp_str_match(target: "leaf 31" , len: 7, data: value) || |
3307 | __kmp_str_match(target: "leaf1f" , len: 6, data: value) || |
3308 | __kmp_str_match(target: "leaf31" , len: 6, data: value)) { |
3309 | __kmp_affinity_top_method = affinity_top_method_x2apicid_1f; |
3310 | } else if (__kmp_str_match(target: "x2apic id" , len: 9, data: value) || |
3311 | __kmp_str_match(target: "x2apic_id" , len: 9, data: value) || |
3312 | __kmp_str_match(target: "x2apic-id" , len: 9, data: value) || |
3313 | __kmp_str_match(target: "x2apicid" , len: 8, data: value) || |
3314 | __kmp_str_match(target: "cpuid leaf 11" , len: 13, data: value) || |
3315 | __kmp_str_match(target: "cpuid_leaf_11" , len: 13, data: value) || |
3316 | __kmp_str_match(target: "cpuid-leaf-11" , len: 13, data: value) || |
3317 | __kmp_str_match(target: "cpuid leaf11" , len: 12, data: value) || |
3318 | __kmp_str_match(target: "cpuid_leaf11" , len: 12, data: value) || |
3319 | __kmp_str_match(target: "cpuid-leaf11" , len: 12, data: value) || |
3320 | __kmp_str_match(target: "cpuidleaf 11" , len: 12, data: value) || |
3321 | __kmp_str_match(target: "cpuidleaf_11" , len: 12, data: value) || |
3322 | __kmp_str_match(target: "cpuidleaf-11" , len: 12, data: value) || |
3323 | __kmp_str_match(target: "cpuidleaf11" , len: 11, data: value) || |
3324 | __kmp_str_match(target: "cpuid 11" , len: 8, data: value) || |
3325 | __kmp_str_match(target: "cpuid_11" , len: 8, data: value) || |
3326 | __kmp_str_match(target: "cpuid-11" , len: 8, data: value) || |
3327 | __kmp_str_match(target: "cpuid11" , len: 7, data: value) || |
3328 | __kmp_str_match(target: "leaf 11" , len: 7, data: value) || |
3329 | __kmp_str_match(target: "leaf_11" , len: 7, data: value) || |
3330 | __kmp_str_match(target: "leaf-11" , len: 7, data: value) || |
3331 | __kmp_str_match(target: "leaf11" , len: 6, data: value)) { |
3332 | __kmp_affinity_top_method = affinity_top_method_x2apicid; |
3333 | } else if (__kmp_str_match(target: "apic id" , len: 7, data: value) || |
3334 | __kmp_str_match(target: "apic_id" , len: 7, data: value) || |
3335 | __kmp_str_match(target: "apic-id" , len: 7, data: value) || |
3336 | __kmp_str_match(target: "apicid" , len: 6, data: value) || |
3337 | __kmp_str_match(target: "cpuid leaf 4" , len: 12, data: value) || |
3338 | __kmp_str_match(target: "cpuid_leaf_4" , len: 12, data: value) || |
3339 | __kmp_str_match(target: "cpuid-leaf-4" , len: 12, data: value) || |
3340 | __kmp_str_match(target: "cpuid leaf4" , len: 11, data: value) || |
3341 | __kmp_str_match(target: "cpuid_leaf4" , len: 11, data: value) || |
3342 | __kmp_str_match(target: "cpuid-leaf4" , len: 11, data: value) || |
3343 | __kmp_str_match(target: "cpuidleaf 4" , len: 11, data: value) || |
3344 | __kmp_str_match(target: "cpuidleaf_4" , len: 11, data: value) || |
3345 | __kmp_str_match(target: "cpuidleaf-4" , len: 11, data: value) || |
3346 | __kmp_str_match(target: "cpuidleaf4" , len: 10, data: value) || |
3347 | __kmp_str_match(target: "cpuid 4" , len: 7, data: value) || |
3348 | __kmp_str_match(target: "cpuid_4" , len: 7, data: value) || |
3349 | __kmp_str_match(target: "cpuid-4" , len: 7, data: value) || |
3350 | __kmp_str_match(target: "cpuid4" , len: 6, data: value) || |
3351 | __kmp_str_match(target: "leaf 4" , len: 6, data: value) || |
3352 | __kmp_str_match(target: "leaf_4" , len: 6, data: value) || |
3353 | __kmp_str_match(target: "leaf-4" , len: 6, data: value) || |
3354 | __kmp_str_match(target: "leaf4" , len: 5, data: value)) { |
3355 | __kmp_affinity_top_method = affinity_top_method_apicid; |
3356 | } |
3357 | #endif /* KMP_ARCH_X86 || KMP_ARCH_X86_64 */ |
3358 | else if (__kmp_str_match(target: "/proc/cpuinfo" , len: 2, data: value) || |
3359 | __kmp_str_match(target: "cpuinfo" , len: 5, data: value)) { |
3360 | __kmp_affinity_top_method = affinity_top_method_cpuinfo; |
3361 | } |
3362 | #if KMP_GROUP_AFFINITY |
3363 | else if (__kmp_str_match("group" , 1, value)) { |
3364 | KMP_WARNING(StgDeprecatedValue, name, value, "all" ); |
3365 | __kmp_affinity_top_method = affinity_top_method_group; |
3366 | } |
3367 | #endif /* KMP_GROUP_AFFINITY */ |
3368 | else if (__kmp_str_match(target: "flat" , len: 1, data: value)) { |
3369 | __kmp_affinity_top_method = affinity_top_method_flat; |
3370 | } else { |
3371 | KMP_WARNING(StgInvalidValue, name, value); |
3372 | } |
3373 | } // __kmp_stg_parse_topology_method |
3374 | |
3375 | static void __kmp_stg_print_topology_method(kmp_str_buf_t *buffer, |
3376 | char const *name, void *data) { |
3377 | char const *value = NULL; |
3378 | |
3379 | switch (__kmp_affinity_top_method) { |
3380 | case affinity_top_method_default: |
3381 | value = "default" ; |
3382 | break; |
3383 | |
3384 | case affinity_top_method_all: |
3385 | value = "all" ; |
3386 | break; |
3387 | |
3388 | #if KMP_ARCH_X86 || KMP_ARCH_X86_64 |
3389 | case affinity_top_method_x2apicid_1f: |
3390 | value = "x2APIC id leaf 0x1f" ; |
3391 | break; |
3392 | |
3393 | case affinity_top_method_x2apicid: |
3394 | value = "x2APIC id leaf 0xb" ; |
3395 | break; |
3396 | |
3397 | case affinity_top_method_apicid: |
3398 | value = "APIC id" ; |
3399 | break; |
3400 | #endif /* KMP_ARCH_X86 || KMP_ARCH_X86_64 */ |
3401 | |
3402 | #if KMP_USE_HWLOC |
3403 | case affinity_top_method_hwloc: |
3404 | value = "hwloc" ; |
3405 | break; |
3406 | #endif |
3407 | |
3408 | case affinity_top_method_cpuinfo: |
3409 | value = "cpuinfo" ; |
3410 | break; |
3411 | |
3412 | #if KMP_GROUP_AFFINITY |
3413 | case affinity_top_method_group: |
3414 | value = "group" ; |
3415 | break; |
3416 | #endif /* KMP_GROUP_AFFINITY */ |
3417 | |
3418 | case affinity_top_method_flat: |
3419 | value = "flat" ; |
3420 | break; |
3421 | } |
3422 | |
3423 | if (value != NULL) { |
3424 | __kmp_stg_print_str(buffer, name, value); |
3425 | } |
3426 | } // __kmp_stg_print_topology_method |
3427 | |
3428 | // KMP_TEAMS_PROC_BIND |
3429 | struct kmp_proc_bind_info_t { |
3430 | const char *name; |
3431 | kmp_proc_bind_t proc_bind; |
3432 | }; |
3433 | static kmp_proc_bind_info_t proc_bind_table[] = { |
3434 | {.name: "spread" , .proc_bind: proc_bind_spread}, |
3435 | {.name: "true" , .proc_bind: proc_bind_spread}, |
3436 | {.name: "close" , .proc_bind: proc_bind_close}, |
3437 | // teams-bind = false means "replicate the primary thread's affinity" |
3438 | {.name: "false" , .proc_bind: proc_bind_primary}, |
3439 | {.name: "primary" , .proc_bind: proc_bind_primary}}; |
3440 | static void __kmp_stg_parse_teams_proc_bind(char const *name, char const *value, |
3441 | void *data) { |
3442 | int valid; |
3443 | const char *end; |
3444 | valid = 0; |
3445 | for (size_t i = 0; i < sizeof(proc_bind_table) / sizeof(proc_bind_table[0]); |
3446 | ++i) { |
3447 | if (__kmp_match_str(token: proc_bind_table[i].name, buf: value, end: &end)) { |
3448 | __kmp_teams_proc_bind = proc_bind_table[i].proc_bind; |
3449 | valid = 1; |
3450 | break; |
3451 | } |
3452 | } |
3453 | if (!valid) { |
3454 | KMP_WARNING(StgInvalidValue, name, value); |
3455 | } |
3456 | } |
3457 | static void __kmp_stg_print_teams_proc_bind(kmp_str_buf_t *buffer, |
3458 | char const *name, void *data) { |
3459 | const char *value = KMP_I18N_STR(NotDefined); |
3460 | for (size_t i = 0; i < sizeof(proc_bind_table) / sizeof(proc_bind_table[0]); |
3461 | ++i) { |
3462 | if (__kmp_teams_proc_bind == proc_bind_table[i].proc_bind) { |
3463 | value = proc_bind_table[i].name; |
3464 | break; |
3465 | } |
3466 | } |
3467 | __kmp_stg_print_str(buffer, name, value); |
3468 | } |
3469 | #endif /* KMP_AFFINITY_SUPPORTED */ |
3470 | |
3471 | // OMP_PROC_BIND / bind-var is functional on all 4.0 builds, including OS X* |
3472 | // OMP_PLACES / place-partition-var is not. |
3473 | static void __kmp_stg_parse_proc_bind(char const *name, char const *value, |
3474 | void *data) { |
3475 | kmp_setting_t **rivals = (kmp_setting_t **)data; |
3476 | int rc; |
3477 | |
3478 | rc = __kmp_stg_check_rivals(name, value, rivals); |
3479 | if (rc) { |
3480 | return; |
3481 | } |
3482 | |
3483 | // In OMP 4.0 OMP_PROC_BIND is a vector of proc_bind types. |
3484 | KMP_DEBUG_ASSERT((__kmp_nested_proc_bind.bind_types != NULL) && |
3485 | (__kmp_nested_proc_bind.used > 0)); |
3486 | |
3487 | const char *buf = value; |
3488 | const char *next; |
3489 | int num; |
3490 | SKIP_WS(buf); |
3491 | if ((*buf >= '0') && (*buf <= '9')) { |
3492 | next = buf; |
3493 | SKIP_DIGITS(next); |
3494 | num = __kmp_str_to_int(str: buf, sentinel: *next); |
3495 | KMP_ASSERT(num >= 0); |
3496 | buf = next; |
3497 | SKIP_WS(buf); |
3498 | } else { |
3499 | num = -1; |
3500 | } |
3501 | |
3502 | next = buf; |
3503 | if (__kmp_match_str(token: "disabled" , buf, end: &next)) { |
3504 | buf = next; |
3505 | SKIP_WS(buf); |
3506 | #if KMP_AFFINITY_SUPPORTED |
3507 | __kmp_affinity.type = affinity_disabled; |
3508 | #endif /* KMP_AFFINITY_SUPPORTED */ |
3509 | __kmp_nested_proc_bind.used = 1; |
3510 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
3511 | } else if ((num == (int)proc_bind_false) || |
3512 | __kmp_match_str(token: "false" , buf, end: &next)) { |
3513 | buf = next; |
3514 | SKIP_WS(buf); |
3515 | #if KMP_AFFINITY_SUPPORTED |
3516 | __kmp_affinity.type = affinity_none; |
3517 | #endif /* KMP_AFFINITY_SUPPORTED */ |
3518 | __kmp_nested_proc_bind.used = 1; |
3519 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
3520 | } else if ((num == (int)proc_bind_true) || |
3521 | __kmp_match_str(token: "true" , buf, end: &next)) { |
3522 | buf = next; |
3523 | SKIP_WS(buf); |
3524 | __kmp_nested_proc_bind.used = 1; |
3525 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_true; |
3526 | } else { |
3527 | // Count the number of values in the env var string |
3528 | const char *scan; |
3529 | int nelem = 1; |
3530 | for (scan = buf; *scan != '\0'; scan++) { |
3531 | if (*scan == ',') { |
3532 | nelem++; |
3533 | } |
3534 | } |
3535 | |
3536 | // Create / expand the nested proc_bind array as needed |
3537 | if (__kmp_nested_proc_bind.size < nelem) { |
3538 | __kmp_nested_proc_bind.bind_types = |
3539 | (kmp_proc_bind_t *)KMP_INTERNAL_REALLOC( |
3540 | __kmp_nested_proc_bind.bind_types, |
3541 | sizeof(kmp_proc_bind_t) * nelem); |
3542 | if (__kmp_nested_proc_bind.bind_types == NULL) { |
3543 | KMP_FATAL(MemoryAllocFailed); |
3544 | } |
3545 | __kmp_nested_proc_bind.size = nelem; |
3546 | } |
3547 | __kmp_nested_proc_bind.used = nelem; |
3548 | |
3549 | if (nelem > 1 && !__kmp_dflt_max_active_levels_set) |
3550 | __kmp_dflt_max_active_levels = KMP_MAX_ACTIVE_LEVELS_LIMIT; |
3551 | |
3552 | // Save values in the nested proc_bind array |
3553 | int i = 0; |
3554 | for (;;) { |
3555 | enum kmp_proc_bind_t bind; |
3556 | |
3557 | if ((num == (int)proc_bind_primary) || |
3558 | __kmp_match_str(token: "master" , buf, end: &next) || |
3559 | __kmp_match_str(token: "primary" , buf, end: &next)) { |
3560 | buf = next; |
3561 | SKIP_WS(buf); |
3562 | bind = proc_bind_primary; |
3563 | } else if ((num == (int)proc_bind_close) || |
3564 | __kmp_match_str(token: "close" , buf, end: &next)) { |
3565 | buf = next; |
3566 | SKIP_WS(buf); |
3567 | bind = proc_bind_close; |
3568 | } else if ((num == (int)proc_bind_spread) || |
3569 | __kmp_match_str(token: "spread" , buf, end: &next)) { |
3570 | buf = next; |
3571 | SKIP_WS(buf); |
3572 | bind = proc_bind_spread; |
3573 | } else { |
3574 | KMP_WARNING(StgInvalidValue, name, value); |
3575 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
3576 | __kmp_nested_proc_bind.used = 1; |
3577 | return; |
3578 | } |
3579 | |
3580 | __kmp_nested_proc_bind.bind_types[i++] = bind; |
3581 | if (i >= nelem) { |
3582 | break; |
3583 | } |
3584 | KMP_DEBUG_ASSERT(*buf == ','); |
3585 | buf++; |
3586 | SKIP_WS(buf); |
3587 | |
3588 | // Read next value if it was specified as an integer |
3589 | if ((*buf >= '0') && (*buf <= '9')) { |
3590 | next = buf; |
3591 | SKIP_DIGITS(next); |
3592 | num = __kmp_str_to_int(str: buf, sentinel: *next); |
3593 | KMP_ASSERT(num >= 0); |
3594 | buf = next; |
3595 | SKIP_WS(buf); |
3596 | } else { |
3597 | num = -1; |
3598 | } |
3599 | } |
3600 | SKIP_WS(buf); |
3601 | } |
3602 | if (*buf != '\0') { |
3603 | KMP_WARNING(ParseExtraCharsWarn, name, buf); |
3604 | } |
3605 | } |
3606 | |
3607 | static void __kmp_stg_print_proc_bind(kmp_str_buf_t *buffer, char const *name, |
3608 | void *data) { |
3609 | int nelem = __kmp_nested_proc_bind.used; |
3610 | if (__kmp_env_format) { |
3611 | KMP_STR_BUF_PRINT_NAME; |
3612 | } else { |
3613 | __kmp_str_buf_print(buffer, format: " %s" , name); |
3614 | } |
3615 | if (nelem == 0) { |
3616 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
3617 | } else { |
3618 | int i; |
3619 | __kmp_str_buf_print(buffer, format: "='" , name); |
3620 | for (i = 0; i < nelem; i++) { |
3621 | switch (__kmp_nested_proc_bind.bind_types[i]) { |
3622 | case proc_bind_false: |
3623 | __kmp_str_buf_print(buffer, format: "false" ); |
3624 | break; |
3625 | |
3626 | case proc_bind_true: |
3627 | __kmp_str_buf_print(buffer, format: "true" ); |
3628 | break; |
3629 | |
3630 | case proc_bind_primary: |
3631 | __kmp_str_buf_print(buffer, format: "primary" ); |
3632 | break; |
3633 | |
3634 | case proc_bind_close: |
3635 | __kmp_str_buf_print(buffer, format: "close" ); |
3636 | break; |
3637 | |
3638 | case proc_bind_spread: |
3639 | __kmp_str_buf_print(buffer, format: "spread" ); |
3640 | break; |
3641 | |
3642 | case proc_bind_intel: |
3643 | __kmp_str_buf_print(buffer, format: "intel" ); |
3644 | break; |
3645 | |
3646 | case proc_bind_default: |
3647 | __kmp_str_buf_print(buffer, format: "default" ); |
3648 | break; |
3649 | } |
3650 | if (i < nelem - 1) { |
3651 | __kmp_str_buf_print(buffer, format: "," ); |
3652 | } |
3653 | } |
3654 | __kmp_str_buf_print(buffer, format: "'\n" ); |
3655 | } |
3656 | } |
3657 | |
3658 | static void __kmp_stg_parse_display_affinity(char const *name, |
3659 | char const *value, void *data) { |
3660 | __kmp_stg_parse_bool(name, value, out: &__kmp_display_affinity); |
3661 | } |
3662 | static void __kmp_stg_print_display_affinity(kmp_str_buf_t *buffer, |
3663 | char const *name, void *data) { |
3664 | __kmp_stg_print_bool(buffer, name, value: __kmp_display_affinity); |
3665 | } |
3666 | static void __kmp_stg_parse_affinity_format(char const *name, char const *value, |
3667 | void *data) { |
3668 | size_t length = KMP_STRLEN(s: value); |
3669 | __kmp_strncpy_truncate(buffer: __kmp_affinity_format, buf_size: KMP_AFFINITY_FORMAT_SIZE, src: value, |
3670 | src_size: length); |
3671 | } |
3672 | static void __kmp_stg_print_affinity_format(kmp_str_buf_t *buffer, |
3673 | char const *name, void *data) { |
3674 | if (__kmp_env_format) { |
3675 | KMP_STR_BUF_PRINT_NAME_EX(name); |
3676 | } else { |
3677 | __kmp_str_buf_print(buffer, format: " %s='" , name); |
3678 | } |
3679 | __kmp_str_buf_print(buffer, format: "%s'\n" , __kmp_affinity_format); |
3680 | } |
3681 | |
3682 | /*----------------------------------------------------------------------------- |
3683 | OMP_ALLOCATOR sets default allocator. Here is the grammar: |
3684 | |
3685 | <allocator> |= <predef-allocator> | <predef-mem-space> | |
3686 | <predef-mem-space>:<traits> |
3687 | <traits> |= <trait>=<value> | <trait>=<value>,<traits> |
3688 | <predef-allocator> |= omp_default_mem_alloc | omp_large_cap_mem_alloc | |
3689 | omp_const_mem_alloc | omp_high_bw_mem_alloc | |
3690 | omp_low_lat_mem_alloc | omp_cgroup_mem_alloc | |
3691 | omp_pteam_mem_alloc | omp_thread_mem_alloc |
3692 | <predef-mem-space> |= omp_default_mem_space | omp_large_cap_mem_space | |
3693 | omp_const_mem_space | omp_high_bw_mem_space | |
3694 | omp_low_lat_mem_space |
3695 | <trait> |= sync_hint | alignment | access | pool_size | fallback | |
3696 | fb_data | pinned | partition |
3697 | <value> |= one of the allowed values of trait | |
3698 | non-negative integer | <predef-allocator> |
3699 | -----------------------------------------------------------------------------*/ |
3700 | |
3701 | static void __kmp_stg_parse_allocator(char const *name, char const *value, |
3702 | void *data) { |
3703 | const char *buf = value; |
3704 | const char *next, *scan, *start; |
3705 | char *key; |
3706 | omp_allocator_handle_t al; |
3707 | omp_memspace_handle_t ms = omp_default_mem_space; |
3708 | bool is_memspace = false; |
3709 | int ntraits = 0, count = 0; |
3710 | |
3711 | SKIP_WS(buf); |
3712 | next = buf; |
3713 | const char *delim = strchr(s: buf, c: ':'); |
3714 | const char *predef_mem_space = strstr(haystack: buf, needle: "mem_space" ); |
3715 | |
3716 | bool is_memalloc = (!predef_mem_space && !delim) ? true : false; |
3717 | |
3718 | // Count the number of traits in the env var string |
3719 | if (delim) { |
3720 | ntraits = 1; |
3721 | for (scan = buf; *scan != '\0'; scan++) { |
3722 | if (*scan == ',') |
3723 | ntraits++; |
3724 | } |
3725 | } |
3726 | omp_alloctrait_t *traits = |
3727 | (omp_alloctrait_t *)KMP_ALLOCA(ntraits * sizeof(omp_alloctrait_t)); |
3728 | |
3729 | // Helper macros |
3730 | #define IS_POWER_OF_TWO(n) (((n) & ((n)-1)) == 0) |
3731 | |
3732 | #define GET_NEXT(sentinel) \ |
3733 | { \ |
3734 | SKIP_WS(next); \ |
3735 | if (*next == sentinel) \ |
3736 | next++; \ |
3737 | SKIP_WS(next); \ |
3738 | scan = next; \ |
3739 | } |
3740 | |
3741 | #define SKIP_PAIR(key) \ |
3742 | { \ |
3743 | char const str_delimiter[] = {',', 0}; \ |
3744 | char *value = __kmp_str_token(CCAST(char *, scan), str_delimiter, \ |
3745 | CCAST(char **, &next)); \ |
3746 | KMP_WARNING(StgInvalidValue, key, value); \ |
3747 | ntraits--; \ |
3748 | SKIP_WS(next); \ |
3749 | scan = next; \ |
3750 | } |
3751 | |
3752 | #define SET_KEY() \ |
3753 | { \ |
3754 | char const str_delimiter[] = {'=', 0}; \ |
3755 | key = __kmp_str_token(CCAST(char *, start), str_delimiter, \ |
3756 | CCAST(char **, &next)); \ |
3757 | scan = next; \ |
3758 | } |
3759 | |
3760 | scan = next; |
3761 | while (*next != '\0') { |
3762 | if (is_memalloc || |
3763 | __kmp_match_str(token: "fb_data" , buf: scan, end: &next)) { // allocator check |
3764 | start = scan; |
3765 | GET_NEXT('='); |
3766 | // check HBW and LCAP first as the only non-default supported |
3767 | if (__kmp_match_str(token: "omp_high_bw_mem_alloc" , buf: scan, end: &next)) { |
3768 | SKIP_WS(next); |
3769 | if (is_memalloc) { |
3770 | if (__kmp_memkind_available) { |
3771 | __kmp_def_allocator = omp_high_bw_mem_alloc; |
3772 | return; |
3773 | } else { |
3774 | KMP_WARNING(OmpNoAllocator, "omp_high_bw_mem_alloc" ); |
3775 | } |
3776 | } else { |
3777 | traits[count].key = omp_atk_fb_data; |
3778 | traits[count].value = RCAST(omp_uintptr_t, omp_high_bw_mem_alloc); |
3779 | } |
3780 | } else if (__kmp_match_str(token: "omp_large_cap_mem_alloc" , buf: scan, end: &next)) { |
3781 | SKIP_WS(next); |
3782 | if (is_memalloc) { |
3783 | if (__kmp_memkind_available) { |
3784 | __kmp_def_allocator = omp_large_cap_mem_alloc; |
3785 | return; |
3786 | } else { |
3787 | KMP_WARNING(OmpNoAllocator, "omp_large_cap_mem_alloc" ); |
3788 | } |
3789 | } else { |
3790 | traits[count].key = omp_atk_fb_data; |
3791 | traits[count].value = RCAST(omp_uintptr_t, omp_large_cap_mem_alloc); |
3792 | } |
3793 | } else if (__kmp_match_str(token: "omp_default_mem_alloc" , buf: scan, end: &next)) { |
3794 | // default requested |
3795 | SKIP_WS(next); |
3796 | if (!is_memalloc) { |
3797 | traits[count].key = omp_atk_fb_data; |
3798 | traits[count].value = RCAST(omp_uintptr_t, omp_default_mem_alloc); |
3799 | } |
3800 | } else if (__kmp_match_str(token: "omp_const_mem_alloc" , buf: scan, end: &next)) { |
3801 | SKIP_WS(next); |
3802 | if (is_memalloc) { |
3803 | KMP_WARNING(OmpNoAllocator, "omp_const_mem_alloc" ); |
3804 | } else { |
3805 | traits[count].key = omp_atk_fb_data; |
3806 | traits[count].value = RCAST(omp_uintptr_t, omp_const_mem_alloc); |
3807 | } |
3808 | } else if (__kmp_match_str(token: "omp_low_lat_mem_alloc" , buf: scan, end: &next)) { |
3809 | SKIP_WS(next); |
3810 | if (is_memalloc) { |
3811 | KMP_WARNING(OmpNoAllocator, "omp_low_lat_mem_alloc" ); |
3812 | } else { |
3813 | traits[count].key = omp_atk_fb_data; |
3814 | traits[count].value = RCAST(omp_uintptr_t, omp_low_lat_mem_alloc); |
3815 | } |
3816 | } else if (__kmp_match_str(token: "omp_cgroup_mem_alloc" , buf: scan, end: &next)) { |
3817 | SKIP_WS(next); |
3818 | if (is_memalloc) { |
3819 | KMP_WARNING(OmpNoAllocator, "omp_cgroup_mem_alloc" ); |
3820 | } else { |
3821 | traits[count].key = omp_atk_fb_data; |
3822 | traits[count].value = RCAST(omp_uintptr_t, omp_cgroup_mem_alloc); |
3823 | } |
3824 | } else if (__kmp_match_str(token: "omp_pteam_mem_alloc" , buf: scan, end: &next)) { |
3825 | SKIP_WS(next); |
3826 | if (is_memalloc) { |
3827 | KMP_WARNING(OmpNoAllocator, "omp_pteam_mem_alloc" ); |
3828 | } else { |
3829 | traits[count].key = omp_atk_fb_data; |
3830 | traits[count].value = RCAST(omp_uintptr_t, omp_pteam_mem_alloc); |
3831 | } |
3832 | } else if (__kmp_match_str(token: "omp_thread_mem_alloc" , buf: scan, end: &next)) { |
3833 | SKIP_WS(next); |
3834 | if (is_memalloc) { |
3835 | KMP_WARNING(OmpNoAllocator, "omp_thread_mem_alloc" ); |
3836 | } else { |
3837 | traits[count].key = omp_atk_fb_data; |
3838 | traits[count].value = RCAST(omp_uintptr_t, omp_thread_mem_alloc); |
3839 | } |
3840 | } else { |
3841 | if (!is_memalloc) { |
3842 | SET_KEY(); |
3843 | SKIP_PAIR(key); |
3844 | continue; |
3845 | } |
3846 | } |
3847 | if (is_memalloc) { |
3848 | __kmp_def_allocator = omp_default_mem_alloc; |
3849 | if (next == buf || *next != '\0') { |
3850 | // either no match or extra symbols present after the matched token |
3851 | KMP_WARNING(StgInvalidValue, name, value); |
3852 | } |
3853 | return; |
3854 | } else { |
3855 | ++count; |
3856 | if (count == ntraits) |
3857 | break; |
3858 | GET_NEXT(','); |
3859 | } |
3860 | } else { // memspace |
3861 | if (!is_memspace) { |
3862 | if (__kmp_match_str(token: "omp_default_mem_space" , buf: scan, end: &next)) { |
3863 | SKIP_WS(next); |
3864 | ms = omp_default_mem_space; |
3865 | } else if (__kmp_match_str(token: "omp_large_cap_mem_space" , buf: scan, end: &next)) { |
3866 | SKIP_WS(next); |
3867 | ms = omp_large_cap_mem_space; |
3868 | } else if (__kmp_match_str(token: "omp_const_mem_space" , buf: scan, end: &next)) { |
3869 | SKIP_WS(next); |
3870 | ms = omp_const_mem_space; |
3871 | } else if (__kmp_match_str(token: "omp_high_bw_mem_space" , buf: scan, end: &next)) { |
3872 | SKIP_WS(next); |
3873 | ms = omp_high_bw_mem_space; |
3874 | } else if (__kmp_match_str(token: "omp_low_lat_mem_space" , buf: scan, end: &next)) { |
3875 | SKIP_WS(next); |
3876 | ms = omp_low_lat_mem_space; |
3877 | } else { |
3878 | __kmp_def_allocator = omp_default_mem_alloc; |
3879 | if (next == buf || *next != '\0') { |
3880 | // either no match or extra symbols present after the matched token |
3881 | KMP_WARNING(StgInvalidValue, name, value); |
3882 | } |
3883 | return; |
3884 | } |
3885 | is_memspace = true; |
3886 | } |
3887 | if (delim) { // traits |
3888 | GET_NEXT(':'); |
3889 | start = scan; |
3890 | if (__kmp_match_str(token: "sync_hint" , buf: scan, end: &next)) { |
3891 | GET_NEXT('='); |
3892 | traits[count].key = omp_atk_sync_hint; |
3893 | if (__kmp_match_str(token: "contended" , buf: scan, end: &next)) { |
3894 | traits[count].value = omp_atv_contended; |
3895 | } else if (__kmp_match_str(token: "uncontended" , buf: scan, end: &next)) { |
3896 | traits[count].value = omp_atv_uncontended; |
3897 | } else if (__kmp_match_str(token: "serialized" , buf: scan, end: &next)) { |
3898 | traits[count].value = omp_atv_serialized; |
3899 | } else if (__kmp_match_str(token: "private" , buf: scan, end: &next)) { |
3900 | traits[count].value = omp_atv_private; |
3901 | } else { |
3902 | SET_KEY(); |
3903 | SKIP_PAIR(key); |
3904 | continue; |
3905 | } |
3906 | } else if (__kmp_match_str(token: "alignment" , buf: scan, end: &next)) { |
3907 | GET_NEXT('='); |
3908 | if (!isdigit(*next)) { |
3909 | SET_KEY(); |
3910 | SKIP_PAIR(key); |
3911 | continue; |
3912 | } |
3913 | SKIP_DIGITS(next); |
3914 | int n = __kmp_str_to_int(str: scan, sentinel: ','); |
3915 | if (n < 0 || !IS_POWER_OF_TWO(n)) { |
3916 | SET_KEY(); |
3917 | SKIP_PAIR(key); |
3918 | continue; |
3919 | } |
3920 | traits[count].key = omp_atk_alignment; |
3921 | traits[count].value = n; |
3922 | } else if (__kmp_match_str(token: "access" , buf: scan, end: &next)) { |
3923 | GET_NEXT('='); |
3924 | traits[count].key = omp_atk_access; |
3925 | if (__kmp_match_str(token: "all" , buf: scan, end: &next)) { |
3926 | traits[count].value = omp_atv_all; |
3927 | } else if (__kmp_match_str(token: "cgroup" , buf: scan, end: &next)) { |
3928 | traits[count].value = omp_atv_cgroup; |
3929 | } else if (__kmp_match_str(token: "pteam" , buf: scan, end: &next)) { |
3930 | traits[count].value = omp_atv_pteam; |
3931 | } else if (__kmp_match_str(token: "thread" , buf: scan, end: &next)) { |
3932 | traits[count].value = omp_atv_thread; |
3933 | } else { |
3934 | SET_KEY(); |
3935 | SKIP_PAIR(key); |
3936 | continue; |
3937 | } |
3938 | } else if (__kmp_match_str(token: "pool_size" , buf: scan, end: &next)) { |
3939 | GET_NEXT('='); |
3940 | if (!isdigit(*next)) { |
3941 | SET_KEY(); |
3942 | SKIP_PAIR(key); |
3943 | continue; |
3944 | } |
3945 | SKIP_DIGITS(next); |
3946 | int n = __kmp_str_to_int(str: scan, sentinel: ','); |
3947 | if (n < 0) { |
3948 | SET_KEY(); |
3949 | SKIP_PAIR(key); |
3950 | continue; |
3951 | } |
3952 | traits[count].key = omp_atk_pool_size; |
3953 | traits[count].value = n; |
3954 | } else if (__kmp_match_str(token: "fallback" , buf: scan, end: &next)) { |
3955 | GET_NEXT('='); |
3956 | traits[count].key = omp_atk_fallback; |
3957 | if (__kmp_match_str(token: "default_mem_fb" , buf: scan, end: &next)) { |
3958 | traits[count].value = omp_atv_default_mem_fb; |
3959 | } else if (__kmp_match_str(token: "null_fb" , buf: scan, end: &next)) { |
3960 | traits[count].value = omp_atv_null_fb; |
3961 | } else if (__kmp_match_str(token: "abort_fb" , buf: scan, end: &next)) { |
3962 | traits[count].value = omp_atv_abort_fb; |
3963 | } else if (__kmp_match_str(token: "allocator_fb" , buf: scan, end: &next)) { |
3964 | traits[count].value = omp_atv_allocator_fb; |
3965 | } else { |
3966 | SET_KEY(); |
3967 | SKIP_PAIR(key); |
3968 | continue; |
3969 | } |
3970 | } else if (__kmp_match_str(token: "pinned" , buf: scan, end: &next)) { |
3971 | GET_NEXT('='); |
3972 | traits[count].key = omp_atk_pinned; |
3973 | if (__kmp_str_match_true(data: next)) { |
3974 | traits[count].value = omp_atv_true; |
3975 | } else if (__kmp_str_match_false(data: next)) { |
3976 | traits[count].value = omp_atv_false; |
3977 | } else { |
3978 | SET_KEY(); |
3979 | SKIP_PAIR(key); |
3980 | continue; |
3981 | } |
3982 | } else if (__kmp_match_str(token: "partition" , buf: scan, end: &next)) { |
3983 | GET_NEXT('='); |
3984 | traits[count].key = omp_atk_partition; |
3985 | if (__kmp_match_str(token: "environment" , buf: scan, end: &next)) { |
3986 | traits[count].value = omp_atv_environment; |
3987 | } else if (__kmp_match_str(token: "nearest" , buf: scan, end: &next)) { |
3988 | traits[count].value = omp_atv_nearest; |
3989 | } else if (__kmp_match_str(token: "blocked" , buf: scan, end: &next)) { |
3990 | traits[count].value = omp_atv_blocked; |
3991 | } else if (__kmp_match_str(token: "interleaved" , buf: scan, end: &next)) { |
3992 | traits[count].value = omp_atv_interleaved; |
3993 | } else { |
3994 | SET_KEY(); |
3995 | SKIP_PAIR(key); |
3996 | continue; |
3997 | } |
3998 | } else { |
3999 | SET_KEY(); |
4000 | SKIP_PAIR(key); |
4001 | continue; |
4002 | } |
4003 | SKIP_WS(next); |
4004 | ++count; |
4005 | if (count == ntraits) |
4006 | break; |
4007 | GET_NEXT(','); |
4008 | } // traits |
4009 | } // memspace |
4010 | } // while |
4011 | al = __kmpc_init_allocator(__kmp_get_gtid(), ms, ntraits, traits); |
4012 | __kmp_def_allocator = (al == omp_null_allocator) ? omp_default_mem_alloc : al; |
4013 | } |
4014 | |
4015 | static void __kmp_stg_print_allocator(kmp_str_buf_t *buffer, char const *name, |
4016 | void *data) { |
4017 | if (__kmp_def_allocator == omp_default_mem_alloc) { |
4018 | __kmp_stg_print_str(buffer, name, value: "omp_default_mem_alloc" ); |
4019 | } else if (__kmp_def_allocator == omp_high_bw_mem_alloc) { |
4020 | __kmp_stg_print_str(buffer, name, value: "omp_high_bw_mem_alloc" ); |
4021 | } else if (__kmp_def_allocator == omp_large_cap_mem_alloc) { |
4022 | __kmp_stg_print_str(buffer, name, value: "omp_large_cap_mem_alloc" ); |
4023 | } else if (__kmp_def_allocator == omp_const_mem_alloc) { |
4024 | __kmp_stg_print_str(buffer, name, value: "omp_const_mem_alloc" ); |
4025 | } else if (__kmp_def_allocator == omp_low_lat_mem_alloc) { |
4026 | __kmp_stg_print_str(buffer, name, value: "omp_low_lat_mem_alloc" ); |
4027 | } else if (__kmp_def_allocator == omp_cgroup_mem_alloc) { |
4028 | __kmp_stg_print_str(buffer, name, value: "omp_cgroup_mem_alloc" ); |
4029 | } else if (__kmp_def_allocator == omp_pteam_mem_alloc) { |
4030 | __kmp_stg_print_str(buffer, name, value: "omp_pteam_mem_alloc" ); |
4031 | } else if (__kmp_def_allocator == omp_thread_mem_alloc) { |
4032 | __kmp_stg_print_str(buffer, name, value: "omp_thread_mem_alloc" ); |
4033 | } |
4034 | } |
4035 | |
4036 | // ----------------------------------------------------------------------------- |
4037 | // OMP_DYNAMIC |
4038 | |
4039 | static void __kmp_stg_parse_omp_dynamic(char const *name, char const *value, |
4040 | void *data) { |
4041 | __kmp_stg_parse_bool(name, value, out: &(__kmp_global.g.g_dynamic)); |
4042 | } // __kmp_stg_parse_omp_dynamic |
4043 | |
4044 | static void __kmp_stg_print_omp_dynamic(kmp_str_buf_t *buffer, char const *name, |
4045 | void *data) { |
4046 | __kmp_stg_print_bool(buffer, name, value: __kmp_global.g.g_dynamic); |
4047 | } // __kmp_stg_print_omp_dynamic |
4048 | |
4049 | static void __kmp_stg_parse_kmp_dynamic_mode(char const *name, |
4050 | char const *value, void *data) { |
4051 | if (TCR_4(__kmp_init_parallel)) { |
4052 | KMP_WARNING(EnvParallelWarn, name); |
4053 | __kmp_env_toPrint(name, flag: 0); |
4054 | return; |
4055 | } |
4056 | #ifdef USE_LOAD_BALANCE |
4057 | else if (__kmp_str_match(target: "load balance" , len: 2, data: value) || |
4058 | __kmp_str_match(target: "load_balance" , len: 2, data: value) || |
4059 | __kmp_str_match(target: "load-balance" , len: 2, data: value) || |
4060 | __kmp_str_match(target: "loadbalance" , len: 2, data: value) || |
4061 | __kmp_str_match(target: "balance" , len: 1, data: value)) { |
4062 | __kmp_global.g.g_dynamic_mode = dynamic_load_balance; |
4063 | } |
4064 | #endif /* USE_LOAD_BALANCE */ |
4065 | else if (__kmp_str_match(target: "thread limit" , len: 1, data: value) || |
4066 | __kmp_str_match(target: "thread_limit" , len: 1, data: value) || |
4067 | __kmp_str_match(target: "thread-limit" , len: 1, data: value) || |
4068 | __kmp_str_match(target: "threadlimit" , len: 1, data: value) || |
4069 | __kmp_str_match(target: "limit" , len: 2, data: value)) { |
4070 | __kmp_global.g.g_dynamic_mode = dynamic_thread_limit; |
4071 | } else if (__kmp_str_match(target: "random" , len: 1, data: value)) { |
4072 | __kmp_global.g.g_dynamic_mode = dynamic_random; |
4073 | } else { |
4074 | KMP_WARNING(StgInvalidValue, name, value); |
4075 | } |
4076 | } //__kmp_stg_parse_kmp_dynamic_mode |
4077 | |
4078 | static void __kmp_stg_print_kmp_dynamic_mode(kmp_str_buf_t *buffer, |
4079 | char const *name, void *data) { |
4080 | #if KMP_DEBUG |
4081 | if (__kmp_global.g.g_dynamic_mode == dynamic_default) { |
4082 | __kmp_str_buf_print(buffer, " %s: %s \n" , name, KMP_I18N_STR(NotDefined)); |
4083 | } |
4084 | #ifdef USE_LOAD_BALANCE |
4085 | else if (__kmp_global.g.g_dynamic_mode == dynamic_load_balance) { |
4086 | __kmp_stg_print_str(buffer, name, value: "load balance" ); |
4087 | } |
4088 | #endif /* USE_LOAD_BALANCE */ |
4089 | else if (__kmp_global.g.g_dynamic_mode == dynamic_thread_limit) { |
4090 | __kmp_stg_print_str(buffer, name, value: "thread limit" ); |
4091 | } else if (__kmp_global.g.g_dynamic_mode == dynamic_random) { |
4092 | __kmp_stg_print_str(buffer, name, value: "random" ); |
4093 | } else { |
4094 | KMP_ASSERT(0); |
4095 | } |
4096 | #endif /* KMP_DEBUG */ |
4097 | } // __kmp_stg_print_kmp_dynamic_mode |
4098 | |
4099 | #ifdef USE_LOAD_BALANCE |
4100 | |
4101 | // ----------------------------------------------------------------------------- |
4102 | // KMP_LOAD_BALANCE_INTERVAL |
4103 | |
4104 | static void __kmp_stg_parse_ld_balance_interval(char const *name, |
4105 | char const *value, void *data) { |
4106 | double interval = __kmp_convert_to_double(s: value); |
4107 | if (interval >= 0) { |
4108 | __kmp_load_balance_interval = interval; |
4109 | } else { |
4110 | KMP_WARNING(StgInvalidValue, name, value); |
4111 | } |
4112 | } // __kmp_stg_parse_load_balance_interval |
4113 | |
4114 | static void __kmp_stg_print_ld_balance_interval(kmp_str_buf_t *buffer, |
4115 | char const *name, void *data) { |
4116 | #if KMP_DEBUG |
4117 | __kmp_str_buf_print(buffer, format: " %s=%8.6f\n" , name, |
4118 | __kmp_load_balance_interval); |
4119 | #endif /* KMP_DEBUG */ |
4120 | } // __kmp_stg_print_load_balance_interval |
4121 | |
4122 | #endif /* USE_LOAD_BALANCE */ |
4123 | |
4124 | // ----------------------------------------------------------------------------- |
4125 | // KMP_INIT_AT_FORK |
4126 | |
4127 | static void __kmp_stg_parse_init_at_fork(char const *name, char const *value, |
4128 | void *data) { |
4129 | __kmp_stg_parse_bool(name, value, out: &__kmp_need_register_atfork); |
4130 | if (__kmp_need_register_atfork) { |
4131 | __kmp_need_register_atfork_specified = TRUE; |
4132 | } |
4133 | } // __kmp_stg_parse_init_at_fork |
4134 | |
4135 | static void __kmp_stg_print_init_at_fork(kmp_str_buf_t *buffer, |
4136 | char const *name, void *data) { |
4137 | __kmp_stg_print_bool(buffer, name, value: __kmp_need_register_atfork_specified); |
4138 | } // __kmp_stg_print_init_at_fork |
4139 | |
4140 | // ----------------------------------------------------------------------------- |
4141 | // KMP_SCHEDULE |
4142 | |
4143 | static void __kmp_stg_parse_schedule(char const *name, char const *value, |
4144 | void *data) { |
4145 | |
4146 | if (value != NULL) { |
4147 | size_t length = KMP_STRLEN(s: value); |
4148 | if (length > INT_MAX) { |
4149 | KMP_WARNING(LongValue, name); |
4150 | } else { |
4151 | const char *semicolon; |
4152 | if (value[length - 1] == '"' || value[length - 1] == '\'') |
4153 | KMP_WARNING(UnbalancedQuotes, name); |
4154 | do { |
4155 | char sentinel; |
4156 | |
4157 | semicolon = strchr(s: value, c: ';'); |
4158 | if (*value && semicolon != value) { |
4159 | const char *comma = strchr(s: value, c: ','); |
4160 | |
4161 | if (comma) { |
4162 | ++comma; |
4163 | sentinel = ','; |
4164 | } else |
4165 | sentinel = ';'; |
4166 | if (!__kmp_strcasecmp_with_sentinel(a: "static" , b: value, sentinel)) { |
4167 | if (!__kmp_strcasecmp_with_sentinel(a: "greedy" , b: comma, sentinel: ';')) { |
4168 | __kmp_static = kmp_sch_static_greedy; |
4169 | continue; |
4170 | } else if (!__kmp_strcasecmp_with_sentinel(a: "balanced" , b: comma, |
4171 | sentinel: ';')) { |
4172 | __kmp_static = kmp_sch_static_balanced; |
4173 | continue; |
4174 | } |
4175 | } else if (!__kmp_strcasecmp_with_sentinel(a: "guided" , b: value, |
4176 | sentinel)) { |
4177 | if (!__kmp_strcasecmp_with_sentinel(a: "iterative" , b: comma, sentinel: ';')) { |
4178 | __kmp_guided = kmp_sch_guided_iterative_chunked; |
4179 | continue; |
4180 | } else if (!__kmp_strcasecmp_with_sentinel(a: "analytical" , b: comma, |
4181 | sentinel: ';')) { |
4182 | /* analytical not allowed for too many threads */ |
4183 | __kmp_guided = kmp_sch_guided_analytical_chunked; |
4184 | continue; |
4185 | } |
4186 | } |
4187 | KMP_WARNING(InvalidClause, name, value); |
4188 | } else |
4189 | KMP_WARNING(EmptyClause, name); |
4190 | } while ((value = semicolon ? semicolon + 1 : NULL)); |
4191 | } |
4192 | } |
4193 | |
4194 | } // __kmp_stg_parse__schedule |
4195 | |
4196 | static void __kmp_stg_print_schedule(kmp_str_buf_t *buffer, char const *name, |
4197 | void *data) { |
4198 | if (__kmp_env_format) { |
4199 | KMP_STR_BUF_PRINT_NAME_EX(name); |
4200 | } else { |
4201 | __kmp_str_buf_print(buffer, format: " %s='" , name); |
4202 | } |
4203 | if (__kmp_static == kmp_sch_static_greedy) { |
4204 | __kmp_str_buf_print(buffer, format: "%s" , "static,greedy" ); |
4205 | } else if (__kmp_static == kmp_sch_static_balanced) { |
4206 | __kmp_str_buf_print(buffer, format: "%s" , "static,balanced" ); |
4207 | } |
4208 | if (__kmp_guided == kmp_sch_guided_iterative_chunked) { |
4209 | __kmp_str_buf_print(buffer, format: ";%s'\n" , "guided,iterative" ); |
4210 | } else if (__kmp_guided == kmp_sch_guided_analytical_chunked) { |
4211 | __kmp_str_buf_print(buffer, format: ";%s'\n" , "guided,analytical" ); |
4212 | } |
4213 | } // __kmp_stg_print_schedule |
4214 | |
4215 | // ----------------------------------------------------------------------------- |
4216 | // OMP_SCHEDULE |
4217 | |
4218 | static inline void __kmp_omp_schedule_restore() { |
4219 | #if KMP_USE_HIER_SCHED |
4220 | __kmp_hier_scheds.deallocate(); |
4221 | #endif |
4222 | __kmp_chunk = 0; |
4223 | __kmp_sched = kmp_sch_default; |
4224 | } |
4225 | |
4226 | // if parse_hier = true: |
4227 | // Parse [HW,][modifier:]kind[,chunk] |
4228 | // else: |
4229 | // Parse [modifier:]kind[,chunk] |
4230 | static const char *__kmp_parse_single_omp_schedule(const char *name, |
4231 | const char *value, |
4232 | bool parse_hier = false) { |
4233 | /* get the specified scheduling style */ |
4234 | const char *ptr = value; |
4235 | const char *delim; |
4236 | int chunk = 0; |
4237 | enum sched_type sched = kmp_sch_default; |
4238 | if (*ptr == '\0') |
4239 | return NULL; |
4240 | delim = ptr; |
4241 | while (*delim != ',' && *delim != ':' && *delim != '\0') |
4242 | delim++; |
4243 | #if KMP_USE_HIER_SCHED |
4244 | kmp_hier_layer_e layer = kmp_hier_layer_e::LAYER_THREAD; |
4245 | if (parse_hier) { |
4246 | if (*delim == ',') { |
4247 | if (!__kmp_strcasecmp_with_sentinel("L1" , ptr, ',')) { |
4248 | layer = kmp_hier_layer_e::LAYER_L1; |
4249 | } else if (!__kmp_strcasecmp_with_sentinel("L2" , ptr, ',')) { |
4250 | layer = kmp_hier_layer_e::LAYER_L2; |
4251 | } else if (!__kmp_strcasecmp_with_sentinel("L3" , ptr, ',')) { |
4252 | layer = kmp_hier_layer_e::LAYER_L3; |
4253 | } else if (!__kmp_strcasecmp_with_sentinel("NUMA" , ptr, ',')) { |
4254 | layer = kmp_hier_layer_e::LAYER_NUMA; |
4255 | } |
4256 | } |
4257 | if (layer != kmp_hier_layer_e::LAYER_THREAD && *delim != ',') { |
4258 | // If there is no comma after the layer, then this schedule is invalid |
4259 | KMP_WARNING(StgInvalidValue, name, value); |
4260 | __kmp_omp_schedule_restore(); |
4261 | return NULL; |
4262 | } else if (layer != kmp_hier_layer_e::LAYER_THREAD) { |
4263 | ptr = ++delim; |
4264 | while (*delim != ',' && *delim != ':' && *delim != '\0') |
4265 | delim++; |
4266 | } |
4267 | } |
4268 | #endif // KMP_USE_HIER_SCHED |
4269 | // Read in schedule modifier if specified |
4270 | enum sched_type sched_modifier = (enum sched_type)0; |
4271 | if (*delim == ':') { |
4272 | if (!__kmp_strcasecmp_with_sentinel(a: "monotonic" , b: ptr, sentinel: *delim)) { |
4273 | sched_modifier = sched_type::kmp_sch_modifier_monotonic; |
4274 | ptr = ++delim; |
4275 | while (*delim != ',' && *delim != ':' && *delim != '\0') |
4276 | delim++; |
4277 | } else if (!__kmp_strcasecmp_with_sentinel(a: "nonmonotonic" , b: ptr, sentinel: *delim)) { |
4278 | sched_modifier = sched_type::kmp_sch_modifier_nonmonotonic; |
4279 | ptr = ++delim; |
4280 | while (*delim != ',' && *delim != ':' && *delim != '\0') |
4281 | delim++; |
4282 | } else if (!parse_hier) { |
4283 | // If there is no proper schedule modifier, then this schedule is invalid |
4284 | KMP_WARNING(StgInvalidValue, name, value); |
4285 | __kmp_omp_schedule_restore(); |
4286 | return NULL; |
4287 | } |
4288 | } |
4289 | // Read in schedule kind (required) |
4290 | if (!__kmp_strcasecmp_with_sentinel(a: "dynamic" , b: ptr, sentinel: *delim)) |
4291 | sched = kmp_sch_dynamic_chunked; |
4292 | else if (!__kmp_strcasecmp_with_sentinel(a: "guided" , b: ptr, sentinel: *delim)) |
4293 | sched = kmp_sch_guided_chunked; |
4294 | // AC: TODO: probably remove TRAPEZOIDAL (OMP 3.0 does not allow it) |
4295 | else if (!__kmp_strcasecmp_with_sentinel(a: "auto" , b: ptr, sentinel: *delim)) |
4296 | sched = kmp_sch_auto; |
4297 | else if (!__kmp_strcasecmp_with_sentinel(a: "trapezoidal" , b: ptr, sentinel: *delim)) |
4298 | sched = kmp_sch_trapezoidal; |
4299 | else if (!__kmp_strcasecmp_with_sentinel(a: "static" , b: ptr, sentinel: *delim)) |
4300 | sched = kmp_sch_static; |
4301 | #if KMP_STATIC_STEAL_ENABLED |
4302 | else if (!__kmp_strcasecmp_with_sentinel(a: "static_steal" , b: ptr, sentinel: *delim)) { |
4303 | // replace static_steal with dynamic to better cope with ordered loops |
4304 | sched = kmp_sch_dynamic_chunked; |
4305 | sched_modifier = sched_type::kmp_sch_modifier_nonmonotonic; |
4306 | } |
4307 | #endif |
4308 | else { |
4309 | // If there is no proper schedule kind, then this schedule is invalid |
4310 | KMP_WARNING(StgInvalidValue, name, value); |
4311 | __kmp_omp_schedule_restore(); |
4312 | return NULL; |
4313 | } |
4314 | |
4315 | // Read in schedule chunk size if specified |
4316 | if (*delim == ',') { |
4317 | ptr = delim + 1; |
4318 | SKIP_WS(ptr); |
4319 | if (!isdigit(*ptr)) { |
4320 | // If there is no chunk after comma, then this schedule is invalid |
4321 | KMP_WARNING(StgInvalidValue, name, value); |
4322 | __kmp_omp_schedule_restore(); |
4323 | return NULL; |
4324 | } |
4325 | SKIP_DIGITS(ptr); |
4326 | // auto schedule should not specify chunk size |
4327 | if (sched == kmp_sch_auto) { |
4328 | __kmp_msg(kmp_ms_warning, KMP_MSG(IgnoreChunk, name, delim), |
4329 | __kmp_msg_null); |
4330 | } else { |
4331 | if (sched == kmp_sch_static) |
4332 | sched = kmp_sch_static_chunked; |
4333 | chunk = __kmp_str_to_int(str: delim + 1, sentinel: *ptr); |
4334 | if (chunk < 1) { |
4335 | chunk = KMP_DEFAULT_CHUNK; |
4336 | __kmp_msg(kmp_ms_warning, KMP_MSG(InvalidChunk, name, delim), |
4337 | __kmp_msg_null); |
4338 | KMP_INFORM(Using_int_Value, name, __kmp_chunk); |
4339 | // AC: next block commented out until KMP_DEFAULT_CHUNK != KMP_MIN_CHUNK |
4340 | // (to improve code coverage :) |
4341 | // The default chunk size is 1 according to standard, thus making |
4342 | // KMP_MIN_CHUNK not 1 we would introduce mess: |
4343 | // wrong chunk becomes 1, but it will be impossible to explicitly set |
4344 | // to 1 because it becomes KMP_MIN_CHUNK... |
4345 | // } else if ( chunk < KMP_MIN_CHUNK ) { |
4346 | // chunk = KMP_MIN_CHUNK; |
4347 | } else if (chunk > KMP_MAX_CHUNK) { |
4348 | chunk = KMP_MAX_CHUNK; |
4349 | __kmp_msg(kmp_ms_warning, KMP_MSG(LargeChunk, name, delim), |
4350 | __kmp_msg_null); |
4351 | KMP_INFORM(Using_int_Value, name, chunk); |
4352 | } |
4353 | } |
4354 | } else { |
4355 | ptr = delim; |
4356 | } |
4357 | |
4358 | SCHEDULE_SET_MODIFIERS(sched, sched_modifier); |
4359 | |
4360 | #if KMP_USE_HIER_SCHED |
4361 | if (layer != kmp_hier_layer_e::LAYER_THREAD) { |
4362 | __kmp_hier_scheds.append(sched, chunk, layer); |
4363 | } else |
4364 | #endif |
4365 | { |
4366 | __kmp_chunk = chunk; |
4367 | __kmp_sched = sched; |
4368 | } |
4369 | return ptr; |
4370 | } |
4371 | |
4372 | static void __kmp_stg_parse_omp_schedule(char const *name, char const *value, |
4373 | void *data) { |
4374 | size_t length; |
4375 | const char *ptr = value; |
4376 | if (ptr) { |
4377 | SKIP_WS(ptr); |
4378 | length = KMP_STRLEN(s: value); |
4379 | if (length) { |
4380 | if (value[length - 1] == '"' || value[length - 1] == '\'') |
4381 | KMP_WARNING(UnbalancedQuotes, name); |
4382 | /* get the specified scheduling style */ |
4383 | #if KMP_USE_HIER_SCHED |
4384 | if (!__kmp_strcasecmp_with_sentinel("EXPERIMENTAL" , ptr, ' ')) { |
4385 | SKIP_TOKEN(ptr); |
4386 | SKIP_WS(ptr); |
4387 | while ((ptr = __kmp_parse_single_omp_schedule(name, ptr, true))) { |
4388 | while (*ptr == ' ' || *ptr == '\t' || *ptr == ':') |
4389 | ptr++; |
4390 | if (*ptr == '\0') |
4391 | break; |
4392 | } |
4393 | } else |
4394 | #endif |
4395 | __kmp_parse_single_omp_schedule(name, value: ptr); |
4396 | } else |
4397 | KMP_WARNING(EmptyString, name); |
4398 | } |
4399 | #if KMP_USE_HIER_SCHED |
4400 | __kmp_hier_scheds.sort(); |
4401 | #endif |
4402 | K_DIAG(1, ("__kmp_static == %d\n" , __kmp_static)) |
4403 | K_DIAG(1, ("__kmp_guided == %d\n" , __kmp_guided)) |
4404 | K_DIAG(1, ("__kmp_sched == %d\n" , __kmp_sched)) |
4405 | K_DIAG(1, ("__kmp_chunk == %d\n" , __kmp_chunk)) |
4406 | } // __kmp_stg_parse_omp_schedule |
4407 | |
4408 | static void __kmp_stg_print_omp_schedule(kmp_str_buf_t *buffer, |
4409 | char const *name, void *data) { |
4410 | if (__kmp_env_format) { |
4411 | KMP_STR_BUF_PRINT_NAME_EX(name); |
4412 | } else { |
4413 | __kmp_str_buf_print(buffer, format: " %s='" , name); |
4414 | } |
4415 | enum sched_type sched = SCHEDULE_WITHOUT_MODIFIERS(__kmp_sched); |
4416 | if (SCHEDULE_HAS_MONOTONIC(__kmp_sched)) { |
4417 | __kmp_str_buf_print(buffer, format: "monotonic:" ); |
4418 | } else if (SCHEDULE_HAS_NONMONOTONIC(__kmp_sched)) { |
4419 | __kmp_str_buf_print(buffer, format: "nonmonotonic:" ); |
4420 | } |
4421 | if (__kmp_chunk) { |
4422 | switch (sched) { |
4423 | case kmp_sch_dynamic_chunked: |
4424 | __kmp_str_buf_print(buffer, format: "%s,%d'\n" , "dynamic" , __kmp_chunk); |
4425 | break; |
4426 | case kmp_sch_guided_iterative_chunked: |
4427 | case kmp_sch_guided_analytical_chunked: |
4428 | __kmp_str_buf_print(buffer, format: "%s,%d'\n" , "guided" , __kmp_chunk); |
4429 | break; |
4430 | case kmp_sch_trapezoidal: |
4431 | __kmp_str_buf_print(buffer, format: "%s,%d'\n" , "trapezoidal" , __kmp_chunk); |
4432 | break; |
4433 | case kmp_sch_static: |
4434 | case kmp_sch_static_chunked: |
4435 | case kmp_sch_static_balanced: |
4436 | case kmp_sch_static_greedy: |
4437 | __kmp_str_buf_print(buffer, format: "%s,%d'\n" , "static" , __kmp_chunk); |
4438 | break; |
4439 | case kmp_sch_static_steal: |
4440 | __kmp_str_buf_print(buffer, format: "%s,%d'\n" , "static_steal" , __kmp_chunk); |
4441 | break; |
4442 | case kmp_sch_auto: |
4443 | __kmp_str_buf_print(buffer, format: "%s,%d'\n" , "auto" , __kmp_chunk); |
4444 | break; |
4445 | default: |
4446 | KMP_ASSERT2(false, "Unhandled sched_type enumeration" ); |
4447 | KMP_BUILTIN_UNREACHABLE; |
4448 | break; |
4449 | } |
4450 | } else { |
4451 | switch (sched) { |
4452 | case kmp_sch_dynamic_chunked: |
4453 | __kmp_str_buf_print(buffer, format: "%s'\n" , "dynamic" ); |
4454 | break; |
4455 | case kmp_sch_guided_iterative_chunked: |
4456 | case kmp_sch_guided_analytical_chunked: |
4457 | __kmp_str_buf_print(buffer, format: "%s'\n" , "guided" ); |
4458 | break; |
4459 | case kmp_sch_trapezoidal: |
4460 | __kmp_str_buf_print(buffer, format: "%s'\n" , "trapezoidal" ); |
4461 | break; |
4462 | case kmp_sch_static: |
4463 | case kmp_sch_static_chunked: |
4464 | case kmp_sch_static_balanced: |
4465 | case kmp_sch_static_greedy: |
4466 | __kmp_str_buf_print(buffer, format: "%s'\n" , "static" ); |
4467 | break; |
4468 | case kmp_sch_static_steal: |
4469 | __kmp_str_buf_print(buffer, format: "%s'\n" , "static_steal" ); |
4470 | break; |
4471 | case kmp_sch_auto: |
4472 | __kmp_str_buf_print(buffer, format: "%s'\n" , "auto" ); |
4473 | break; |
4474 | default: |
4475 | KMP_ASSERT2(false, "Unhandled sched_type enumeration" ); |
4476 | KMP_BUILTIN_UNREACHABLE; |
4477 | break; |
4478 | } |
4479 | } |
4480 | } // __kmp_stg_print_omp_schedule |
4481 | |
4482 | #if KMP_USE_HIER_SCHED |
4483 | // ----------------------------------------------------------------------------- |
4484 | // KMP_DISP_HAND_THREAD |
4485 | static void __kmp_stg_parse_kmp_hand_thread(char const *name, char const *value, |
4486 | void *data) { |
4487 | __kmp_stg_parse_bool(name, value, &(__kmp_dispatch_hand_threading)); |
4488 | } // __kmp_stg_parse_kmp_hand_thread |
4489 | |
4490 | static void __kmp_stg_print_kmp_hand_thread(kmp_str_buf_t *buffer, |
4491 | char const *name, void *data) { |
4492 | __kmp_stg_print_bool(buffer, name, __kmp_dispatch_hand_threading); |
4493 | } // __kmp_stg_print_kmp_hand_thread |
4494 | #endif |
4495 | |
4496 | // ----------------------------------------------------------------------------- |
4497 | // KMP_FORCE_MONOTONIC_DYNAMIC_SCHEDULE |
4498 | static void __kmp_stg_parse_kmp_force_monotonic(char const *name, |
4499 | char const *value, void *data) { |
4500 | __kmp_stg_parse_bool(name, value, out: &(__kmp_force_monotonic)); |
4501 | } // __kmp_stg_parse_kmp_force_monotonic |
4502 | |
4503 | static void __kmp_stg_print_kmp_force_monotonic(kmp_str_buf_t *buffer, |
4504 | char const *name, void *data) { |
4505 | __kmp_stg_print_bool(buffer, name, value: __kmp_force_monotonic); |
4506 | } // __kmp_stg_print_kmp_force_monotonic |
4507 | |
4508 | // ----------------------------------------------------------------------------- |
4509 | // KMP_ATOMIC_MODE |
4510 | |
4511 | static void __kmp_stg_parse_atomic_mode(char const *name, char const *value, |
4512 | void *data) { |
4513 | // Modes: 0 -- do not change default; 1 -- Intel perf mode, 2 -- GOMP |
4514 | // compatibility mode. |
4515 | int mode = 0; |
4516 | int max = 1; |
4517 | #ifdef KMP_GOMP_COMPAT |
4518 | max = 2; |
4519 | #endif /* KMP_GOMP_COMPAT */ |
4520 | __kmp_stg_parse_int(name, value, min: 0, max, out: &mode); |
4521 | // TODO; parse_int is not very suitable for this case. In case of overflow it |
4522 | // is better to use |
4523 | // 0 rather that max value. |
4524 | if (mode > 0) { |
4525 | __kmp_atomic_mode = mode; |
4526 | } |
4527 | } // __kmp_stg_parse_atomic_mode |
4528 | |
4529 | static void __kmp_stg_print_atomic_mode(kmp_str_buf_t *buffer, char const *name, |
4530 | void *data) { |
4531 | __kmp_stg_print_int(buffer, name, value: __kmp_atomic_mode); |
4532 | } // __kmp_stg_print_atomic_mode |
4533 | |
4534 | // ----------------------------------------------------------------------------- |
4535 | // KMP_CONSISTENCY_CHECK |
4536 | |
4537 | static void __kmp_stg_parse_consistency_check(char const *name, |
4538 | char const *value, void *data) { |
4539 | if (!__kmp_strcasecmp_with_sentinel(a: "all" , b: value, sentinel: 0)) { |
4540 | // Note, this will not work from kmp_set_defaults because th_cons stack was |
4541 | // not allocated |
4542 | // for existed thread(s) thus the first __kmp_push_<construct> will break |
4543 | // with assertion. |
4544 | // TODO: allocate th_cons if called from kmp_set_defaults. |
4545 | __kmp_env_consistency_check = TRUE; |
4546 | } else if (!__kmp_strcasecmp_with_sentinel(a: "none" , b: value, sentinel: 0)) { |
4547 | __kmp_env_consistency_check = FALSE; |
4548 | } else { |
4549 | KMP_WARNING(StgInvalidValue, name, value); |
4550 | } |
4551 | } // __kmp_stg_parse_consistency_check |
4552 | |
4553 | static void __kmp_stg_print_consistency_check(kmp_str_buf_t *buffer, |
4554 | char const *name, void *data) { |
4555 | #if KMP_DEBUG |
4556 | const char *value = NULL; |
4557 | |
4558 | if (__kmp_env_consistency_check) { |
4559 | value = "all" ; |
4560 | } else { |
4561 | value = "none" ; |
4562 | } |
4563 | |
4564 | if (value != NULL) { |
4565 | __kmp_stg_print_str(buffer, name, value); |
4566 | } |
4567 | #endif /* KMP_DEBUG */ |
4568 | } // __kmp_stg_print_consistency_check |
4569 | |
4570 | #if USE_ITT_BUILD |
4571 | // ----------------------------------------------------------------------------- |
4572 | // KMP_ITT_PREPARE_DELAY |
4573 | |
4574 | #if USE_ITT_NOTIFY |
4575 | |
4576 | static void __kmp_stg_parse_itt_prepare_delay(char const *name, |
4577 | char const *value, void *data) { |
4578 | // Experimental code: KMP_ITT_PREPARE_DELAY specifies numbert of loop |
4579 | // iterations. |
4580 | int delay = 0; |
4581 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX, out: &delay); |
4582 | __kmp_itt_prepare_delay = delay; |
4583 | } // __kmp_str_parse_itt_prepare_delay |
4584 | |
4585 | static void __kmp_stg_print_itt_prepare_delay(kmp_str_buf_t *buffer, |
4586 | char const *name, void *data) { |
4587 | __kmp_stg_print_uint64(buffer, name, value: __kmp_itt_prepare_delay); |
4588 | |
4589 | } // __kmp_str_print_itt_prepare_delay |
4590 | |
4591 | #endif // USE_ITT_NOTIFY |
4592 | #endif /* USE_ITT_BUILD */ |
4593 | |
4594 | // ----------------------------------------------------------------------------- |
4595 | // KMP_MALLOC_POOL_INCR |
4596 | |
4597 | static void __kmp_stg_parse_malloc_pool_incr(char const *name, |
4598 | char const *value, void *data) { |
4599 | __kmp_stg_parse_size(name, value, KMP_MIN_MALLOC_POOL_INCR, |
4600 | KMP_MAX_MALLOC_POOL_INCR, NULL, out: &__kmp_malloc_pool_incr, |
4601 | factor: 1); |
4602 | } // __kmp_stg_parse_malloc_pool_incr |
4603 | |
4604 | static void __kmp_stg_print_malloc_pool_incr(kmp_str_buf_t *buffer, |
4605 | char const *name, void *data) { |
4606 | __kmp_stg_print_size(buffer, name, value: __kmp_malloc_pool_incr); |
4607 | |
4608 | } // _kmp_stg_print_malloc_pool_incr |
4609 | |
4610 | #ifdef KMP_DEBUG |
4611 | |
4612 | // ----------------------------------------------------------------------------- |
4613 | // KMP_PAR_RANGE |
4614 | |
4615 | static void __kmp_stg_parse_par_range_env(char const *name, char const *value, |
4616 | void *data) { |
4617 | __kmp_stg_parse_par_range(name, value, out_range: &__kmp_par_range, |
4618 | out_routine: __kmp_par_range_routine, out_file: __kmp_par_range_filename, |
4619 | out_lb: &__kmp_par_range_lb, out_ub: &__kmp_par_range_ub); |
4620 | } // __kmp_stg_parse_par_range_env |
4621 | |
4622 | static void __kmp_stg_print_par_range_env(kmp_str_buf_t *buffer, |
4623 | char const *name, void *data) { |
4624 | if (__kmp_par_range != 0) { |
4625 | __kmp_stg_print_str(buffer, name, value: par_range_to_print); |
4626 | } |
4627 | } // __kmp_stg_print_par_range_env |
4628 | |
4629 | #endif |
4630 | |
4631 | // ----------------------------------------------------------------------------- |
4632 | // KMP_GTID_MODE |
4633 | |
4634 | static void __kmp_stg_parse_gtid_mode(char const *name, char const *value, |
4635 | void *data) { |
4636 | // Modes: |
4637 | // 0 -- do not change default |
4638 | // 1 -- sp search |
4639 | // 2 -- use "keyed" TLS var, i.e. |
4640 | // pthread_getspecific(Linux* OS/OS X*) or TlsGetValue(Windows* OS) |
4641 | // 3 -- __declspec(thread) TLS var in tdata section |
4642 | int mode = 0; |
4643 | int max = 2; |
4644 | #ifdef KMP_TDATA_GTID |
4645 | max = 3; |
4646 | #endif /* KMP_TDATA_GTID */ |
4647 | __kmp_stg_parse_int(name, value, min: 0, max, out: &mode); |
4648 | // TODO; parse_int is not very suitable for this case. In case of overflow it |
4649 | // is better to use 0 rather that max value. |
4650 | if (mode == 0) { |
4651 | __kmp_adjust_gtid_mode = TRUE; |
4652 | } else { |
4653 | __kmp_gtid_mode = mode; |
4654 | __kmp_adjust_gtid_mode = FALSE; |
4655 | } |
4656 | } // __kmp_str_parse_gtid_mode |
4657 | |
4658 | static void __kmp_stg_print_gtid_mode(kmp_str_buf_t *buffer, char const *name, |
4659 | void *data) { |
4660 | if (__kmp_adjust_gtid_mode) { |
4661 | __kmp_stg_print_int(buffer, name, value: 0); |
4662 | } else { |
4663 | __kmp_stg_print_int(buffer, name, value: __kmp_gtid_mode); |
4664 | } |
4665 | } // __kmp_stg_print_gtid_mode |
4666 | |
4667 | // ----------------------------------------------------------------------------- |
4668 | // KMP_NUM_LOCKS_IN_BLOCK |
4669 | |
4670 | static void __kmp_stg_parse_lock_block(char const *name, char const *value, |
4671 | void *data) { |
4672 | __kmp_stg_parse_int(name, value, min: 0, KMP_INT_MAX, out: &__kmp_num_locks_in_block); |
4673 | } // __kmp_str_parse_lock_block |
4674 | |
4675 | static void __kmp_stg_print_lock_block(kmp_str_buf_t *buffer, char const *name, |
4676 | void *data) { |
4677 | __kmp_stg_print_int(buffer, name, value: __kmp_num_locks_in_block); |
4678 | } // __kmp_stg_print_lock_block |
4679 | |
4680 | // ----------------------------------------------------------------------------- |
4681 | // KMP_LOCK_KIND |
4682 | |
4683 | #if KMP_USE_DYNAMIC_LOCK |
4684 | #define KMP_STORE_LOCK_SEQ(a) (__kmp_user_lock_seq = lockseq_##a) |
4685 | #else |
4686 | #define KMP_STORE_LOCK_SEQ(a) |
4687 | #endif |
4688 | |
4689 | static void __kmp_stg_parse_lock_kind(char const *name, char const *value, |
4690 | void *data) { |
4691 | if (__kmp_init_user_locks) { |
4692 | KMP_WARNING(EnvLockWarn, name); |
4693 | return; |
4694 | } |
4695 | |
4696 | if (__kmp_str_match(target: "tas" , len: 2, data: value) || |
4697 | __kmp_str_match(target: "test and set" , len: 2, data: value) || |
4698 | __kmp_str_match(target: "test_and_set" , len: 2, data: value) || |
4699 | __kmp_str_match(target: "test-and-set" , len: 2, data: value) || |
4700 | __kmp_str_match(target: "test andset" , len: 2, data: value) || |
4701 | __kmp_str_match(target: "test_andset" , len: 2, data: value) || |
4702 | __kmp_str_match(target: "test-andset" , len: 2, data: value) || |
4703 | __kmp_str_match(target: "testand set" , len: 2, data: value) || |
4704 | __kmp_str_match(target: "testand_set" , len: 2, data: value) || |
4705 | __kmp_str_match(target: "testand-set" , len: 2, data: value) || |
4706 | __kmp_str_match(target: "testandset" , len: 2, data: value)) { |
4707 | __kmp_user_lock_kind = lk_tas; |
4708 | KMP_STORE_LOCK_SEQ(tas); |
4709 | } |
4710 | #if KMP_USE_FUTEX |
4711 | else if (__kmp_str_match(target: "futex" , len: 1, data: value)) { |
4712 | if (__kmp_futex_determine_capable()) { |
4713 | __kmp_user_lock_kind = lk_futex; |
4714 | KMP_STORE_LOCK_SEQ(futex); |
4715 | } else { |
4716 | KMP_WARNING(FutexNotSupported, name, value); |
4717 | } |
4718 | } |
4719 | #endif |
4720 | else if (__kmp_str_match(target: "ticket" , len: 2, data: value)) { |
4721 | __kmp_user_lock_kind = lk_ticket; |
4722 | KMP_STORE_LOCK_SEQ(ticket); |
4723 | } else if (__kmp_str_match(target: "queuing" , len: 1, data: value) || |
4724 | __kmp_str_match(target: "queue" , len: 1, data: value)) { |
4725 | __kmp_user_lock_kind = lk_queuing; |
4726 | KMP_STORE_LOCK_SEQ(queuing); |
4727 | } else if (__kmp_str_match(target: "drdpa ticket" , len: 1, data: value) || |
4728 | __kmp_str_match(target: "drdpa_ticket" , len: 1, data: value) || |
4729 | __kmp_str_match(target: "drdpa-ticket" , len: 1, data: value) || |
4730 | __kmp_str_match(target: "drdpaticket" , len: 1, data: value) || |
4731 | __kmp_str_match(target: "drdpa" , len: 1, data: value)) { |
4732 | __kmp_user_lock_kind = lk_drdpa; |
4733 | KMP_STORE_LOCK_SEQ(drdpa); |
4734 | } |
4735 | #if KMP_USE_ADAPTIVE_LOCKS |
4736 | else if (__kmp_str_match(target: "adaptive" , len: 1, data: value)) { |
4737 | if (__kmp_cpuinfo.flags.rtm) { // ??? Is cpuinfo available here? |
4738 | __kmp_user_lock_kind = lk_adaptive; |
4739 | KMP_STORE_LOCK_SEQ(adaptive); |
4740 | } else { |
4741 | KMP_WARNING(AdaptiveNotSupported, name, value); |
4742 | __kmp_user_lock_kind = lk_queuing; |
4743 | KMP_STORE_LOCK_SEQ(queuing); |
4744 | } |
4745 | } |
4746 | #endif // KMP_USE_ADAPTIVE_LOCKS |
4747 | #if KMP_USE_DYNAMIC_LOCK && KMP_USE_TSX |
4748 | else if (__kmp_str_match(target: "rtm_queuing" , len: 1, data: value)) { |
4749 | if (__kmp_cpuinfo.flags.rtm) { |
4750 | __kmp_user_lock_kind = lk_rtm_queuing; |
4751 | KMP_STORE_LOCK_SEQ(rtm_queuing); |
4752 | } else { |
4753 | KMP_WARNING(AdaptiveNotSupported, name, value); |
4754 | __kmp_user_lock_kind = lk_queuing; |
4755 | KMP_STORE_LOCK_SEQ(queuing); |
4756 | } |
4757 | } else if (__kmp_str_match(target: "rtm_spin" , len: 1, data: value)) { |
4758 | if (__kmp_cpuinfo.flags.rtm) { |
4759 | __kmp_user_lock_kind = lk_rtm_spin; |
4760 | KMP_STORE_LOCK_SEQ(rtm_spin); |
4761 | } else { |
4762 | KMP_WARNING(AdaptiveNotSupported, name, value); |
4763 | __kmp_user_lock_kind = lk_tas; |
4764 | KMP_STORE_LOCK_SEQ(queuing); |
4765 | } |
4766 | } else if (__kmp_str_match(target: "hle" , len: 1, data: value)) { |
4767 | __kmp_user_lock_kind = lk_hle; |
4768 | KMP_STORE_LOCK_SEQ(hle); |
4769 | } |
4770 | #endif |
4771 | else { |
4772 | KMP_WARNING(StgInvalidValue, name, value); |
4773 | } |
4774 | } |
4775 | |
4776 | static void __kmp_stg_print_lock_kind(kmp_str_buf_t *buffer, char const *name, |
4777 | void *data) { |
4778 | const char *value = NULL; |
4779 | |
4780 | switch (__kmp_user_lock_kind) { |
4781 | case lk_default: |
4782 | value = "default" ; |
4783 | break; |
4784 | |
4785 | case lk_tas: |
4786 | value = "tas" ; |
4787 | break; |
4788 | |
4789 | #if KMP_USE_FUTEX |
4790 | case lk_futex: |
4791 | value = "futex" ; |
4792 | break; |
4793 | #endif |
4794 | |
4795 | #if KMP_USE_DYNAMIC_LOCK && KMP_USE_TSX |
4796 | case lk_rtm_queuing: |
4797 | value = "rtm_queuing" ; |
4798 | break; |
4799 | |
4800 | case lk_rtm_spin: |
4801 | value = "rtm_spin" ; |
4802 | break; |
4803 | |
4804 | case lk_hle: |
4805 | value = "hle" ; |
4806 | break; |
4807 | #endif |
4808 | |
4809 | case lk_ticket: |
4810 | value = "ticket" ; |
4811 | break; |
4812 | |
4813 | case lk_queuing: |
4814 | value = "queuing" ; |
4815 | break; |
4816 | |
4817 | case lk_drdpa: |
4818 | value = "drdpa" ; |
4819 | break; |
4820 | #if KMP_USE_ADAPTIVE_LOCKS |
4821 | case lk_adaptive: |
4822 | value = "adaptive" ; |
4823 | break; |
4824 | #endif |
4825 | } |
4826 | |
4827 | if (value != NULL) { |
4828 | __kmp_stg_print_str(buffer, name, value); |
4829 | } |
4830 | } |
4831 | |
4832 | // ----------------------------------------------------------------------------- |
4833 | // KMP_SPIN_BACKOFF_PARAMS |
4834 | |
4835 | // KMP_SPIN_BACKOFF_PARAMS=max_backoff[,min_tick] (max backoff size, min tick |
4836 | // for machine pause) |
4837 | static void __kmp_stg_parse_spin_backoff_params(const char *name, |
4838 | const char *value, void *data) { |
4839 | const char *next = value; |
4840 | |
4841 | int total = 0; // Count elements that were set. It'll be used as an array size |
4842 | int prev_comma = FALSE; // For correct processing sequential commas |
4843 | int i; |
4844 | |
4845 | kmp_uint32 max_backoff = __kmp_spin_backoff_params.max_backoff; |
4846 | kmp_uint32 min_tick = __kmp_spin_backoff_params.min_tick; |
4847 | |
4848 | // Run only 3 iterations because it is enough to read two values or find a |
4849 | // syntax error |
4850 | for (i = 0; i < 3; i++) { |
4851 | SKIP_WS(next); |
4852 | |
4853 | if (*next == '\0') { |
4854 | break; |
4855 | } |
4856 | // Next character is not an integer or not a comma OR number of values > 2 |
4857 | // => end of list |
4858 | if (((*next < '0' || *next > '9') && *next != ',') || total > 2) { |
4859 | KMP_WARNING(EnvSyntaxError, name, value); |
4860 | return; |
4861 | } |
4862 | // The next character is ',' |
4863 | if (*next == ',') { |
4864 | // ',' is the first character |
4865 | if (total == 0 || prev_comma) { |
4866 | total++; |
4867 | } |
4868 | prev_comma = TRUE; |
4869 | next++; // skip ',' |
4870 | SKIP_WS(next); |
4871 | } |
4872 | // Next character is a digit |
4873 | if (*next >= '0' && *next <= '9') { |
4874 | int num; |
4875 | const char *buf = next; |
4876 | char const *msg = NULL; |
4877 | prev_comma = FALSE; |
4878 | SKIP_DIGITS(next); |
4879 | total++; |
4880 | |
4881 | const char *tmp = next; |
4882 | SKIP_WS(tmp); |
4883 | if ((*next == ' ' || *next == '\t') && (*tmp >= '0' && *tmp <= '9')) { |
4884 | KMP_WARNING(EnvSpacesNotAllowed, name, value); |
4885 | return; |
4886 | } |
4887 | |
4888 | num = __kmp_str_to_int(str: buf, sentinel: *next); |
4889 | if (num <= 0) { // The number of retries should be > 0 |
4890 | msg = KMP_I18N_STR(ValueTooSmall); |
4891 | num = 1; |
4892 | } |
4893 | if (msg != NULL) { |
4894 | // Message is not empty. Print warning. |
4895 | KMP_WARNING(ParseSizeIntWarn, name, value, msg); |
4896 | KMP_INFORM(Using_int_Value, name, num); |
4897 | } |
4898 | if (total == 1) { |
4899 | max_backoff = num; |
4900 | } else if (total == 2) { |
4901 | min_tick = num; |
4902 | } |
4903 | } |
4904 | } |
4905 | KMP_DEBUG_ASSERT(total > 0); |
4906 | if (total <= 0) { |
4907 | KMP_WARNING(EnvSyntaxError, name, value); |
4908 | return; |
4909 | } |
4910 | __kmp_spin_backoff_params.max_backoff = max_backoff; |
4911 | __kmp_spin_backoff_params.min_tick = min_tick; |
4912 | } |
4913 | |
4914 | static void __kmp_stg_print_spin_backoff_params(kmp_str_buf_t *buffer, |
4915 | char const *name, void *data) { |
4916 | if (__kmp_env_format) { |
4917 | KMP_STR_BUF_PRINT_NAME_EX(name); |
4918 | } else { |
4919 | __kmp_str_buf_print(buffer, format: " %s='" , name); |
4920 | } |
4921 | __kmp_str_buf_print(buffer, format: "%d,%d'\n" , __kmp_spin_backoff_params.max_backoff, |
4922 | __kmp_spin_backoff_params.min_tick); |
4923 | } |
4924 | |
4925 | #if KMP_USE_ADAPTIVE_LOCKS |
4926 | |
4927 | // ----------------------------------------------------------------------------- |
4928 | // KMP_ADAPTIVE_LOCK_PROPS, KMP_SPECULATIVE_STATSFILE |
4929 | |
4930 | // Parse out values for the tunable parameters from a string of the form |
4931 | // KMP_ADAPTIVE_LOCK_PROPS=max_soft_retries[,max_badness] |
4932 | static void __kmp_stg_parse_adaptive_lock_props(const char *name, |
4933 | const char *value, void *data) { |
4934 | int max_retries = 0; |
4935 | int max_badness = 0; |
4936 | |
4937 | const char *next = value; |
4938 | |
4939 | int total = 0; // Count elements that were set. It'll be used as an array size |
4940 | int prev_comma = FALSE; // For correct processing sequential commas |
4941 | int i; |
4942 | |
4943 | // Save values in the structure __kmp_speculative_backoff_params |
4944 | // Run only 3 iterations because it is enough to read two values or find a |
4945 | // syntax error |
4946 | for (i = 0; i < 3; i++) { |
4947 | SKIP_WS(next); |
4948 | |
4949 | if (*next == '\0') { |
4950 | break; |
4951 | } |
4952 | // Next character is not an integer or not a comma OR number of values > 2 |
4953 | // => end of list |
4954 | if (((*next < '0' || *next > '9') && *next != ',') || total > 2) { |
4955 | KMP_WARNING(EnvSyntaxError, name, value); |
4956 | return; |
4957 | } |
4958 | // The next character is ',' |
4959 | if (*next == ',') { |
4960 | // ',' is the first character |
4961 | if (total == 0 || prev_comma) { |
4962 | total++; |
4963 | } |
4964 | prev_comma = TRUE; |
4965 | next++; // skip ',' |
4966 | SKIP_WS(next); |
4967 | } |
4968 | // Next character is a digit |
4969 | if (*next >= '0' && *next <= '9') { |
4970 | int num; |
4971 | const char *buf = next; |
4972 | char const *msg = NULL; |
4973 | prev_comma = FALSE; |
4974 | SKIP_DIGITS(next); |
4975 | total++; |
4976 | |
4977 | const char *tmp = next; |
4978 | SKIP_WS(tmp); |
4979 | if ((*next == ' ' || *next == '\t') && (*tmp >= '0' && *tmp <= '9')) { |
4980 | KMP_WARNING(EnvSpacesNotAllowed, name, value); |
4981 | return; |
4982 | } |
4983 | |
4984 | num = __kmp_str_to_int(str: buf, sentinel: *next); |
4985 | if (num < 0) { // The number of retries should be >= 0 |
4986 | msg = KMP_I18N_STR(ValueTooSmall); |
4987 | num = 1; |
4988 | } |
4989 | if (msg != NULL) { |
4990 | // Message is not empty. Print warning. |
4991 | KMP_WARNING(ParseSizeIntWarn, name, value, msg); |
4992 | KMP_INFORM(Using_int_Value, name, num); |
4993 | } |
4994 | if (total == 1) { |
4995 | max_retries = num; |
4996 | } else if (total == 2) { |
4997 | max_badness = num; |
4998 | } |
4999 | } |
5000 | } |
5001 | KMP_DEBUG_ASSERT(total > 0); |
5002 | if (total <= 0) { |
5003 | KMP_WARNING(EnvSyntaxError, name, value); |
5004 | return; |
5005 | } |
5006 | __kmp_adaptive_backoff_params.max_soft_retries = max_retries; |
5007 | __kmp_adaptive_backoff_params.max_badness = max_badness; |
5008 | } |
5009 | |
5010 | static void __kmp_stg_print_adaptive_lock_props(kmp_str_buf_t *buffer, |
5011 | char const *name, void *data) { |
5012 | if (__kmp_env_format) { |
5013 | KMP_STR_BUF_PRINT_NAME_EX(name); |
5014 | } else { |
5015 | __kmp_str_buf_print(buffer, format: " %s='" , name); |
5016 | } |
5017 | __kmp_str_buf_print(buffer, format: "%d,%d'\n" , |
5018 | __kmp_adaptive_backoff_params.max_soft_retries, |
5019 | __kmp_adaptive_backoff_params.max_badness); |
5020 | } // __kmp_stg_print_adaptive_lock_props |
5021 | |
5022 | #if KMP_DEBUG_ADAPTIVE_LOCKS |
5023 | |
5024 | static void __kmp_stg_parse_speculative_statsfile(char const *name, |
5025 | char const *value, |
5026 | void *data) { |
5027 | __kmp_stg_parse_file(name, value, "" , |
5028 | CCAST(char **, &__kmp_speculative_statsfile)); |
5029 | } // __kmp_stg_parse_speculative_statsfile |
5030 | |
5031 | static void __kmp_stg_print_speculative_statsfile(kmp_str_buf_t *buffer, |
5032 | char const *name, |
5033 | void *data) { |
5034 | if (__kmp_str_match("-" , 0, __kmp_speculative_statsfile)) { |
5035 | __kmp_stg_print_str(buffer, name, "stdout" ); |
5036 | } else { |
5037 | __kmp_stg_print_str(buffer, name, __kmp_speculative_statsfile); |
5038 | } |
5039 | |
5040 | } // __kmp_stg_print_speculative_statsfile |
5041 | |
5042 | #endif // KMP_DEBUG_ADAPTIVE_LOCKS |
5043 | |
5044 | #endif // KMP_USE_ADAPTIVE_LOCKS |
5045 | |
5046 | // ----------------------------------------------------------------------------- |
5047 | // KMP_HW_SUBSET (was KMP_PLACE_THREADS) |
5048 | // 2s16c,2t => 2S16C,2T => 2S16C \0 2T |
5049 | |
5050 | // Return KMP_HW_SUBSET preferred hardware type in case a token is ambiguously |
5051 | // short. The original KMP_HW_SUBSET environment variable had single letters: |
5052 | // s, c, t for sockets, cores, threads repsectively. |
5053 | static kmp_hw_t __kmp_hw_subset_break_tie(const kmp_hw_t *possible, |
5054 | size_t num_possible) { |
5055 | for (size_t i = 0; i < num_possible; ++i) { |
5056 | if (possible[i] == KMP_HW_THREAD) |
5057 | return KMP_HW_THREAD; |
5058 | else if (possible[i] == KMP_HW_CORE) |
5059 | return KMP_HW_CORE; |
5060 | else if (possible[i] == KMP_HW_SOCKET) |
5061 | return KMP_HW_SOCKET; |
5062 | } |
5063 | return KMP_HW_UNKNOWN; |
5064 | } |
5065 | |
5066 | // Return hardware type from string or HW_UNKNOWN if string cannot be parsed |
5067 | // This algorithm is very forgiving to the user in that, the instant it can |
5068 | // reduce the search space to one, it assumes that is the topology level the |
5069 | // user wanted, even if it is misspelled later in the token. |
5070 | static kmp_hw_t __kmp_stg_parse_hw_subset_name(char const *token) { |
5071 | size_t index, num_possible, token_length; |
5072 | kmp_hw_t possible[KMP_HW_LAST]; |
5073 | const char *end; |
5074 | |
5075 | // Find the end of the hardware token string |
5076 | end = token; |
5077 | token_length = 0; |
5078 | while (isalnum(*end) || *end == '_') { |
5079 | token_length++; |
5080 | end++; |
5081 | } |
5082 | |
5083 | // Set the possibilities to all hardware types |
5084 | num_possible = 0; |
5085 | KMP_FOREACH_HW_TYPE(type) { possible[num_possible++] = type; } |
5086 | |
5087 | // Eliminate hardware types by comparing the front of the token |
5088 | // with hardware names |
5089 | // In most cases, the first letter in the token will indicate exactly |
5090 | // which hardware type is parsed, e.g., 'C' = Core |
5091 | index = 0; |
5092 | while (num_possible > 1 && index < token_length) { |
5093 | size_t n = num_possible; |
5094 | char token_char = (char)toupper(c: token[index]); |
5095 | for (size_t i = 0; i < n; ++i) { |
5096 | const char *s; |
5097 | kmp_hw_t type = possible[i]; |
5098 | s = __kmp_hw_get_keyword(type, plural: false); |
5099 | if (index < KMP_STRLEN(s: s)) { |
5100 | char c = (char)toupper(c: s[index]); |
5101 | // Mark hardware types for removal when the characters do not match |
5102 | if (c != token_char) { |
5103 | possible[i] = KMP_HW_UNKNOWN; |
5104 | num_possible--; |
5105 | } |
5106 | } |
5107 | } |
5108 | // Remove hardware types that this token cannot be |
5109 | size_t start = 0; |
5110 | for (size_t i = 0; i < n; ++i) { |
5111 | if (possible[i] != KMP_HW_UNKNOWN) { |
5112 | kmp_hw_t temp = possible[i]; |
5113 | possible[i] = possible[start]; |
5114 | possible[start] = temp; |
5115 | start++; |
5116 | } |
5117 | } |
5118 | KMP_ASSERT(start == num_possible); |
5119 | index++; |
5120 | } |
5121 | |
5122 | // Attempt to break a tie if user has very short token |
5123 | // (e.g., is 'T' tile or thread?) |
5124 | if (num_possible > 1) |
5125 | return __kmp_hw_subset_break_tie(possible, num_possible); |
5126 | if (num_possible == 1) |
5127 | return possible[0]; |
5128 | return KMP_HW_UNKNOWN; |
5129 | } |
5130 | |
5131 | // The longest observable sequence of items can only be HW_LAST length |
5132 | // The input string is usually short enough, let's use 512 limit for now |
5133 | #define MAX_T_LEVEL KMP_HW_LAST |
5134 | #define MAX_STR_LEN 512 |
5135 | static void __kmp_stg_parse_hw_subset(char const *name, char const *value, |
5136 | void *data) { |
5137 | // Value example: 1s,5c@3,2T |
5138 | // Which means "use 1 socket, 5 cores with offset 3, 2 threads per core" |
5139 | kmp_setting_t **rivals = (kmp_setting_t **)data; |
5140 | if (strcmp(s1: name, s2: "KMP_PLACE_THREADS" ) == 0) { |
5141 | KMP_INFORM(EnvVarDeprecated, name, "KMP_HW_SUBSET" ); |
5142 | } |
5143 | if (__kmp_stg_check_rivals(name, value, rivals)) { |
5144 | return; |
5145 | } |
5146 | |
5147 | char *components[MAX_T_LEVEL]; |
5148 | char const *digits = "0123456789" ; |
5149 | char input[MAX_STR_LEN]; |
5150 | size_t len = 0, mlen = MAX_STR_LEN; |
5151 | int level = 0; |
5152 | bool absolute = false; |
5153 | // Canonicalize the string (remove spaces, unify delimiters, etc.) |
5154 | char *pos = CCAST(char *, value); |
5155 | while (*pos && mlen) { |
5156 | if (*pos != ' ') { // skip spaces |
5157 | if (len == 0 && *pos == ':') { |
5158 | absolute = true; |
5159 | } else { |
5160 | input[len] = (char)(toupper(c: *pos)); |
5161 | if (input[len] == 'X') |
5162 | input[len] = ','; // unify delimiters of levels |
5163 | if (input[len] == 'O' && strchr(s: digits, c: *(pos + 1))) |
5164 | input[len] = '@'; // unify delimiters of offset |
5165 | len++; |
5166 | } |
5167 | } |
5168 | mlen--; |
5169 | pos++; |
5170 | } |
5171 | if (len == 0 || mlen == 0) { |
5172 | goto err; // contents is either empty or too long |
5173 | } |
5174 | input[len] = '\0'; |
5175 | // Split by delimiter |
5176 | pos = input; |
5177 | components[level++] = pos; |
5178 | while ((pos = strchr(s: pos, c: ','))) { |
5179 | if (level >= MAX_T_LEVEL) |
5180 | goto err; // too many components provided |
5181 | *pos = '\0'; // modify input and avoid more copying |
5182 | components[level++] = ++pos; // expect something after "," |
5183 | } |
5184 | |
5185 | __kmp_hw_subset = kmp_hw_subset_t::allocate(); |
5186 | if (absolute) |
5187 | __kmp_hw_subset->set_absolute(); |
5188 | |
5189 | // Check each component |
5190 | for (int i = 0; i < level; ++i) { |
5191 | int core_level = 0; |
5192 | char *core_components[MAX_T_LEVEL]; |
5193 | // Split possible core components by '&' delimiter |
5194 | pos = components[i]; |
5195 | core_components[core_level++] = pos; |
5196 | while ((pos = strchr(s: pos, c: '&'))) { |
5197 | if (core_level >= MAX_T_LEVEL) |
5198 | goto err; // too many different core types |
5199 | *pos = '\0'; // modify input and avoid more copying |
5200 | core_components[core_level++] = ++pos; // expect something after '&' |
5201 | } |
5202 | |
5203 | for (int j = 0; j < core_level; ++j) { |
5204 | char *offset_ptr; |
5205 | char *attr_ptr; |
5206 | int offset = 0; |
5207 | kmp_hw_attr_t attr; |
5208 | int num; |
5209 | // components may begin with an optional count of the number of resources |
5210 | if (isdigit(*core_components[j])) { |
5211 | num = atoi(nptr: core_components[j]); |
5212 | if (num <= 0) { |
5213 | goto err; // only positive integers are valid for count |
5214 | } |
5215 | pos = core_components[j] + strspn(s: core_components[j], accept: digits); |
5216 | } else if (*core_components[j] == '*') { |
5217 | num = kmp_hw_subset_t::USE_ALL; |
5218 | pos = core_components[j] + 1; |
5219 | } else { |
5220 | num = kmp_hw_subset_t::USE_ALL; |
5221 | pos = core_components[j]; |
5222 | } |
5223 | |
5224 | offset_ptr = strchr(s: core_components[j], c: '@'); |
5225 | attr_ptr = strchr(s: core_components[j], c: ':'); |
5226 | |
5227 | if (offset_ptr) { |
5228 | offset = atoi(nptr: offset_ptr + 1); // save offset |
5229 | *offset_ptr = '\0'; // cut the offset from the component |
5230 | } |
5231 | if (attr_ptr) { |
5232 | attr.clear(); |
5233 | // save the attribute |
5234 | #if KMP_ARCH_X86 || KMP_ARCH_X86_64 |
5235 | if (__kmp_str_match(target: "intel_core" , len: -1, data: attr_ptr + 1)) { |
5236 | attr.set_core_type(KMP_HW_CORE_TYPE_CORE); |
5237 | } else if (__kmp_str_match(target: "intel_atom" , len: -1, data: attr_ptr + 1)) { |
5238 | attr.set_core_type(KMP_HW_CORE_TYPE_ATOM); |
5239 | } else |
5240 | #endif |
5241 | if (__kmp_str_match(target: "eff" , len: 3, data: attr_ptr + 1)) { |
5242 | const char *number = attr_ptr + 1; |
5243 | // skip the eff[iciency] token |
5244 | while (isalpha(*number)) |
5245 | number++; |
5246 | if (!isdigit(*number)) { |
5247 | goto err; |
5248 | } |
5249 | int efficiency = atoi(nptr: number); |
5250 | attr.set_core_eff(efficiency); |
5251 | } else { |
5252 | goto err; |
5253 | } |
5254 | *attr_ptr = '\0'; // cut the attribute from the component |
5255 | } |
5256 | // detect the component type |
5257 | kmp_hw_t type = __kmp_stg_parse_hw_subset_name(token: pos); |
5258 | if (type == KMP_HW_UNKNOWN) { |
5259 | goto err; |
5260 | } |
5261 | // Only the core type can have attributes |
5262 | if (attr && type != KMP_HW_CORE) |
5263 | goto err; |
5264 | // Must allow core be specified more than once |
5265 | if (type != KMP_HW_CORE && __kmp_hw_subset->specified(type)) { |
5266 | goto err; |
5267 | } |
5268 | __kmp_hw_subset->push_back(num, type, offset, attr); |
5269 | } |
5270 | } |
5271 | return; |
5272 | err: |
5273 | KMP_WARNING(AffHWSubsetInvalid, name, value); |
5274 | if (__kmp_hw_subset) { |
5275 | kmp_hw_subset_t::deallocate(subset: __kmp_hw_subset); |
5276 | __kmp_hw_subset = nullptr; |
5277 | } |
5278 | return; |
5279 | } |
5280 | |
5281 | static void __kmp_stg_print_hw_subset(kmp_str_buf_t *buffer, char const *name, |
5282 | void *data) { |
5283 | kmp_str_buf_t buf; |
5284 | int depth; |
5285 | if (!__kmp_hw_subset) |
5286 | return; |
5287 | __kmp_str_buf_init(&buf); |
5288 | if (__kmp_env_format) |
5289 | KMP_STR_BUF_PRINT_NAME_EX(name); |
5290 | else |
5291 | __kmp_str_buf_print(buffer, format: " %s='" , name); |
5292 | |
5293 | depth = __kmp_hw_subset->get_depth(); |
5294 | for (int i = 0; i < depth; ++i) { |
5295 | const auto &item = __kmp_hw_subset->at(index: i); |
5296 | if (i > 0) |
5297 | __kmp_str_buf_print(buffer: &buf, format: "%c" , ','); |
5298 | for (int j = 0; j < item.num_attrs; ++j) { |
5299 | __kmp_str_buf_print(buffer: &buf, format: "%s%d%s" , (j > 0 ? "&" : "" ), item.num[j], |
5300 | __kmp_hw_get_keyword(type: item.type)); |
5301 | if (item.attr[j].is_core_type_valid()) |
5302 | __kmp_str_buf_print( |
5303 | buffer: &buf, format: ":%s" , |
5304 | __kmp_hw_get_core_type_keyword(type: item.attr[j].get_core_type())); |
5305 | if (item.attr[j].is_core_eff_valid()) |
5306 | __kmp_str_buf_print(buffer: &buf, format: ":eff%d" , item.attr[j].get_core_eff()); |
5307 | if (item.offset[j]) |
5308 | __kmp_str_buf_print(buffer: &buf, format: "@%d" , item.offset[j]); |
5309 | } |
5310 | } |
5311 | __kmp_str_buf_print(buffer, format: "%s'\n" , buf.str); |
5312 | __kmp_str_buf_free(buffer: &buf); |
5313 | } |
5314 | |
5315 | #if USE_ITT_BUILD |
5316 | // ----------------------------------------------------------------------------- |
5317 | // KMP_FORKJOIN_FRAMES |
5318 | |
5319 | static void __kmp_stg_parse_forkjoin_frames(char const *name, char const *value, |
5320 | void *data) { |
5321 | __kmp_stg_parse_bool(name, value, out: &__kmp_forkjoin_frames); |
5322 | } // __kmp_stg_parse_forkjoin_frames |
5323 | |
5324 | static void __kmp_stg_print_forkjoin_frames(kmp_str_buf_t *buffer, |
5325 | char const *name, void *data) { |
5326 | __kmp_stg_print_bool(buffer, name, value: __kmp_forkjoin_frames); |
5327 | } // __kmp_stg_print_forkjoin_frames |
5328 | |
5329 | // ----------------------------------------------------------------------------- |
5330 | // KMP_FORKJOIN_FRAMES_MODE |
5331 | |
5332 | static void __kmp_stg_parse_forkjoin_frames_mode(char const *name, |
5333 | char const *value, |
5334 | void *data) { |
5335 | __kmp_stg_parse_int(name, value, min: 0, max: 3, out: &__kmp_forkjoin_frames_mode); |
5336 | } // __kmp_stg_parse_forkjoin_frames |
5337 | |
5338 | static void __kmp_stg_print_forkjoin_frames_mode(kmp_str_buf_t *buffer, |
5339 | char const *name, void *data) { |
5340 | __kmp_stg_print_int(buffer, name, value: __kmp_forkjoin_frames_mode); |
5341 | } // __kmp_stg_print_forkjoin_frames |
5342 | #endif /* USE_ITT_BUILD */ |
5343 | |
5344 | // ----------------------------------------------------------------------------- |
5345 | // KMP_ENABLE_TASK_THROTTLING |
5346 | |
5347 | static void __kmp_stg_parse_task_throttling(char const *name, char const *value, |
5348 | void *data) { |
5349 | __kmp_stg_parse_bool(name, value, out: &__kmp_enable_task_throttling); |
5350 | } // __kmp_stg_parse_task_throttling |
5351 | |
5352 | static void __kmp_stg_print_task_throttling(kmp_str_buf_t *buffer, |
5353 | char const *name, void *data) { |
5354 | __kmp_stg_print_bool(buffer, name, value: __kmp_enable_task_throttling); |
5355 | } // __kmp_stg_print_task_throttling |
5356 | |
5357 | #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT |
5358 | // ----------------------------------------------------------------------------- |
5359 | // KMP_USER_LEVEL_MWAIT |
5360 | |
5361 | static void __kmp_stg_parse_user_level_mwait(char const *name, |
5362 | char const *value, void *data) { |
5363 | __kmp_stg_parse_bool(name, value, out: &__kmp_user_level_mwait); |
5364 | } // __kmp_stg_parse_user_level_mwait |
5365 | |
5366 | static void __kmp_stg_print_user_level_mwait(kmp_str_buf_t *buffer, |
5367 | char const *name, void *data) { |
5368 | __kmp_stg_print_bool(buffer, name, value: __kmp_user_level_mwait); |
5369 | } // __kmp_stg_print_user_level_mwait |
5370 | |
5371 | // ----------------------------------------------------------------------------- |
5372 | // KMP_MWAIT_HINTS |
5373 | |
5374 | static void __kmp_stg_parse_mwait_hints(char const *name, char const *value, |
5375 | void *data) { |
5376 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX, out: &__kmp_mwait_hints); |
5377 | } // __kmp_stg_parse_mwait_hints |
5378 | |
5379 | static void __kmp_stg_print_mwait_hints(kmp_str_buf_t *buffer, char const *name, |
5380 | void *data) { |
5381 | __kmp_stg_print_int(buffer, name, value: __kmp_mwait_hints); |
5382 | } // __kmp_stg_print_mwait_hints |
5383 | |
5384 | #endif // KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT |
5385 | |
5386 | #if KMP_HAVE_UMWAIT |
5387 | // ----------------------------------------------------------------------------- |
5388 | // KMP_TPAUSE |
5389 | // 0 = don't use TPAUSE, 1 = use C0.1 state, 2 = use C0.2 state |
5390 | |
5391 | static void __kmp_stg_parse_tpause(char const *name, char const *value, |
5392 | void *data) { |
5393 | __kmp_stg_parse_int(name, value, min: 0, INT_MAX, out: &__kmp_tpause_state); |
5394 | if (__kmp_tpause_state != 0) { |
5395 | // The actual hint passed to tpause is: 0 for C0.2 and 1 for C0.1 |
5396 | if (__kmp_tpause_state == 2) // use C0.2 |
5397 | __kmp_tpause_hint = 0; // default was set to 1 for C0.1 |
5398 | } |
5399 | } // __kmp_stg_parse_tpause |
5400 | |
5401 | static void __kmp_stg_print_tpause(kmp_str_buf_t *buffer, char const *name, |
5402 | void *data) { |
5403 | __kmp_stg_print_int(buffer, name, value: __kmp_tpause_state); |
5404 | } // __kmp_stg_print_tpause |
5405 | #endif // KMP_HAVE_UMWAIT |
5406 | |
5407 | // ----------------------------------------------------------------------------- |
5408 | // OMP_DISPLAY_ENV |
5409 | |
5410 | static void __kmp_stg_parse_omp_display_env(char const *name, char const *value, |
5411 | void *data) { |
5412 | if (__kmp_str_match(target: "VERBOSE" , len: 1, data: value)) { |
5413 | __kmp_display_env_verbose = TRUE; |
5414 | } else { |
5415 | __kmp_stg_parse_bool(name, value, out: &__kmp_display_env); |
5416 | } |
5417 | } // __kmp_stg_parse_omp_display_env |
5418 | |
5419 | static void __kmp_stg_print_omp_display_env(kmp_str_buf_t *buffer, |
5420 | char const *name, void *data) { |
5421 | if (__kmp_display_env_verbose) { |
5422 | __kmp_stg_print_str(buffer, name, value: "VERBOSE" ); |
5423 | } else { |
5424 | __kmp_stg_print_bool(buffer, name, value: __kmp_display_env); |
5425 | } |
5426 | } // __kmp_stg_print_omp_display_env |
5427 | |
5428 | static void __kmp_stg_parse_omp_cancellation(char const *name, |
5429 | char const *value, void *data) { |
5430 | if (TCR_4(__kmp_init_parallel)) { |
5431 | KMP_WARNING(EnvParallelWarn, name); |
5432 | return; |
5433 | } // read value before first parallel only |
5434 | __kmp_stg_parse_bool(name, value, out: &__kmp_omp_cancellation); |
5435 | } // __kmp_stg_parse_omp_cancellation |
5436 | |
5437 | static void __kmp_stg_print_omp_cancellation(kmp_str_buf_t *buffer, |
5438 | char const *name, void *data) { |
5439 | __kmp_stg_print_bool(buffer, name, value: __kmp_omp_cancellation); |
5440 | } // __kmp_stg_print_omp_cancellation |
5441 | |
5442 | #if OMPT_SUPPORT |
5443 | int __kmp_tool = 1; |
5444 | |
5445 | static void __kmp_stg_parse_omp_tool(char const *name, char const *value, |
5446 | void *data) { |
5447 | __kmp_stg_parse_bool(name, value, out: &__kmp_tool); |
5448 | } // __kmp_stg_parse_omp_tool |
5449 | |
5450 | static void __kmp_stg_print_omp_tool(kmp_str_buf_t *buffer, char const *name, |
5451 | void *data) { |
5452 | if (__kmp_env_format) { |
5453 | KMP_STR_BUF_PRINT_BOOL_EX(name, __kmp_tool, "enabled" , "disabled" ); |
5454 | } else { |
5455 | __kmp_str_buf_print(buffer, format: " %s=%s\n" , name, |
5456 | __kmp_tool ? "enabled" : "disabled" ); |
5457 | } |
5458 | } // __kmp_stg_print_omp_tool |
5459 | |
5460 | char *__kmp_tool_libraries = NULL; |
5461 | |
5462 | static void __kmp_stg_parse_omp_tool_libraries(char const *name, |
5463 | char const *value, void *data) { |
5464 | __kmp_stg_parse_str(name, value, out: &__kmp_tool_libraries); |
5465 | } // __kmp_stg_parse_omp_tool_libraries |
5466 | |
5467 | static void __kmp_stg_print_omp_tool_libraries(kmp_str_buf_t *buffer, |
5468 | char const *name, void *data) { |
5469 | if (__kmp_tool_libraries) |
5470 | __kmp_stg_print_str(buffer, name, value: __kmp_tool_libraries); |
5471 | else { |
5472 | if (__kmp_env_format) { |
5473 | KMP_STR_BUF_PRINT_NAME; |
5474 | } else { |
5475 | __kmp_str_buf_print(buffer, format: " %s" , name); |
5476 | } |
5477 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
5478 | } |
5479 | } // __kmp_stg_print_omp_tool_libraries |
5480 | |
5481 | char *__kmp_tool_verbose_init = NULL; |
5482 | |
5483 | static void __kmp_stg_parse_omp_tool_verbose_init(char const *name, |
5484 | char const *value, |
5485 | void *data) { |
5486 | __kmp_stg_parse_str(name, value, out: &__kmp_tool_verbose_init); |
5487 | } // __kmp_stg_parse_omp_tool_libraries |
5488 | |
5489 | static void __kmp_stg_print_omp_tool_verbose_init(kmp_str_buf_t *buffer, |
5490 | char const *name, |
5491 | void *data) { |
5492 | if (__kmp_tool_verbose_init) |
5493 | __kmp_stg_print_str(buffer, name, value: __kmp_tool_verbose_init); |
5494 | else { |
5495 | if (__kmp_env_format) { |
5496 | KMP_STR_BUF_PRINT_NAME; |
5497 | } else { |
5498 | __kmp_str_buf_print(buffer, format: " %s" , name); |
5499 | } |
5500 | __kmp_str_buf_print(buffer, ": %s\n" , KMP_I18N_STR(NotDefined)); |
5501 | } |
5502 | } // __kmp_stg_print_omp_tool_verbose_init |
5503 | |
5504 | #endif |
5505 | |
5506 | // Table. |
5507 | |
5508 | static kmp_setting_t __kmp_stg_table[] = { |
5509 | |
5510 | {.name: "KMP_ALL_THREADS" , .parse: __kmp_stg_parse_device_thread_limit, NULL, NULL, .set: 0, .defined: 0}, |
5511 | {.name: "KMP_BLOCKTIME" , .parse: __kmp_stg_parse_blocktime, .print: __kmp_stg_print_blocktime, |
5512 | NULL, .set: 0, .defined: 0}, |
5513 | {.name: "KMP_USE_YIELD" , .parse: __kmp_stg_parse_use_yield, .print: __kmp_stg_print_use_yield, |
5514 | NULL, .set: 0, .defined: 0}, |
5515 | {.name: "KMP_DUPLICATE_LIB_OK" , .parse: __kmp_stg_parse_duplicate_lib_ok, |
5516 | .print: __kmp_stg_print_duplicate_lib_ok, NULL, .set: 0, .defined: 0}, |
5517 | {.name: "KMP_LIBRARY" , .parse: __kmp_stg_parse_wait_policy, .print: __kmp_stg_print_wait_policy, |
5518 | NULL, .set: 0, .defined: 0}, |
5519 | {.name: "KMP_DEVICE_THREAD_LIMIT" , .parse: __kmp_stg_parse_device_thread_limit, |
5520 | .print: __kmp_stg_print_device_thread_limit, NULL, .set: 0, .defined: 0}, |
5521 | #if KMP_USE_MONITOR |
5522 | {"KMP_MONITOR_STACKSIZE" , __kmp_stg_parse_monitor_stacksize, |
5523 | __kmp_stg_print_monitor_stacksize, NULL, 0, 0}, |
5524 | #endif |
5525 | {.name: "KMP_SETTINGS" , .parse: __kmp_stg_parse_settings, .print: __kmp_stg_print_settings, NULL, |
5526 | .set: 0, .defined: 0}, |
5527 | {.name: "KMP_STACKOFFSET" , .parse: __kmp_stg_parse_stackoffset, |
5528 | .print: __kmp_stg_print_stackoffset, NULL, .set: 0, .defined: 0}, |
5529 | {.name: "KMP_STACKSIZE" , .parse: __kmp_stg_parse_stacksize, .print: __kmp_stg_print_stacksize, |
5530 | NULL, .set: 0, .defined: 0}, |
5531 | {.name: "KMP_STACKPAD" , .parse: __kmp_stg_parse_stackpad, .print: __kmp_stg_print_stackpad, NULL, |
5532 | .set: 0, .defined: 0}, |
5533 | {.name: "KMP_VERSION" , .parse: __kmp_stg_parse_version, .print: __kmp_stg_print_version, NULL, .set: 0, |
5534 | .defined: 0}, |
5535 | {.name: "KMP_WARNINGS" , .parse: __kmp_stg_parse_warnings, .print: __kmp_stg_print_warnings, NULL, |
5536 | .set: 0, .defined: 0}, |
5537 | |
5538 | {.name: "KMP_NESTING_MODE" , .parse: __kmp_stg_parse_nesting_mode, |
5539 | .print: __kmp_stg_print_nesting_mode, NULL, .set: 0, .defined: 0}, |
5540 | {.name: "OMP_NESTED" , .parse: __kmp_stg_parse_nested, .print: __kmp_stg_print_nested, NULL, .set: 0, .defined: 0}, |
5541 | {.name: "OMP_NUM_THREADS" , .parse: __kmp_stg_parse_num_threads, |
5542 | .print: __kmp_stg_print_num_threads, NULL, .set: 0, .defined: 0}, |
5543 | {.name: "OMP_STACKSIZE" , .parse: __kmp_stg_parse_stacksize, .print: __kmp_stg_print_stacksize, |
5544 | NULL, .set: 0, .defined: 0}, |
5545 | |
5546 | {.name: "KMP_TASKING" , .parse: __kmp_stg_parse_tasking, .print: __kmp_stg_print_tasking, NULL, .set: 0, |
5547 | .defined: 0}, |
5548 | {.name: "KMP_TASK_STEALING_CONSTRAINT" , .parse: __kmp_stg_parse_task_stealing, |
5549 | .print: __kmp_stg_print_task_stealing, NULL, .set: 0, .defined: 0}, |
5550 | {.name: "OMP_MAX_ACTIVE_LEVELS" , .parse: __kmp_stg_parse_max_active_levels, |
5551 | .print: __kmp_stg_print_max_active_levels, NULL, .set: 0, .defined: 0}, |
5552 | {.name: "OMP_DEFAULT_DEVICE" , .parse: __kmp_stg_parse_default_device, |
5553 | .print: __kmp_stg_print_default_device, NULL, .set: 0, .defined: 0}, |
5554 | {.name: "OMP_TARGET_OFFLOAD" , .parse: __kmp_stg_parse_target_offload, |
5555 | .print: __kmp_stg_print_target_offload, NULL, .set: 0, .defined: 0}, |
5556 | {.name: "OMP_MAX_TASK_PRIORITY" , .parse: __kmp_stg_parse_max_task_priority, |
5557 | .print: __kmp_stg_print_max_task_priority, NULL, .set: 0, .defined: 0}, |
5558 | {.name: "KMP_TASKLOOP_MIN_TASKS" , .parse: __kmp_stg_parse_taskloop_min_tasks, |
5559 | .print: __kmp_stg_print_taskloop_min_tasks, NULL, .set: 0, .defined: 0}, |
5560 | {.name: "OMP_THREAD_LIMIT" , .parse: __kmp_stg_parse_thread_limit, |
5561 | .print: __kmp_stg_print_thread_limit, NULL, .set: 0, .defined: 0}, |
5562 | {.name: "KMP_TEAMS_THREAD_LIMIT" , .parse: __kmp_stg_parse_teams_thread_limit, |
5563 | .print: __kmp_stg_print_teams_thread_limit, NULL, .set: 0, .defined: 0}, |
5564 | {.name: "OMP_NUM_TEAMS" , .parse: __kmp_stg_parse_nteams, .print: __kmp_stg_print_nteams, NULL, .set: 0, |
5565 | .defined: 0}, |
5566 | {.name: "OMP_TEAMS_THREAD_LIMIT" , .parse: __kmp_stg_parse_teams_th_limit, |
5567 | .print: __kmp_stg_print_teams_th_limit, NULL, .set: 0, .defined: 0}, |
5568 | {.name: "OMP_WAIT_POLICY" , .parse: __kmp_stg_parse_wait_policy, |
5569 | .print: __kmp_stg_print_wait_policy, NULL, .set: 0, .defined: 0}, |
5570 | {.name: "KMP_DISP_NUM_BUFFERS" , .parse: __kmp_stg_parse_disp_buffers, |
5571 | .print: __kmp_stg_print_disp_buffers, NULL, .set: 0, .defined: 0}, |
5572 | #if KMP_NESTED_HOT_TEAMS |
5573 | {.name: "KMP_HOT_TEAMS_MAX_LEVEL" , .parse: __kmp_stg_parse_hot_teams_level, |
5574 | .print: __kmp_stg_print_hot_teams_level, NULL, .set: 0, .defined: 0}, |
5575 | {.name: "KMP_HOT_TEAMS_MODE" , .parse: __kmp_stg_parse_hot_teams_mode, |
5576 | .print: __kmp_stg_print_hot_teams_mode, NULL, .set: 0, .defined: 0}, |
5577 | #endif // KMP_NESTED_HOT_TEAMS |
5578 | |
5579 | #if KMP_HANDLE_SIGNALS |
5580 | {.name: "KMP_HANDLE_SIGNALS" , .parse: __kmp_stg_parse_handle_signals, |
5581 | .print: __kmp_stg_print_handle_signals, NULL, .set: 0, .defined: 0}, |
5582 | #endif |
5583 | |
5584 | #if KMP_ARCH_X86 || KMP_ARCH_X86_64 |
5585 | {.name: "KMP_INHERIT_FP_CONTROL" , .parse: __kmp_stg_parse_inherit_fp_control, |
5586 | .print: __kmp_stg_print_inherit_fp_control, NULL, .set: 0, .defined: 0}, |
5587 | #endif /* KMP_ARCH_X86 || KMP_ARCH_X86_64 */ |
5588 | |
5589 | #ifdef KMP_GOMP_COMPAT |
5590 | {.name: "GOMP_STACKSIZE" , .parse: __kmp_stg_parse_stacksize, NULL, NULL, .set: 0, .defined: 0}, |
5591 | #endif |
5592 | |
5593 | #ifdef KMP_DEBUG |
5594 | {.name: "KMP_A_DEBUG" , .parse: __kmp_stg_parse_a_debug, .print: __kmp_stg_print_a_debug, NULL, .set: 0, |
5595 | .defined: 0}, |
5596 | {.name: "KMP_B_DEBUG" , .parse: __kmp_stg_parse_b_debug, .print: __kmp_stg_print_b_debug, NULL, .set: 0, |
5597 | .defined: 0}, |
5598 | {.name: "KMP_C_DEBUG" , .parse: __kmp_stg_parse_c_debug, .print: __kmp_stg_print_c_debug, NULL, .set: 0, |
5599 | .defined: 0}, |
5600 | {.name: "KMP_D_DEBUG" , .parse: __kmp_stg_parse_d_debug, .print: __kmp_stg_print_d_debug, NULL, .set: 0, |
5601 | .defined: 0}, |
5602 | {.name: "KMP_E_DEBUG" , .parse: __kmp_stg_parse_e_debug, .print: __kmp_stg_print_e_debug, NULL, .set: 0, |
5603 | .defined: 0}, |
5604 | {.name: "KMP_F_DEBUG" , .parse: __kmp_stg_parse_f_debug, .print: __kmp_stg_print_f_debug, NULL, .set: 0, |
5605 | .defined: 0}, |
5606 | {.name: "KMP_DEBUG" , .parse: __kmp_stg_parse_debug, NULL, /* no print */ NULL, .set: 0, .defined: 0}, |
5607 | {.name: "KMP_DEBUG_BUF" , .parse: __kmp_stg_parse_debug_buf, .print: __kmp_stg_print_debug_buf, |
5608 | NULL, .set: 0, .defined: 0}, |
5609 | {.name: "KMP_DEBUG_BUF_ATOMIC" , .parse: __kmp_stg_parse_debug_buf_atomic, |
5610 | .print: __kmp_stg_print_debug_buf_atomic, NULL, .set: 0, .defined: 0}, |
5611 | {.name: "KMP_DEBUG_BUF_CHARS" , .parse: __kmp_stg_parse_debug_buf_chars, |
5612 | .print: __kmp_stg_print_debug_buf_chars, NULL, .set: 0, .defined: 0}, |
5613 | {.name: "KMP_DEBUG_BUF_LINES" , .parse: __kmp_stg_parse_debug_buf_lines, |
5614 | .print: __kmp_stg_print_debug_buf_lines, NULL, .set: 0, .defined: 0}, |
5615 | {.name: "KMP_DIAG" , .parse: __kmp_stg_parse_diag, .print: __kmp_stg_print_diag, NULL, .set: 0, .defined: 0}, |
5616 | |
5617 | {.name: "KMP_PAR_RANGE" , .parse: __kmp_stg_parse_par_range_env, |
5618 | .print: __kmp_stg_print_par_range_env, NULL, .set: 0, .defined: 0}, |
5619 | #endif // KMP_DEBUG |
5620 | |
5621 | {.name: "KMP_ALIGN_ALLOC" , .parse: __kmp_stg_parse_align_alloc, |
5622 | .print: __kmp_stg_print_align_alloc, NULL, .set: 0, .defined: 0}, |
5623 | |
5624 | {.name: "KMP_PLAIN_BARRIER" , .parse: __kmp_stg_parse_barrier_branch_bit, |
5625 | .print: __kmp_stg_print_barrier_branch_bit, NULL, .set: 0, .defined: 0}, |
5626 | {.name: "KMP_PLAIN_BARRIER_PATTERN" , .parse: __kmp_stg_parse_barrier_pattern, |
5627 | .print: __kmp_stg_print_barrier_pattern, NULL, .set: 0, .defined: 0}, |
5628 | {.name: "KMP_FORKJOIN_BARRIER" , .parse: __kmp_stg_parse_barrier_branch_bit, |
5629 | .print: __kmp_stg_print_barrier_branch_bit, NULL, .set: 0, .defined: 0}, |
5630 | {.name: "KMP_FORKJOIN_BARRIER_PATTERN" , .parse: __kmp_stg_parse_barrier_pattern, |
5631 | .print: __kmp_stg_print_barrier_pattern, NULL, .set: 0, .defined: 0}, |
5632 | #if KMP_FAST_REDUCTION_BARRIER |
5633 | {.name: "KMP_REDUCTION_BARRIER" , .parse: __kmp_stg_parse_barrier_branch_bit, |
5634 | .print: __kmp_stg_print_barrier_branch_bit, NULL, .set: 0, .defined: 0}, |
5635 | {.name: "KMP_REDUCTION_BARRIER_PATTERN" , .parse: __kmp_stg_parse_barrier_pattern, |
5636 | .print: __kmp_stg_print_barrier_pattern, NULL, .set: 0, .defined: 0}, |
5637 | #endif |
5638 | |
5639 | {.name: "KMP_ABORT_DELAY" , .parse: __kmp_stg_parse_abort_delay, |
5640 | .print: __kmp_stg_print_abort_delay, NULL, .set: 0, .defined: 0}, |
5641 | {.name: "KMP_CPUINFO_FILE" , .parse: __kmp_stg_parse_cpuinfo_file, |
5642 | .print: __kmp_stg_print_cpuinfo_file, NULL, .set: 0, .defined: 0}, |
5643 | {.name: "KMP_FORCE_REDUCTION" , .parse: __kmp_stg_parse_force_reduction, |
5644 | .print: __kmp_stg_print_force_reduction, NULL, .set: 0, .defined: 0}, |
5645 | {.name: "KMP_DETERMINISTIC_REDUCTION" , .parse: __kmp_stg_parse_force_reduction, |
5646 | .print: __kmp_stg_print_force_reduction, NULL, .set: 0, .defined: 0}, |
5647 | {.name: "KMP_STORAGE_MAP" , .parse: __kmp_stg_parse_storage_map, |
5648 | .print: __kmp_stg_print_storage_map, NULL, .set: 0, .defined: 0}, |
5649 | {.name: "KMP_ALL_THREADPRIVATE" , .parse: __kmp_stg_parse_all_threadprivate, |
5650 | .print: __kmp_stg_print_all_threadprivate, NULL, .set: 0, .defined: 0}, |
5651 | {.name: "KMP_FOREIGN_THREADS_THREADPRIVATE" , |
5652 | .parse: __kmp_stg_parse_foreign_threads_threadprivate, |
5653 | .print: __kmp_stg_print_foreign_threads_threadprivate, NULL, .set: 0, .defined: 0}, |
5654 | |
5655 | #if KMP_AFFINITY_SUPPORTED |
5656 | {.name: "KMP_AFFINITY" , .parse: __kmp_stg_parse_affinity, .print: __kmp_stg_print_affinity, NULL, |
5657 | .set: 0, .defined: 0}, |
5658 | {.name: "KMP_HIDDEN_HELPER_AFFINITY" , .parse: __kmp_stg_parse_hh_affinity, |
5659 | .print: __kmp_stg_print_hh_affinity, NULL, .set: 0, .defined: 0}, |
5660 | #ifdef KMP_GOMP_COMPAT |
5661 | {.name: "GOMP_CPU_AFFINITY" , .parse: __kmp_stg_parse_gomp_cpu_affinity, NULL, |
5662 | /* no print */ NULL, .set: 0, .defined: 0}, |
5663 | #endif /* KMP_GOMP_COMPAT */ |
5664 | {.name: "OMP_PROC_BIND" , .parse: __kmp_stg_parse_proc_bind, .print: __kmp_stg_print_proc_bind, |
5665 | NULL, .set: 0, .defined: 0}, |
5666 | {.name: "KMP_TEAMS_PROC_BIND" , .parse: __kmp_stg_parse_teams_proc_bind, |
5667 | .print: __kmp_stg_print_teams_proc_bind, NULL, .set: 0, .defined: 0}, |
5668 | {.name: "OMP_PLACES" , .parse: __kmp_stg_parse_places, .print: __kmp_stg_print_places, NULL, .set: 0, .defined: 0}, |
5669 | {.name: "KMP_TOPOLOGY_METHOD" , .parse: __kmp_stg_parse_topology_method, |
5670 | .print: __kmp_stg_print_topology_method, NULL, .set: 0, .defined: 0}, |
5671 | |
5672 | #else |
5673 | |
5674 | // KMP_AFFINITY is not supported on OS X*, nor is OMP_PLACES. |
5675 | // OMP_PROC_BIND and proc-bind-var are supported, however. |
5676 | {"OMP_PROC_BIND" , __kmp_stg_parse_proc_bind, __kmp_stg_print_proc_bind, |
5677 | NULL, 0, 0}, |
5678 | |
5679 | #endif // KMP_AFFINITY_SUPPORTED |
5680 | {.name: "OMP_DISPLAY_AFFINITY" , .parse: __kmp_stg_parse_display_affinity, |
5681 | .print: __kmp_stg_print_display_affinity, NULL, .set: 0, .defined: 0}, |
5682 | {.name: "OMP_AFFINITY_FORMAT" , .parse: __kmp_stg_parse_affinity_format, |
5683 | .print: __kmp_stg_print_affinity_format, NULL, .set: 0, .defined: 0}, |
5684 | {.name: "KMP_INIT_AT_FORK" , .parse: __kmp_stg_parse_init_at_fork, |
5685 | .print: __kmp_stg_print_init_at_fork, NULL, .set: 0, .defined: 0}, |
5686 | {.name: "KMP_SCHEDULE" , .parse: __kmp_stg_parse_schedule, .print: __kmp_stg_print_schedule, NULL, |
5687 | .set: 0, .defined: 0}, |
5688 | {.name: "OMP_SCHEDULE" , .parse: __kmp_stg_parse_omp_schedule, .print: __kmp_stg_print_omp_schedule, |
5689 | NULL, .set: 0, .defined: 0}, |
5690 | #if KMP_USE_HIER_SCHED |
5691 | {"KMP_DISP_HAND_THREAD" , __kmp_stg_parse_kmp_hand_thread, |
5692 | __kmp_stg_print_kmp_hand_thread, NULL, 0, 0}, |
5693 | #endif |
5694 | {.name: "KMP_FORCE_MONOTONIC_DYNAMIC_SCHEDULE" , |
5695 | .parse: __kmp_stg_parse_kmp_force_monotonic, .print: __kmp_stg_print_kmp_force_monotonic, |
5696 | NULL, .set: 0, .defined: 0}, |
5697 | {.name: "KMP_ATOMIC_MODE" , .parse: __kmp_stg_parse_atomic_mode, |
5698 | .print: __kmp_stg_print_atomic_mode, NULL, .set: 0, .defined: 0}, |
5699 | {.name: "KMP_CONSISTENCY_CHECK" , .parse: __kmp_stg_parse_consistency_check, |
5700 | .print: __kmp_stg_print_consistency_check, NULL, .set: 0, .defined: 0}, |
5701 | |
5702 | #if USE_ITT_BUILD && USE_ITT_NOTIFY |
5703 | {.name: "KMP_ITT_PREPARE_DELAY" , .parse: __kmp_stg_parse_itt_prepare_delay, |
5704 | .print: __kmp_stg_print_itt_prepare_delay, NULL, .set: 0, .defined: 0}, |
5705 | #endif /* USE_ITT_BUILD && USE_ITT_NOTIFY */ |
5706 | {.name: "KMP_MALLOC_POOL_INCR" , .parse: __kmp_stg_parse_malloc_pool_incr, |
5707 | .print: __kmp_stg_print_malloc_pool_incr, NULL, .set: 0, .defined: 0}, |
5708 | {.name: "KMP_GTID_MODE" , .parse: __kmp_stg_parse_gtid_mode, .print: __kmp_stg_print_gtid_mode, |
5709 | NULL, .set: 0, .defined: 0}, |
5710 | {.name: "OMP_DYNAMIC" , .parse: __kmp_stg_parse_omp_dynamic, .print: __kmp_stg_print_omp_dynamic, |
5711 | NULL, .set: 0, .defined: 0}, |
5712 | {.name: "KMP_DYNAMIC_MODE" , .parse: __kmp_stg_parse_kmp_dynamic_mode, |
5713 | .print: __kmp_stg_print_kmp_dynamic_mode, NULL, .set: 0, .defined: 0}, |
5714 | |
5715 | #ifdef USE_LOAD_BALANCE |
5716 | {.name: "KMP_LOAD_BALANCE_INTERVAL" , .parse: __kmp_stg_parse_ld_balance_interval, |
5717 | .print: __kmp_stg_print_ld_balance_interval, NULL, .set: 0, .defined: 0}, |
5718 | #endif |
5719 | |
5720 | {.name: "KMP_NUM_LOCKS_IN_BLOCK" , .parse: __kmp_stg_parse_lock_block, |
5721 | .print: __kmp_stg_print_lock_block, NULL, .set: 0, .defined: 0}, |
5722 | {.name: "KMP_LOCK_KIND" , .parse: __kmp_stg_parse_lock_kind, .print: __kmp_stg_print_lock_kind, |
5723 | NULL, .set: 0, .defined: 0}, |
5724 | {.name: "KMP_SPIN_BACKOFF_PARAMS" , .parse: __kmp_stg_parse_spin_backoff_params, |
5725 | .print: __kmp_stg_print_spin_backoff_params, NULL, .set: 0, .defined: 0}, |
5726 | #if KMP_USE_ADAPTIVE_LOCKS |
5727 | {.name: "KMP_ADAPTIVE_LOCK_PROPS" , .parse: __kmp_stg_parse_adaptive_lock_props, |
5728 | .print: __kmp_stg_print_adaptive_lock_props, NULL, .set: 0, .defined: 0}, |
5729 | #if KMP_DEBUG_ADAPTIVE_LOCKS |
5730 | {"KMP_SPECULATIVE_STATSFILE" , __kmp_stg_parse_speculative_statsfile, |
5731 | __kmp_stg_print_speculative_statsfile, NULL, 0, 0}, |
5732 | #endif |
5733 | #endif // KMP_USE_ADAPTIVE_LOCKS |
5734 | {.name: "KMP_PLACE_THREADS" , .parse: __kmp_stg_parse_hw_subset, .print: __kmp_stg_print_hw_subset, |
5735 | NULL, .set: 0, .defined: 0}, |
5736 | {.name: "KMP_HW_SUBSET" , .parse: __kmp_stg_parse_hw_subset, .print: __kmp_stg_print_hw_subset, |
5737 | NULL, .set: 0, .defined: 0}, |
5738 | #if USE_ITT_BUILD |
5739 | {.name: "KMP_FORKJOIN_FRAMES" , .parse: __kmp_stg_parse_forkjoin_frames, |
5740 | .print: __kmp_stg_print_forkjoin_frames, NULL, .set: 0, .defined: 0}, |
5741 | {.name: "KMP_FORKJOIN_FRAMES_MODE" , .parse: __kmp_stg_parse_forkjoin_frames_mode, |
5742 | .print: __kmp_stg_print_forkjoin_frames_mode, NULL, .set: 0, .defined: 0}, |
5743 | #endif |
5744 | {.name: "KMP_ENABLE_TASK_THROTTLING" , .parse: __kmp_stg_parse_task_throttling, |
5745 | .print: __kmp_stg_print_task_throttling, NULL, .set: 0, .defined: 0}, |
5746 | |
5747 | {.name: "OMP_DISPLAY_ENV" , .parse: __kmp_stg_parse_omp_display_env, |
5748 | .print: __kmp_stg_print_omp_display_env, NULL, .set: 0, .defined: 0}, |
5749 | {.name: "OMP_CANCELLATION" , .parse: __kmp_stg_parse_omp_cancellation, |
5750 | .print: __kmp_stg_print_omp_cancellation, NULL, .set: 0, .defined: 0}, |
5751 | {.name: "OMP_ALLOCATOR" , .parse: __kmp_stg_parse_allocator, .print: __kmp_stg_print_allocator, |
5752 | NULL, .set: 0, .defined: 0}, |
5753 | {.name: "LIBOMP_USE_HIDDEN_HELPER_TASK" , .parse: __kmp_stg_parse_use_hidden_helper, |
5754 | .print: __kmp_stg_print_use_hidden_helper, NULL, .set: 0, .defined: 0}, |
5755 | {.name: "LIBOMP_NUM_HIDDEN_HELPER_THREADS" , |
5756 | .parse: __kmp_stg_parse_num_hidden_helper_threads, |
5757 | .print: __kmp_stg_print_num_hidden_helper_threads, NULL, .set: 0, .defined: 0}, |
5758 | #if OMPX_TASKGRAPH |
5759 | {"KMP_MAX_TDGS" , __kmp_stg_parse_max_tdgs, __kmp_std_print_max_tdgs, NULL, |
5760 | 0, 0}, |
5761 | {"KMP_TDG_DOT" , __kmp_stg_parse_tdg_dot, __kmp_stg_print_tdg_dot, NULL, 0, 0}, |
5762 | #endif |
5763 | |
5764 | #if OMPT_SUPPORT |
5765 | {.name: "OMP_TOOL" , .parse: __kmp_stg_parse_omp_tool, .print: __kmp_stg_print_omp_tool, NULL, .set: 0, |
5766 | .defined: 0}, |
5767 | {.name: "OMP_TOOL_LIBRARIES" , .parse: __kmp_stg_parse_omp_tool_libraries, |
5768 | .print: __kmp_stg_print_omp_tool_libraries, NULL, .set: 0, .defined: 0}, |
5769 | {.name: "OMP_TOOL_VERBOSE_INIT" , .parse: __kmp_stg_parse_omp_tool_verbose_init, |
5770 | .print: __kmp_stg_print_omp_tool_verbose_init, NULL, .set: 0, .defined: 0}, |
5771 | #endif |
5772 | |
5773 | #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT |
5774 | {.name: "KMP_USER_LEVEL_MWAIT" , .parse: __kmp_stg_parse_user_level_mwait, |
5775 | .print: __kmp_stg_print_user_level_mwait, NULL, .set: 0, .defined: 0}, |
5776 | {.name: "KMP_MWAIT_HINTS" , .parse: __kmp_stg_parse_mwait_hints, |
5777 | .print: __kmp_stg_print_mwait_hints, NULL, .set: 0, .defined: 0}, |
5778 | #endif |
5779 | |
5780 | #if KMP_HAVE_UMWAIT |
5781 | {.name: "KMP_TPAUSE" , .parse: __kmp_stg_parse_tpause, .print: __kmp_stg_print_tpause, NULL, .set: 0, .defined: 0}, |
5782 | #endif |
5783 | {.name: "" , NULL, NULL, NULL, .set: 0, .defined: 0}}; // settings |
5784 | |
5785 | static int const __kmp_stg_count = |
5786 | sizeof(__kmp_stg_table) / sizeof(kmp_setting_t); |
5787 | |
5788 | static inline kmp_setting_t *__kmp_stg_find(char const *name) { |
5789 | |
5790 | int i; |
5791 | if (name != NULL) { |
5792 | for (i = 0; i < __kmp_stg_count; ++i) { |
5793 | if (strcmp(s1: __kmp_stg_table[i].name, s2: name) == 0) { |
5794 | return &__kmp_stg_table[i]; |
5795 | } |
5796 | } |
5797 | } |
5798 | return NULL; |
5799 | |
5800 | } // __kmp_stg_find |
5801 | |
5802 | static int __kmp_stg_cmp(void const *_a, void const *_b) { |
5803 | const kmp_setting_t *a = RCAST(const kmp_setting_t *, _a); |
5804 | const kmp_setting_t *b = RCAST(const kmp_setting_t *, _b); |
5805 | |
5806 | // Process KMP_AFFINITY last. |
5807 | // It needs to come after OMP_PLACES and GOMP_CPU_AFFINITY. |
5808 | if (strcmp(s1: a->name, s2: "KMP_AFFINITY" ) == 0) { |
5809 | if (strcmp(s1: b->name, s2: "KMP_AFFINITY" ) == 0) { |
5810 | return 0; |
5811 | } |
5812 | return 1; |
5813 | } else if (strcmp(s1: b->name, s2: "KMP_AFFINITY" ) == 0) { |
5814 | return -1; |
5815 | } |
5816 | return strcmp(s1: a->name, s2: b->name); |
5817 | } // __kmp_stg_cmp |
5818 | |
5819 | static void __kmp_stg_init(void) { |
5820 | |
5821 | static int initialized = 0; |
5822 | |
5823 | if (!initialized) { |
5824 | |
5825 | // Sort table. |
5826 | qsort(base: __kmp_stg_table, nmemb: __kmp_stg_count - 1, size: sizeof(kmp_setting_t), |
5827 | compar: __kmp_stg_cmp); |
5828 | |
5829 | { // Initialize *_STACKSIZE data. |
5830 | kmp_setting_t *kmp_stacksize = |
5831 | __kmp_stg_find(name: "KMP_STACKSIZE" ); // 1st priority. |
5832 | #ifdef KMP_GOMP_COMPAT |
5833 | kmp_setting_t *gomp_stacksize = |
5834 | __kmp_stg_find(name: "GOMP_STACKSIZE" ); // 2nd priority. |
5835 | #endif |
5836 | kmp_setting_t *omp_stacksize = |
5837 | __kmp_stg_find(name: "OMP_STACKSIZE" ); // 3rd priority. |
5838 | |
5839 | // !!! volatile keyword is Intel(R) C Compiler bug CQ49908 workaround. |
5840 | // !!! Compiler does not understand rivals is used and optimizes out |
5841 | // assignments |
5842 | // !!! rivals[ i ++ ] = ...; |
5843 | static kmp_setting_t *volatile rivals[4]; |
5844 | static kmp_stg_ss_data_t kmp_data = {.factor: 1, CCAST(kmp_setting_t **, rivals)}; |
5845 | #ifdef KMP_GOMP_COMPAT |
5846 | static kmp_stg_ss_data_t gomp_data = {.factor: 1024, |
5847 | CCAST(kmp_setting_t **, rivals)}; |
5848 | #endif |
5849 | static kmp_stg_ss_data_t omp_data = {.factor: 1024, |
5850 | CCAST(kmp_setting_t **, rivals)}; |
5851 | int i = 0; |
5852 | |
5853 | rivals[i++] = kmp_stacksize; |
5854 | #ifdef KMP_GOMP_COMPAT |
5855 | if (gomp_stacksize != NULL) { |
5856 | rivals[i++] = gomp_stacksize; |
5857 | } |
5858 | #endif |
5859 | rivals[i++] = omp_stacksize; |
5860 | rivals[i++] = NULL; |
5861 | |
5862 | kmp_stacksize->data = &kmp_data; |
5863 | #ifdef KMP_GOMP_COMPAT |
5864 | if (gomp_stacksize != NULL) { |
5865 | gomp_stacksize->data = &gomp_data; |
5866 | } |
5867 | #endif |
5868 | omp_stacksize->data = &omp_data; |
5869 | } |
5870 | |
5871 | { // Initialize KMP_LIBRARY and OMP_WAIT_POLICY data. |
5872 | kmp_setting_t *kmp_library = |
5873 | __kmp_stg_find(name: "KMP_LIBRARY" ); // 1st priority. |
5874 | kmp_setting_t *omp_wait_policy = |
5875 | __kmp_stg_find(name: "OMP_WAIT_POLICY" ); // 2nd priority. |
5876 | |
5877 | // !!! volatile keyword is Intel(R) C Compiler bug CQ49908 workaround. |
5878 | static kmp_setting_t *volatile rivals[3]; |
5879 | static kmp_stg_wp_data_t kmp_data = {.omp: 0, CCAST(kmp_setting_t **, rivals)}; |
5880 | static kmp_stg_wp_data_t omp_data = {.omp: 1, CCAST(kmp_setting_t **, rivals)}; |
5881 | int i = 0; |
5882 | |
5883 | rivals[i++] = kmp_library; |
5884 | if (omp_wait_policy != NULL) { |
5885 | rivals[i++] = omp_wait_policy; |
5886 | } |
5887 | rivals[i++] = NULL; |
5888 | |
5889 | kmp_library->data = &kmp_data; |
5890 | if (omp_wait_policy != NULL) { |
5891 | omp_wait_policy->data = &omp_data; |
5892 | } |
5893 | } |
5894 | |
5895 | { // Initialize KMP_DEVICE_THREAD_LIMIT and KMP_ALL_THREADS |
5896 | kmp_setting_t *kmp_device_thread_limit = |
5897 | __kmp_stg_find(name: "KMP_DEVICE_THREAD_LIMIT" ); // 1st priority. |
5898 | kmp_setting_t *kmp_all_threads = |
5899 | __kmp_stg_find(name: "KMP_ALL_THREADS" ); // 2nd priority. |
5900 | |
5901 | // !!! volatile keyword is Intel(R) C Compiler bug CQ49908 workaround. |
5902 | static kmp_setting_t *volatile rivals[3]; |
5903 | int i = 0; |
5904 | |
5905 | rivals[i++] = kmp_device_thread_limit; |
5906 | rivals[i++] = kmp_all_threads; |
5907 | rivals[i++] = NULL; |
5908 | |
5909 | kmp_device_thread_limit->data = CCAST(kmp_setting_t **, rivals); |
5910 | kmp_all_threads->data = CCAST(kmp_setting_t **, rivals); |
5911 | } |
5912 | |
5913 | { // Initialize KMP_HW_SUBSET and KMP_PLACE_THREADS |
5914 | // 1st priority |
5915 | kmp_setting_t *kmp_hw_subset = __kmp_stg_find(name: "KMP_HW_SUBSET" ); |
5916 | // 2nd priority |
5917 | kmp_setting_t *kmp_place_threads = __kmp_stg_find(name: "KMP_PLACE_THREADS" ); |
5918 | |
5919 | // !!! volatile keyword is Intel(R) C Compiler bug CQ49908 workaround. |
5920 | static kmp_setting_t *volatile rivals[3]; |
5921 | int i = 0; |
5922 | |
5923 | rivals[i++] = kmp_hw_subset; |
5924 | rivals[i++] = kmp_place_threads; |
5925 | rivals[i++] = NULL; |
5926 | |
5927 | kmp_hw_subset->data = CCAST(kmp_setting_t **, rivals); |
5928 | kmp_place_threads->data = CCAST(kmp_setting_t **, rivals); |
5929 | } |
5930 | |
5931 | #if KMP_AFFINITY_SUPPORTED |
5932 | { // Initialize KMP_AFFINITY, GOMP_CPU_AFFINITY, and OMP_PROC_BIND data. |
5933 | kmp_setting_t *kmp_affinity = |
5934 | __kmp_stg_find(name: "KMP_AFFINITY" ); // 1st priority. |
5935 | KMP_DEBUG_ASSERT(kmp_affinity != NULL); |
5936 | |
5937 | #ifdef KMP_GOMP_COMPAT |
5938 | kmp_setting_t *gomp_cpu_affinity = |
5939 | __kmp_stg_find(name: "GOMP_CPU_AFFINITY" ); // 2nd priority. |
5940 | KMP_DEBUG_ASSERT(gomp_cpu_affinity != NULL); |
5941 | #endif |
5942 | |
5943 | kmp_setting_t *omp_proc_bind = |
5944 | __kmp_stg_find(name: "OMP_PROC_BIND" ); // 3rd priority. |
5945 | KMP_DEBUG_ASSERT(omp_proc_bind != NULL); |
5946 | |
5947 | // !!! volatile keyword is Intel(R) C Compiler bug CQ49908 workaround. |
5948 | static kmp_setting_t *volatile rivals[4]; |
5949 | int i = 0; |
5950 | |
5951 | rivals[i++] = kmp_affinity; |
5952 | |
5953 | #ifdef KMP_GOMP_COMPAT |
5954 | rivals[i++] = gomp_cpu_affinity; |
5955 | gomp_cpu_affinity->data = CCAST(kmp_setting_t **, rivals); |
5956 | #endif |
5957 | |
5958 | rivals[i++] = omp_proc_bind; |
5959 | omp_proc_bind->data = CCAST(kmp_setting_t **, rivals); |
5960 | rivals[i++] = NULL; |
5961 | |
5962 | static kmp_setting_t *volatile places_rivals[4]; |
5963 | i = 0; |
5964 | |
5965 | kmp_setting_t *omp_places = __kmp_stg_find(name: "OMP_PLACES" ); // 3rd priority. |
5966 | KMP_DEBUG_ASSERT(omp_places != NULL); |
5967 | |
5968 | places_rivals[i++] = kmp_affinity; |
5969 | #ifdef KMP_GOMP_COMPAT |
5970 | places_rivals[i++] = gomp_cpu_affinity; |
5971 | #endif |
5972 | places_rivals[i++] = omp_places; |
5973 | omp_places->data = CCAST(kmp_setting_t **, places_rivals); |
5974 | places_rivals[i++] = NULL; |
5975 | } |
5976 | #else |
5977 | // KMP_AFFINITY not supported, so OMP_PROC_BIND has no rivals. |
5978 | // OMP_PLACES not supported yet. |
5979 | #endif // KMP_AFFINITY_SUPPORTED |
5980 | |
5981 | { // Initialize KMP_DETERMINISTIC_REDUCTION and KMP_FORCE_REDUCTION data. |
5982 | kmp_setting_t *kmp_force_red = |
5983 | __kmp_stg_find(name: "KMP_FORCE_REDUCTION" ); // 1st priority. |
5984 | kmp_setting_t *kmp_determ_red = |
5985 | __kmp_stg_find(name: "KMP_DETERMINISTIC_REDUCTION" ); // 2nd priority. |
5986 | |
5987 | // !!! volatile keyword is Intel(R) C Compiler bug CQ49908 workaround. |
5988 | static kmp_setting_t *volatile rivals[3]; |
5989 | static kmp_stg_fr_data_t force_data = {.force: 1, |
5990 | CCAST(kmp_setting_t **, rivals)}; |
5991 | static kmp_stg_fr_data_t determ_data = {.force: 0, |
5992 | CCAST(kmp_setting_t **, rivals)}; |
5993 | int i = 0; |
5994 | |
5995 | rivals[i++] = kmp_force_red; |
5996 | if (kmp_determ_red != NULL) { |
5997 | rivals[i++] = kmp_determ_red; |
5998 | } |
5999 | rivals[i++] = NULL; |
6000 | |
6001 | kmp_force_red->data = &force_data; |
6002 | if (kmp_determ_red != NULL) { |
6003 | kmp_determ_red->data = &determ_data; |
6004 | } |
6005 | } |
6006 | |
6007 | initialized = 1; |
6008 | } |
6009 | |
6010 | // Reset flags. |
6011 | int i; |
6012 | for (i = 0; i < __kmp_stg_count; ++i) { |
6013 | __kmp_stg_table[i].set = 0; |
6014 | } |
6015 | |
6016 | } // __kmp_stg_init |
6017 | |
6018 | static void __kmp_stg_parse(char const *name, char const *value) { |
6019 | // On Windows* OS there are some nameless variables like "C:=C:\" (yeah, |
6020 | // really nameless, they are presented in environment block as |
6021 | // "=C:=C\\\x00=D:=D:\\\x00...", so let us skip them. |
6022 | if (name[0] == 0) { |
6023 | return; |
6024 | } |
6025 | |
6026 | if (value != NULL) { |
6027 | kmp_setting_t *setting = __kmp_stg_find(name); |
6028 | if (setting != NULL) { |
6029 | setting->parse(name, value, setting->data); |
6030 | setting->defined = 1; |
6031 | } |
6032 | } |
6033 | |
6034 | } // __kmp_stg_parse |
6035 | |
6036 | static int __kmp_stg_check_rivals( // 0 -- Ok, 1 -- errors found. |
6037 | char const *name, // Name of variable. |
6038 | char const *value, // Value of the variable. |
6039 | kmp_setting_t **rivals // List of rival settings (must include current one). |
6040 | ) { |
6041 | |
6042 | if (rivals == NULL) { |
6043 | return 0; |
6044 | } |
6045 | |
6046 | // Loop thru higher priority settings (listed before current). |
6047 | int i = 0; |
6048 | for (; strcmp(s1: rivals[i]->name, s2: name) != 0; i++) { |
6049 | KMP_DEBUG_ASSERT(rivals[i] != NULL); |
6050 | |
6051 | #if KMP_AFFINITY_SUPPORTED |
6052 | if (rivals[i] == __kmp_affinity_notype) { |
6053 | // If KMP_AFFINITY is specified without a type name, |
6054 | // it does not rival OMP_PROC_BIND or GOMP_CPU_AFFINITY. |
6055 | continue; |
6056 | } |
6057 | #endif |
6058 | |
6059 | if (rivals[i]->set) { |
6060 | KMP_WARNING(StgIgnored, name, rivals[i]->name); |
6061 | return 1; |
6062 | } |
6063 | } |
6064 | |
6065 | ++i; // Skip current setting. |
6066 | return 0; |
6067 | |
6068 | } // __kmp_stg_check_rivals |
6069 | |
6070 | static int __kmp_env_toPrint(char const *name, int flag) { |
6071 | int rc = 0; |
6072 | kmp_setting_t *setting = __kmp_stg_find(name); |
6073 | if (setting != NULL) { |
6074 | rc = setting->defined; |
6075 | if (flag >= 0) { |
6076 | setting->defined = flag; |
6077 | } |
6078 | } |
6079 | return rc; |
6080 | } |
6081 | |
6082 | #if defined(KMP_DEBUG) && KMP_AFFINITY_SUPPORTED |
6083 | static void __kmp_print_affinity_settings(const kmp_affinity_t *affinity) { |
6084 | K_DIAG(1, ("%s:\n" , affinity->env_var)); |
6085 | K_DIAG(1, (" type : %d\n" , affinity->type)); |
6086 | K_DIAG(1, (" compact : %d\n" , affinity->compact)); |
6087 | K_DIAG(1, (" offset : %d\n" , affinity->offset)); |
6088 | K_DIAG(1, (" verbose : %u\n" , affinity->flags.verbose)); |
6089 | K_DIAG(1, (" warnings : %u\n" , affinity->flags.warnings)); |
6090 | K_DIAG(1, (" respect : %u\n" , affinity->flags.respect)); |
6091 | K_DIAG(1, (" reset : %u\n" , affinity->flags.reset)); |
6092 | K_DIAG(1, (" dups : %u\n" , affinity->flags.dups)); |
6093 | K_DIAG(1, (" gran : %d\n" , (int)affinity->gran)); |
6094 | KMP_DEBUG_ASSERT(affinity->type != affinity_default); |
6095 | } |
6096 | #endif |
6097 | |
6098 | static void __kmp_aux_env_initialize(kmp_env_blk_t *block) { |
6099 | |
6100 | char const *value; |
6101 | |
6102 | /* OMP_NUM_THREADS */ |
6103 | value = __kmp_env_blk_var(block, name: "OMP_NUM_THREADS" ); |
6104 | if (value) { |
6105 | ompc_set_num_threads(arg: __kmp_dflt_team_nth); |
6106 | } |
6107 | |
6108 | /* KMP_BLOCKTIME */ |
6109 | value = __kmp_env_blk_var(block, name: "KMP_BLOCKTIME" ); |
6110 | if (value) { |
6111 | int gtid, tid; |
6112 | kmp_info_t *thread; |
6113 | |
6114 | gtid = __kmp_entry_gtid(); |
6115 | tid = __kmp_tid_from_gtid(gtid); |
6116 | thread = __kmp_thread_from_gtid(gtid); |
6117 | __kmp_aux_set_blocktime(arg: __kmp_dflt_blocktime, thread, tid); |
6118 | } |
6119 | |
6120 | /* OMP_NESTED */ |
6121 | value = __kmp_env_blk_var(block, name: "OMP_NESTED" ); |
6122 | if (value) { |
6123 | ompc_set_nested(flag: __kmp_dflt_max_active_levels > 1); |
6124 | } |
6125 | |
6126 | /* OMP_DYNAMIC */ |
6127 | value = __kmp_env_blk_var(block, name: "OMP_DYNAMIC" ); |
6128 | if (value) { |
6129 | ompc_set_dynamic(flag: __kmp_global.g.g_dynamic); |
6130 | } |
6131 | } |
6132 | |
6133 | void __kmp_env_initialize(char const *string) { |
6134 | |
6135 | kmp_env_blk_t block; |
6136 | int i; |
6137 | |
6138 | __kmp_stg_init(); |
6139 | |
6140 | // Hack!!! |
6141 | if (string == NULL) { |
6142 | // __kmp_max_nth = __kmp_sys_max_nth; |
6143 | __kmp_threads_capacity = |
6144 | __kmp_initial_threads_capacity(req_nproc: __kmp_dflt_team_nth_ub); |
6145 | } |
6146 | __kmp_env_blk_init(block: &block, bulk: string); |
6147 | |
6148 | // update the set flag on all entries that have an env var |
6149 | for (i = 0; i < block.count; ++i) { |
6150 | if ((block.vars[i].name == NULL) || (*block.vars[i].name == '\0')) { |
6151 | continue; |
6152 | } |
6153 | if (block.vars[i].value == NULL) { |
6154 | continue; |
6155 | } |
6156 | kmp_setting_t *setting = __kmp_stg_find(name: block.vars[i].name); |
6157 | if (setting != NULL) { |
6158 | setting->set = 1; |
6159 | } |
6160 | } |
6161 | |
6162 | // We need to know if blocktime was set when processing OMP_WAIT_POLICY |
6163 | blocktime_str = __kmp_env_blk_var(block: &block, name: "KMP_BLOCKTIME" ); |
6164 | |
6165 | // Special case. If we parse environment, not a string, process KMP_WARNINGS |
6166 | // first. |
6167 | if (string == NULL) { |
6168 | char const *name = "KMP_WARNINGS" ; |
6169 | char const *value = __kmp_env_blk_var(block: &block, name); |
6170 | __kmp_stg_parse(name, value); |
6171 | } |
6172 | |
6173 | #if KMP_AFFINITY_SUPPORTED |
6174 | // Special case. KMP_AFFINITY is not a rival to other affinity env vars |
6175 | // if no affinity type is specified. We want to allow |
6176 | // KMP_AFFINITY=[no],verbose/[no]warnings/etc. to be enabled when |
6177 | // specifying the affinity type via GOMP_CPU_AFFINITY or the OMP 4.0 |
6178 | // affinity mechanism. |
6179 | __kmp_affinity_notype = NULL; |
6180 | char const *aff_str = __kmp_env_blk_var(block: &block, name: "KMP_AFFINITY" ); |
6181 | if (aff_str != NULL) { |
6182 | // Check if the KMP_AFFINITY type is specified in the string. |
6183 | // We just search the string for "compact", "scatter", etc. |
6184 | // without really parsing the string. The syntax of the |
6185 | // KMP_AFFINITY env var is such that none of the affinity |
6186 | // type names can appear anywhere other that the type |
6187 | // specifier, even as substrings. |
6188 | // |
6189 | // I can't find a case-insensitive version of strstr on Windows* OS. |
6190 | // Use the case-sensitive version for now. AIX does the same. |
6191 | |
6192 | #if KMP_OS_WINDOWS || KMP_OS_AIX |
6193 | #define FIND strstr |
6194 | #else |
6195 | #define FIND strcasestr |
6196 | #endif |
6197 | |
6198 | if ((FIND(haystack: aff_str, needle: "none" ) == NULL) && |
6199 | (FIND(haystack: aff_str, needle: "physical" ) == NULL) && |
6200 | (FIND(haystack: aff_str, needle: "logical" ) == NULL) && |
6201 | (FIND(haystack: aff_str, needle: "compact" ) == NULL) && |
6202 | (FIND(haystack: aff_str, needle: "scatter" ) == NULL) && |
6203 | (FIND(haystack: aff_str, needle: "explicit" ) == NULL) && |
6204 | (FIND(haystack: aff_str, needle: "balanced" ) == NULL) && |
6205 | (FIND(haystack: aff_str, needle: "disabled" ) == NULL)) { |
6206 | __kmp_affinity_notype = __kmp_stg_find(name: "KMP_AFFINITY" ); |
6207 | } else { |
6208 | // A new affinity type is specified. |
6209 | // Reset the affinity flags to their default values, |
6210 | // in case this is called from kmp_set_defaults(). |
6211 | __kmp_affinity.type = affinity_default; |
6212 | __kmp_affinity.gran = KMP_HW_UNKNOWN; |
6213 | __kmp_affinity_top_method = affinity_top_method_default; |
6214 | __kmp_affinity.flags.respect = affinity_respect_mask_default; |
6215 | } |
6216 | #undef FIND |
6217 | |
6218 | // Also reset the affinity flags if OMP_PROC_BIND is specified. |
6219 | aff_str = __kmp_env_blk_var(block: &block, name: "OMP_PROC_BIND" ); |
6220 | if (aff_str != NULL) { |
6221 | __kmp_affinity.type = affinity_default; |
6222 | __kmp_affinity.gran = KMP_HW_UNKNOWN; |
6223 | __kmp_affinity_top_method = affinity_top_method_default; |
6224 | __kmp_affinity.flags.respect = affinity_respect_mask_default; |
6225 | } |
6226 | } |
6227 | |
6228 | #endif /* KMP_AFFINITY_SUPPORTED */ |
6229 | |
6230 | // Set up the nested proc bind type vector. |
6231 | if (__kmp_nested_proc_bind.bind_types == NULL) { |
6232 | __kmp_nested_proc_bind.bind_types = |
6233 | (kmp_proc_bind_t *)KMP_INTERNAL_MALLOC(sizeof(kmp_proc_bind_t)); |
6234 | if (__kmp_nested_proc_bind.bind_types == NULL) { |
6235 | KMP_FATAL(MemoryAllocFailed); |
6236 | } |
6237 | __kmp_nested_proc_bind.size = 1; |
6238 | __kmp_nested_proc_bind.used = 1; |
6239 | #if KMP_AFFINITY_SUPPORTED |
6240 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_default; |
6241 | #else |
6242 | // default proc bind is false if affinity not supported |
6243 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
6244 | #endif |
6245 | } |
6246 | |
6247 | // Set up the affinity format ICV |
6248 | // Grab the default affinity format string from the message catalog |
6249 | kmp_msg_t m = |
6250 | __kmp_msg_format(kmp_i18n_msg_AffFormatDefault, "%P" , "%i" , "%n" , "%A" ); |
6251 | KMP_DEBUG_ASSERT(KMP_STRLEN(m.str) < KMP_AFFINITY_FORMAT_SIZE); |
6252 | |
6253 | if (__kmp_affinity_format == NULL) { |
6254 | __kmp_affinity_format = |
6255 | (char *)KMP_INTERNAL_MALLOC(sizeof(char) * KMP_AFFINITY_FORMAT_SIZE); |
6256 | } |
6257 | KMP_STRCPY_S(__kmp_affinity_format, KMP_AFFINITY_FORMAT_SIZE, m.str); |
6258 | __kmp_str_free(str: &m.str); |
6259 | |
6260 | // Now process all of the settings. |
6261 | for (i = 0; i < block.count; ++i) { |
6262 | __kmp_stg_parse(name: block.vars[i].name, value: block.vars[i].value); |
6263 | } |
6264 | |
6265 | // If user locks have been allocated yet, don't reset the lock vptr table. |
6266 | if (!__kmp_init_user_locks) { |
6267 | if (__kmp_user_lock_kind == lk_default) { |
6268 | __kmp_user_lock_kind = lk_queuing; |
6269 | } |
6270 | #if KMP_USE_DYNAMIC_LOCK |
6271 | __kmp_init_dynamic_user_locks(); |
6272 | #else |
6273 | __kmp_set_user_lock_vptrs(__kmp_user_lock_kind); |
6274 | #endif |
6275 | } else { |
6276 | KMP_DEBUG_ASSERT(string != NULL); // kmp_set_defaults() was called |
6277 | KMP_DEBUG_ASSERT(__kmp_user_lock_kind != lk_default); |
6278 | // Binds lock functions again to follow the transition between different |
6279 | // KMP_CONSISTENCY_CHECK values. Calling this again is harmless as long |
6280 | // as we do not allow lock kind changes after making a call to any |
6281 | // user lock functions (true). |
6282 | #if KMP_USE_DYNAMIC_LOCK |
6283 | __kmp_init_dynamic_user_locks(); |
6284 | #else |
6285 | __kmp_set_user_lock_vptrs(__kmp_user_lock_kind); |
6286 | #endif |
6287 | } |
6288 | |
6289 | #if KMP_AFFINITY_SUPPORTED |
6290 | |
6291 | if (!TCR_4(__kmp_init_middle)) { |
6292 | #if KMP_USE_HWLOC |
6293 | // Force using hwloc when either tiles or numa nodes requested within |
6294 | // KMP_HW_SUBSET or granularity setting and no other topology method |
6295 | // is requested |
6296 | if (__kmp_hw_subset && |
6297 | __kmp_affinity_top_method == affinity_top_method_default) |
6298 | if (__kmp_hw_subset->specified(KMP_HW_NUMA) || |
6299 | __kmp_hw_subset->specified(KMP_HW_TILE) || |
6300 | __kmp_affinity.gran == KMP_HW_TILE || |
6301 | __kmp_affinity.gran == KMP_HW_NUMA) |
6302 | __kmp_affinity_top_method = affinity_top_method_hwloc; |
6303 | // Force using hwloc when tiles or numa nodes requested for OMP_PLACES |
6304 | if (__kmp_affinity.gran == KMP_HW_NUMA || |
6305 | __kmp_affinity.gran == KMP_HW_TILE) |
6306 | __kmp_affinity_top_method = affinity_top_method_hwloc; |
6307 | #endif |
6308 | // Determine if the machine/OS is actually capable of supporting |
6309 | // affinity. |
6310 | const char *var = "KMP_AFFINITY" ; |
6311 | KMPAffinity::pick_api(); |
6312 | #if KMP_USE_HWLOC |
6313 | // If Hwloc topology discovery was requested but affinity was also disabled, |
6314 | // then tell user that Hwloc request is being ignored and use default |
6315 | // topology discovery method. |
6316 | if (__kmp_affinity_top_method == affinity_top_method_hwloc && |
6317 | __kmp_affinity_dispatch->get_api_type() != KMPAffinity::HWLOC) { |
6318 | KMP_WARNING(AffIgnoringHwloc, var); |
6319 | __kmp_affinity_top_method = affinity_top_method_all; |
6320 | } |
6321 | #endif |
6322 | if (__kmp_affinity.type == affinity_disabled) { |
6323 | KMP_AFFINITY_DISABLE(); |
6324 | } else if (!KMP_AFFINITY_CAPABLE()) { |
6325 | __kmp_affinity_dispatch->determine_capable(env_var: var); |
6326 | if (!KMP_AFFINITY_CAPABLE()) { |
6327 | if (__kmp_affinity.flags.verbose || |
6328 | (__kmp_affinity.flags.warnings && |
6329 | (__kmp_affinity.type != affinity_default) && |
6330 | (__kmp_affinity.type != affinity_none) && |
6331 | (__kmp_affinity.type != affinity_disabled))) { |
6332 | KMP_WARNING(AffNotSupported, var); |
6333 | } |
6334 | __kmp_affinity.type = affinity_disabled; |
6335 | __kmp_affinity.flags.respect = FALSE; |
6336 | __kmp_affinity.gran = KMP_HW_THREAD; |
6337 | } |
6338 | } |
6339 | |
6340 | if (__kmp_affinity.type == affinity_disabled) { |
6341 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
6342 | } else if (__kmp_nested_proc_bind.bind_types[0] == proc_bind_true) { |
6343 | // OMP_PROC_BIND=true maps to OMP_PROC_BIND=spread. |
6344 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_spread; |
6345 | } |
6346 | |
6347 | if (KMP_AFFINITY_CAPABLE()) { |
6348 | |
6349 | #if KMP_GROUP_AFFINITY |
6350 | // This checks to see if the initial affinity mask is equal |
6351 | // to a single windows processor group. If it is, then we do |
6352 | // not respect the initial affinity mask and instead, use the |
6353 | // entire machine. |
6354 | bool exactly_one_group = false; |
6355 | if (__kmp_num_proc_groups > 1) { |
6356 | int group; |
6357 | bool within_one_group; |
6358 | // Get the initial affinity mask and determine if it is |
6359 | // contained within a single group. |
6360 | kmp_affin_mask_t *init_mask; |
6361 | KMP_CPU_ALLOC(init_mask); |
6362 | __kmp_get_system_affinity(init_mask, TRUE); |
6363 | group = __kmp_get_proc_group(init_mask); |
6364 | within_one_group = (group >= 0); |
6365 | // If the initial affinity is within a single group, |
6366 | // then determine if it is equal to that single group. |
6367 | if (within_one_group) { |
6368 | DWORD num_bits_in_group = __kmp_GetActiveProcessorCount(group); |
6369 | DWORD num_bits_in_mask = 0; |
6370 | for (int bit = init_mask->begin(); bit != init_mask->end(); |
6371 | bit = init_mask->next(bit)) |
6372 | num_bits_in_mask++; |
6373 | exactly_one_group = (num_bits_in_group == num_bits_in_mask); |
6374 | } |
6375 | KMP_CPU_FREE(init_mask); |
6376 | } |
6377 | |
6378 | // Handle the Win 64 group affinity stuff if there are multiple |
6379 | // processor groups, or if the user requested it, and OMP 4.0 |
6380 | // affinity is not in effect. |
6381 | if (__kmp_num_proc_groups > 1 && |
6382 | __kmp_affinity.type == affinity_default && |
6383 | __kmp_nested_proc_bind.bind_types[0] == proc_bind_default) { |
6384 | // Do not respect the initial processor affinity mask if it is assigned |
6385 | // exactly one Windows Processor Group since this is interpreted as the |
6386 | // default OS assignment. Not respecting the mask allows the runtime to |
6387 | // use all the logical processors in all groups. |
6388 | if (__kmp_affinity.flags.respect == affinity_respect_mask_default && |
6389 | exactly_one_group) { |
6390 | __kmp_affinity.flags.respect = FALSE; |
6391 | } |
6392 | // Use compact affinity with anticipation of pinning to at least the |
6393 | // group granularity since threads can only be bound to one group. |
6394 | if (__kmp_affinity.type == affinity_default) { |
6395 | __kmp_affinity.type = affinity_compact; |
6396 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
6397 | } |
6398 | if (__kmp_hh_affinity.type == affinity_default) |
6399 | __kmp_hh_affinity.type = affinity_compact; |
6400 | if (__kmp_affinity_top_method == affinity_top_method_default) |
6401 | __kmp_affinity_top_method = affinity_top_method_all; |
6402 | if (__kmp_affinity.gran == KMP_HW_UNKNOWN) |
6403 | __kmp_affinity.gran = KMP_HW_PROC_GROUP; |
6404 | if (__kmp_hh_affinity.gran == KMP_HW_UNKNOWN) |
6405 | __kmp_hh_affinity.gran = KMP_HW_PROC_GROUP; |
6406 | } else |
6407 | |
6408 | #endif /* KMP_GROUP_AFFINITY */ |
6409 | |
6410 | { |
6411 | if (__kmp_affinity.flags.respect == affinity_respect_mask_default) { |
6412 | #if KMP_GROUP_AFFINITY |
6413 | if (__kmp_num_proc_groups > 1 && exactly_one_group) { |
6414 | __kmp_affinity.flags.respect = FALSE; |
6415 | } else |
6416 | #endif /* KMP_GROUP_AFFINITY */ |
6417 | { |
6418 | __kmp_affinity.flags.respect = TRUE; |
6419 | } |
6420 | } |
6421 | if ((__kmp_nested_proc_bind.bind_types[0] != proc_bind_intel) && |
6422 | (__kmp_nested_proc_bind.bind_types[0] != proc_bind_default)) { |
6423 | if (__kmp_affinity.type == affinity_default) { |
6424 | __kmp_affinity.type = affinity_compact; |
6425 | __kmp_affinity.flags.dups = FALSE; |
6426 | } |
6427 | } else if (__kmp_affinity.type == affinity_default) { |
6428 | #if KMP_MIC_SUPPORTED |
6429 | if (__kmp_mic_type != non_mic) { |
6430 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel; |
6431 | } else |
6432 | #endif |
6433 | { |
6434 | __kmp_nested_proc_bind.bind_types[0] = proc_bind_false; |
6435 | } |
6436 | #if KMP_MIC_SUPPORTED |
6437 | if (__kmp_mic_type != non_mic) { |
6438 | __kmp_affinity.type = affinity_scatter; |
6439 | } else |
6440 | #endif |
6441 | { |
6442 | __kmp_affinity.type = affinity_none; |
6443 | } |
6444 | } |
6445 | if (__kmp_hh_affinity.type == affinity_default) |
6446 | __kmp_hh_affinity.type = affinity_none; |
6447 | if ((__kmp_affinity.gran == KMP_HW_UNKNOWN) && |
6448 | (__kmp_affinity.gran_levels < 0)) { |
6449 | #if KMP_MIC_SUPPORTED |
6450 | if (__kmp_mic_type != non_mic) { |
6451 | __kmp_affinity.gran = KMP_HW_THREAD; |
6452 | } else |
6453 | #endif |
6454 | { |
6455 | __kmp_affinity.gran = KMP_HW_CORE; |
6456 | } |
6457 | } |
6458 | if ((__kmp_hh_affinity.gran == KMP_HW_UNKNOWN) && |
6459 | (__kmp_hh_affinity.gran_levels < 0)) { |
6460 | #if KMP_MIC_SUPPORTED |
6461 | if (__kmp_mic_type != non_mic) { |
6462 | __kmp_hh_affinity.gran = KMP_HW_THREAD; |
6463 | } else |
6464 | #endif |
6465 | { |
6466 | __kmp_hh_affinity.gran = KMP_HW_CORE; |
6467 | } |
6468 | } |
6469 | if (__kmp_affinity_top_method == affinity_top_method_default) { |
6470 | __kmp_affinity_top_method = affinity_top_method_all; |
6471 | } |
6472 | } |
6473 | } else { |
6474 | // If affinity is disabled, then still need to assign topology method |
6475 | // to attempt machine detection and affinity types |
6476 | if (__kmp_affinity_top_method == affinity_top_method_default) |
6477 | __kmp_affinity_top_method = affinity_top_method_all; |
6478 | if (__kmp_affinity.type == affinity_default) |
6479 | __kmp_affinity.type = affinity_disabled; |
6480 | if (__kmp_hh_affinity.type == affinity_default) |
6481 | __kmp_hh_affinity.type = affinity_disabled; |
6482 | } |
6483 | |
6484 | #ifdef KMP_DEBUG |
6485 | for (const kmp_affinity_t *affinity : __kmp_affinities) |
6486 | __kmp_print_affinity_settings(affinity); |
6487 | KMP_DEBUG_ASSERT(__kmp_nested_proc_bind.bind_types[0] != proc_bind_default); |
6488 | K_DIAG(1, ("__kmp_nested_proc_bind.bind_types[0] == %d\n" , |
6489 | __kmp_nested_proc_bind.bind_types[0])); |
6490 | #endif |
6491 | } |
6492 | |
6493 | #endif /* KMP_AFFINITY_SUPPORTED */ |
6494 | |
6495 | // Post-initialization step: some env. vars need their value's further |
6496 | // processing |
6497 | if (string != NULL) { // kmp_set_defaults() was called |
6498 | __kmp_aux_env_initialize(block: &block); |
6499 | } |
6500 | |
6501 | __kmp_env_blk_free(block: &block); |
6502 | |
6503 | KMP_MB(); |
6504 | |
6505 | } // __kmp_env_initialize |
6506 | |
6507 | void __kmp_env_print() { |
6508 | |
6509 | kmp_env_blk_t block; |
6510 | int i; |
6511 | kmp_str_buf_t buffer; |
6512 | |
6513 | __kmp_stg_init(); |
6514 | __kmp_str_buf_init(&buffer); |
6515 | |
6516 | __kmp_env_blk_init(block: &block, NULL); |
6517 | __kmp_env_blk_sort(block: &block); |
6518 | |
6519 | // Print real environment values. |
6520 | __kmp_str_buf_print(&buffer, "\n%s\n\n" , KMP_I18N_STR(UserSettings)); |
6521 | for (i = 0; i < block.count; ++i) { |
6522 | char const *name = block.vars[i].name; |
6523 | char const *value = block.vars[i].value; |
6524 | if ((KMP_STRLEN(s: name) > 4 && strncmp(s1: name, s2: "KMP_" , n: 4) == 0) || |
6525 | strncmp(s1: name, s2: "OMP_" , n: 4) == 0 |
6526 | #ifdef KMP_GOMP_COMPAT |
6527 | || strncmp(s1: name, s2: "GOMP_" , n: 5) == 0 |
6528 | #endif // KMP_GOMP_COMPAT |
6529 | ) { |
6530 | __kmp_str_buf_print(buffer: &buffer, format: " %s=%s\n" , name, value); |
6531 | } |
6532 | } |
6533 | __kmp_str_buf_print(buffer: &buffer, format: "\n" ); |
6534 | |
6535 | // Print internal (effective) settings. |
6536 | __kmp_str_buf_print(&buffer, "%s\n\n" , KMP_I18N_STR(EffectiveSettings)); |
6537 | for (int i = 0; i < __kmp_stg_count; ++i) { |
6538 | if (__kmp_stg_table[i].print != NULL) { |
6539 | __kmp_stg_table[i].print(&buffer, __kmp_stg_table[i].name, |
6540 | __kmp_stg_table[i].data); |
6541 | } |
6542 | } |
6543 | |
6544 | __kmp_printf(format: "%s" , buffer.str); |
6545 | |
6546 | __kmp_env_blk_free(block: &block); |
6547 | __kmp_str_buf_free(buffer: &buffer); |
6548 | |
6549 | __kmp_printf(format: "\n" ); |
6550 | |
6551 | } // __kmp_env_print |
6552 | |
6553 | void __kmp_env_print_2() { |
6554 | __kmp_display_env_impl(display_env: __kmp_display_env, display_env_verbose: __kmp_display_env_verbose); |
6555 | } // __kmp_env_print_2 |
6556 | |
6557 | void __kmp_display_env_impl(int display_env, int display_env_verbose) { |
6558 | kmp_env_blk_t block; |
6559 | kmp_str_buf_t buffer; |
6560 | |
6561 | __kmp_env_format = 1; |
6562 | |
6563 | __kmp_stg_init(); |
6564 | __kmp_str_buf_init(&buffer); |
6565 | |
6566 | __kmp_env_blk_init(block: &block, NULL); |
6567 | __kmp_env_blk_sort(block: &block); |
6568 | |
6569 | __kmp_str_buf_print(&buffer, "\n%s\n" , KMP_I18N_STR(DisplayEnvBegin)); |
6570 | __kmp_str_buf_print(buffer: &buffer, format: " _OPENMP='%d'\n" , __kmp_openmp_version); |
6571 | |
6572 | for (int i = 0; i < __kmp_stg_count; ++i) { |
6573 | if (__kmp_stg_table[i].print != NULL && |
6574 | ((display_env && strncmp(s1: __kmp_stg_table[i].name, s2: "OMP_" , n: 4) == 0) || |
6575 | display_env_verbose)) { |
6576 | __kmp_stg_table[i].print(&buffer, __kmp_stg_table[i].name, |
6577 | __kmp_stg_table[i].data); |
6578 | } |
6579 | } |
6580 | |
6581 | __kmp_str_buf_print(&buffer, "%s\n" , KMP_I18N_STR(DisplayEnvEnd)); |
6582 | __kmp_str_buf_print(buffer: &buffer, format: "\n" ); |
6583 | |
6584 | __kmp_printf(format: "%s" , buffer.str); |
6585 | |
6586 | __kmp_env_blk_free(block: &block); |
6587 | __kmp_str_buf_free(buffer: &buffer); |
6588 | |
6589 | __kmp_printf(format: "\n" ); |
6590 | } |
6591 | |
6592 | #if OMPD_SUPPORT |
6593 | // Dump environment variables for OMPD |
6594 | void __kmp_env_dump() { |
6595 | |
6596 | kmp_env_blk_t block; |
6597 | kmp_str_buf_t buffer, env, notdefined; |
6598 | |
6599 | __kmp_stg_init(); |
6600 | __kmp_str_buf_init(&buffer); |
6601 | __kmp_str_buf_init(&env); |
6602 | __kmp_str_buf_init(¬defined); |
6603 | |
6604 | __kmp_env_blk_init(block: &block, NULL); |
6605 | __kmp_env_blk_sort(block: &block); |
6606 | |
6607 | __kmp_str_buf_print(¬defined, ": %s" , KMP_I18N_STR(NotDefined)); |
6608 | |
6609 | for (int i = 0; i < __kmp_stg_count; ++i) { |
6610 | if (__kmp_stg_table[i].print == NULL) |
6611 | continue; |
6612 | __kmp_str_buf_clear(buffer: &env); |
6613 | __kmp_stg_table[i].print(&env, __kmp_stg_table[i].name, |
6614 | __kmp_stg_table[i].data); |
6615 | if (env.used < 4) // valid definition must have indents (3) and a new line |
6616 | continue; |
6617 | if (strstr(haystack: env.str, needle: notdefined.str)) |
6618 | // normalize the string |
6619 | __kmp_str_buf_print(buffer: &buffer, format: "%s=undefined\n" , __kmp_stg_table[i].name); |
6620 | else |
6621 | __kmp_str_buf_cat(buffer: &buffer, str: env.str + 3, len: env.used - 3); |
6622 | } |
6623 | |
6624 | ompd_env_block = (char *)__kmp_allocate(buffer.used + 1); |
6625 | KMP_MEMCPY(dest: ompd_env_block, src: buffer.str, n: buffer.used + 1); |
6626 | ompd_env_block_size = (ompd_size_t)KMP_STRLEN(s: ompd_env_block); |
6627 | |
6628 | __kmp_env_blk_free(block: &block); |
6629 | __kmp_str_buf_free(buffer: &buffer); |
6630 | __kmp_str_buf_free(buffer: &env); |
6631 | __kmp_str_buf_free(buffer: ¬defined); |
6632 | } |
6633 | #endif // OMPD_SUPPORT |
6634 | |
6635 | // end of file |
6636 | |