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
__kmp_convert_to_double(char const * s)40 static double __kmp_convert_to_double(char const *s) {
41 double result;
42
43 if (KMP_SSCANF(s, "%lf", &result) < 1) {
44 result = 0.0;
45 }
46
47 return result;
48 }
49 #endif
50
51 #ifdef KMP_DEBUG
__kmp_readstr_with_sentinel(char * dest,char const * src,size_t len,char sentinel)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
__kmp_match_with_sentinel(char const * a,char const * b,size_t len,char sentinel)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
__kmp_match_str(char const * token,char const * buf,const char ** end)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
__kmp_round4k(size_t size)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
__kmp_strcasecmp_with_sentinel(char const * a,char const * b,char sentinel)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;
kmp_trimmed_str_tkmp_trimmed_str_t224 kmp_trimmed_str_t(const char *str) {
225 __kmp_str_buf_init(&buf);
226 size_t len = KMP_STRLEN(str);
227 if (len == 0)
228 return;
229 const char *begin = str;
230 const char *end = str + KMP_STRLEN(str) - 1;
231 SKIP_WS(begin);
232 while (begin < end && *end == ' ')
233 end--;
234 __kmp_str_buf_cat(&buf, begin, end - begin + 1);
235 }
~kmp_trimmed_str_tkmp_trimmed_str_t236 ~kmp_trimmed_str_t() { __kmp_str_buf_free(&buf); }
getkmp_trimmed_str_t237 const char *get() { return buf.str; }
238 };
239
240 // -----------------------------------------------------------------------------
241 // Helper parse functions.
242
__kmp_stg_parse_bool(char const * name,char const * value,int * out)243 static void __kmp_stg_parse_bool(char const *name, char const *value,
244 int *out) {
245 if (__kmp_str_match_true(value)) {
246 *out = TRUE;
247 } else if (__kmp_str_match_false(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
__kmp_check_stksize(size_t * val)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
__kmp_stg_parse_size(char const * name,char const * value,size_t size_min,size_t size_max,int * is_specified,size_t * out,size_t factor)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(value, out, factor, &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(&buf, *out);
318 KMP_WARNING(ParseSizeIntWarn, name, value, msg);
319 KMP_INFORM(Using_str_Value, name, buf.str);
320 __kmp_str_buf_free(&buf);
321 }
322 }
323 } // __kmp_stg_parse_size
324
__kmp_stg_parse_str(char const * name,char const * value,char ** out)325 static void __kmp_stg_parse_str(char const *name, char const *value,
326 char **out) {
327 __kmp_str_free(out);
328 *out = __kmp_str_format("%s", value);
329 } // __kmp_stg_parse_str
330
__kmp_stg_parse_int(char const * name,char const * value,int min,int max,int * out)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(value, &uint, &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(&buf, "%" KMP_UINT64_SPEC "", uint);
365 KMP_INFORM(Using_uint64_Value, name, buf.str);
366 __kmp_str_buf_free(&buf);
367 }
368 __kmp_type_convert(uint, out);
369 } // __kmp_stg_parse_int
370
371 #if KMP_DEBUG_ADAPTIVE_LOCKS
__kmp_stg_parse_file(char const * name,char const * value,const char * suffix,char ** out)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
__kmp_stg_parse_par_range(char const * name,char const * value,int * out_range,char * out_routine,char * out_file,int * out_lb,int * out_ub)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(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("routine", value, '=')) {
407 par_range_value = strchr(value, '=') + 1;
408 if (!par_range_value)
409 goto par_range_error;
410 value = par_range_value;
411 len = __kmp_readstr_with_sentinel(out_routine, value,
412 KMP_PAR_RANGE_ROUTINE_LEN - 1, ',');
413 if (len == 0) {
414 goto par_range_error;
415 }
416 value = strchr(value, ',');
417 if (value != NULL) {
418 value++;
419 }
420 continue;
421 }
422 if (!__kmp_strcasecmp_with_sentinel("filename", value, '=')) {
423 par_range_value = strchr(value, '=') + 1;
424 if (!par_range_value)
425 goto par_range_error;
426 value = par_range_value;
427 len = __kmp_readstr_with_sentinel(out_file, value,
428 KMP_PAR_RANGE_FILENAME_LEN - 1, ',');
429 if (len == 0) {
430 goto par_range_error;
431 }
432 value = strchr(value, ',');
433 if (value != NULL) {
434 value++;
435 }
436 continue;
437 }
438 if ((!__kmp_strcasecmp_with_sentinel("range", value, '=')) ||
439 (!__kmp_strcasecmp_with_sentinel("incl_range", value, '='))) {
440 par_range_value = strchr(value, '=') + 1;
441 if (!par_range_value)
442 goto par_range_error;
443 value = par_range_value;
444 if (KMP_SSCANF(value, "%d:%d", out_lb, out_ub) != 2) {
445 goto par_range_error;
446 }
447 *out_range = +1;
448 value = strchr(value, ',');
449 if (value != NULL) {
450 value++;
451 }
452 continue;
453 }
454 if (!__kmp_strcasecmp_with_sentinel("excl_range", value, '=')) {
455 par_range_value = strchr(value, '=') + 1;
456 if (!par_range_value)
457 goto par_range_error;
458 value = par_range_value;
459 if (KMP_SSCANF(value, "%d:%d", out_lb, out_ub) != 2) {
460 goto par_range_error;
461 }
462 *out_range = -1;
463 value = strchr(value, ',');
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
__kmp_initial_threads_capacity(int req_nproc)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
__kmp_default_tp_capacity(int req_nproc,int max_nth,int all_threads_specified)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
__kmp_stg_print_bool(kmp_str_buf_t * buffer,char const * name,int value)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, " %s=%s\n", name, value ? "true" : "false");
527 }
528 } // __kmp_stg_print_bool
529
__kmp_stg_print_int(kmp_str_buf_t * buffer,char const * name,int value)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, " %s=%d\n", name, value);
536 }
537 } // __kmp_stg_print_int
538
__kmp_stg_print_uint64(kmp_str_buf_t * buffer,char const * name,kmp_uint64 value)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, " %s=%" KMP_UINT64_SPEC "\n", name, value);
545 }
546 } // __kmp_stg_print_uint64
547
__kmp_stg_print_str(kmp_str_buf_t * buffer,char const * name,char const * value)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, " %s=%s\n", name, value);
554 }
555 } // __kmp_stg_print_str
556
__kmp_stg_print_size(kmp_str_buf_t * buffer,char const * name,size_t value)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, value);
562 __kmp_str_buf_print(buffer, "'\n");
563 } else {
564 __kmp_str_buf_print(buffer, " %s=", name);
565 __kmp_str_buf_print_size(buffer, value);
566 __kmp_str_buf_print(buffer, "\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
__kmp_stg_parse_device_thread_limit(char const * name,char const * value,void * data)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(name, "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("all", value, 0)) {
589 __kmp_max_nth = __kmp_xproc;
590 __kmp_allThreadsSpecified = 1;
591 } else {
592 __kmp_stg_parse_int(name, value, 1, __kmp_sys_max_nth, &__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
__kmp_stg_print_device_thread_limit(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_max_nth);
602 } // __kmp_stg_print_device_thread_limit
603
604 // -----------------------------------------------------------------------------
605 // OMP_THREAD_LIMIT
__kmp_stg_parse_thread_limit(char const * name,char const * value,void * data)606 static void __kmp_stg_parse_thread_limit(char const *name, char const *value,
607 void *data) {
608 __kmp_stg_parse_int(name, value, 1, __kmp_sys_max_nth, &__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
__kmp_stg_print_thread_limit(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_cg_max_nth);
616 } // __kmp_stg_print_thread_limit
617
618 // -----------------------------------------------------------------------------
619 // OMP_NUM_TEAMS
__kmp_stg_parse_nteams(char const * name,char const * value,void * data)620 static void __kmp_stg_parse_nteams(char const *name, char const *value,
621 void *data) {
622 __kmp_stg_parse_int(name, value, 1, __kmp_sys_max_nth, &__kmp_nteams);
623 K_DIAG(1, ("__kmp_nteams == %d\n", __kmp_nteams));
624 } // __kmp_stg_parse_nteams
625
__kmp_stg_print_nteams(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_nteams);
629 } // __kmp_stg_print_nteams
630
631 // -----------------------------------------------------------------------------
632 // OMP_TEAMS_THREAD_LIMIT
__kmp_stg_parse_teams_th_limit(char const * name,char const * value,void * data)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, 1, __kmp_sys_max_nth,
636 &__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
__kmp_stg_print_teams_th_limit(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_teams_thread_limit);
643 } // __kmp_stg_print_teams_th_limit
644
645 // -----------------------------------------------------------------------------
646 // KMP_TEAMS_THREAD_LIMIT
__kmp_stg_parse_teams_thread_limit(char const * name,char const * value,void * data)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, 1, __kmp_sys_max_nth, &__kmp_teams_max_nth);
650 } // __kmp_stg_teams_thread_limit
651
__kmp_stg_print_teams_thread_limit(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_teams_max_nth);
655 } // __kmp_stg_print_teams_thread_limit
656
657 // -----------------------------------------------------------------------------
658 // KMP_USE_YIELD
__kmp_stg_parse_use_yield(char const * name,char const * value,void * data)659 static void __kmp_stg_parse_use_yield(char const *name, char const *value,
660 void *data) {
661 __kmp_stg_parse_int(name, value, 0, 2, &__kmp_use_yield);
662 __kmp_use_yield_exp_set = 1;
663 } // __kmp_stg_parse_use_yield
664
__kmp_stg_print_use_yield(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_use_yield);
668 } // __kmp_stg_print_use_yield
669
670 // -----------------------------------------------------------------------------
671 // KMP_BLOCKTIME
672
__kmp_stg_parse_blocktime(char const * name,char const * value,void * data)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(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("ms", buf, &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("us", buf, &next)) {
702 // units are in us
703 __kmp_dflt_blocktime = num;
704 __kmp_blocktime_units = 'u';
705 } else if (__kmp_match_str("infinite", buf, &next) ||
706 __kmp_match_str("infinity", buf, &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
__kmp_stg_print_blocktime(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s=", name);
769 }
770 __kmp_str_buf_print(buffer, "%d", num);
771 __kmp_str_buf_print(buffer, "%cs\n", __kmp_blocktime_units);
772 } // __kmp_stg_print_blocktime
773
774 // -----------------------------------------------------------------------------
775 // KMP_DUPLICATE_LIB_OK
776
__kmp_stg_parse_duplicate_lib_ok(char const * name,char const * value,void * data)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, &__kmp_duplicate_library_ok);
782 } // __kmp_stg_parse_duplicate_lib_ok
783
__kmp_stg_print_duplicate_lib_ok(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_inherit_fp_control(char const * name,char const * value,void * data)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, &__kmp_inherit_fp_control);
797 } // __kmp_stg_parse_inherit_fp_control
798
__kmp_stg_print_inherit_fp_control(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_wait_policy(char const * name,char const * value,void * data)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, wait->rivals);
821 if (rc) {
822 return;
823 }
824
825 if (wait->omp) {
826 if (__kmp_str_match("ACTIVE", 1, 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("PASSIVE", 1, 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("serial", 1, value)) { /* S */
844 __kmp_library = library_serial;
845 } else if (__kmp_str_match("throughput", 2, 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("turnaround", 2, value)) { /* TU */
852 __kmp_library = library_turnaround;
853 } else if (__kmp_str_match("dedicated", 1, value)) { /* D */
854 __kmp_library = library_turnaround;
855 } else if (__kmp_str_match("multiuser", 1, 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
__kmp_stg_print_wait_policy(kmp_str_buf_t * buffer,char const * name,void * data)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
__kmp_stg_parse_monitor_stacksize(char const * name,char const * value,void * data)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
__kmp_stg_print_monitor_stacksize(kmp_str_buf_t * buffer,char const * name,void * data)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
__kmp_stg_parse_settings(char const * name,char const * value,void * data)942 static void __kmp_stg_parse_settings(char const *name, char const *value,
943 void *data) {
944 __kmp_stg_parse_bool(name, value, &__kmp_settings);
945 } // __kmp_stg_parse_settings
946
__kmp_stg_print_settings(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_settings);
950 } // __kmp_stg_print_settings
951
952 // -----------------------------------------------------------------------------
953 // KMP_STACKPAD
954
__kmp_stg_parse_stackpad(char const * name,char const * value,void * data)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 &__kmp_stkpadding // Var to initialize
962 );
963 } // __kmp_stg_parse_stackpad
964
__kmp_stg_print_stackpad(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_stkpadding);
968 } // __kmp_stg_print_stackpad
969
970 // -----------------------------------------------------------------------------
971 // KMP_STACKOFFSET
972
__kmp_stg_parse_stackoffset(char const * name,char const * value,void * data)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 &__kmp_stkoffset, // Var to initialize
981 1);
982 } // __kmp_stg_parse_stackoffset
983
__kmp_stg_print_stackoffset(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_stkoffset);
987 } // __kmp_stg_print_stackoffset
988
989 // -----------------------------------------------------------------------------
990 // KMP_STACKSIZE, OMP_STACKSIZE, GOMP_STACKSIZE
991
__kmp_stg_parse_stacksize(char const * name,char const * value,void * data)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, stacksize->rivals);
999 if (rc) {
1000 return;
1001 }
1002 __kmp_stg_parse_size(name, // Env var name
1003 value, // Env var value
1004 __kmp_sys_min_stksize, // Min value
1005 KMP_MAX_STKSIZE, // Max value
1006 &__kmp_env_stksize, //
1007 &__kmp_stksize, // Var to initialize
1008 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.
__kmp_stg_print_stacksize(kmp_str_buf_t * buffer,char const * name,void * data)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, (__kmp_stksize % 1024)
1022 ? __kmp_stksize / stacksize->factor
1023 : __kmp_stksize);
1024 __kmp_str_buf_print(buffer, "'\n");
1025 } else {
1026 __kmp_str_buf_print(buffer, " %s=", name);
1027 __kmp_str_buf_print_size(buffer, (__kmp_stksize % 1024)
1028 ? __kmp_stksize / stacksize->factor
1029 : __kmp_stksize);
1030 __kmp_str_buf_print(buffer, "\n");
1031 }
1032 } // __kmp_stg_print_stacksize
1033
1034 // -----------------------------------------------------------------------------
1035 // KMP_VERSION
1036
__kmp_stg_parse_version(char const * name,char const * value,void * data)1037 static void __kmp_stg_parse_version(char const *name, char const *value,
1038 void *data) {
1039 __kmp_stg_parse_bool(name, value, &__kmp_version);
1040 } // __kmp_stg_parse_version
1041
__kmp_stg_print_version(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_version);
1045 } // __kmp_stg_print_version
1046
1047 // -----------------------------------------------------------------------------
1048 // KMP_WARNINGS
1049
__kmp_stg_parse_warnings(char const * name,char const * value,void * data)1050 static void __kmp_stg_parse_warnings(char const *name, char const *value,
1051 void *data) {
1052 __kmp_stg_parse_bool(name, value, &__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
__kmp_stg_print_warnings(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_generate_warnings);
1064 } // __kmp_stg_print_warnings
1065
1066 // -----------------------------------------------------------------------------
1067 // KMP_NESTING_MODE
1068
__kmp_stg_parse_nesting_mode(char const * name,char const * value,void * data)1069 static void __kmp_stg_parse_nesting_mode(char const *name, char const *value,
1070 void *data) {
1071 __kmp_stg_parse_int(name, value, 0, INT_MAX, &__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
__kmp_stg_print_nesting_mode(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s", name);
1084 }
1085 __kmp_str_buf_print(buffer, "=%d\n", __kmp_nesting_mode);
1086 } // __kmp_stg_print_nesting_mode
1087
1088 // -----------------------------------------------------------------------------
1089 // OMP_NESTED, OMP_NUM_THREADS
1090
__kmp_stg_parse_nested(char const * name,char const * value,void * data)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, &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
__kmp_stg_print_nested(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s", name);
1111 }
1112 __kmp_str_buf_print(buffer, ": deprecated; max-active-levels-var=%d\n",
1113 __kmp_dflt_max_active_levels);
1114 } // __kmp_stg_print_nested
1115
__kmp_parse_nested_num_threads(const char * var,const char * env,kmp_nested_nthreads_t * nth_array)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(buf, *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
__kmp_stg_parse_num_threads(char const * name,char const * value,void * data)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("all", value, 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(name, value, &__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
__kmp_stg_parse_max_tdgs(char const * name,char const * value,void * data)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
__kmp_std_print_max_tdgs(kmp_str_buf_t * buffer,char const * name,void * data)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
__kmp_stg_parse_tdg_dot(char const * name,char const * value,void * data)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
__kmp_stg_print_tdg_dot(kmp_str_buf_t * buffer,char const * name,void * data)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
__kmp_stg_parse_num_hidden_helper_threads(char const * name,char const * value,void * data)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, 0, 16, &__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
__kmp_stg_print_num_hidden_helper_threads(kmp_str_buf_t * buffer,char const * name,void * data)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, __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, __kmp_hidden_helper_threads_num - 1);
1315 }
1316 } // __kmp_stg_print_num_hidden_helper_threads
1317
__kmp_stg_parse_use_hidden_helper(char const * name,char const * value,void * data)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, &__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
__kmp_stg_print_use_hidden_helper(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_enable_hidden_helper);
1332 } // __kmp_stg_print_use_hidden_helper
1333
__kmp_stg_print_num_threads(kmp_str_buf_t * buffer,char const * name,void * data)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, " %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(&buf, "%d", __kmp_nested_nth.nth[i]);
1346 if (i < __kmp_nested_nth.used - 1) {
1347 __kmp_str_buf_print(&buf, ",");
1348 }
1349 }
1350 __kmp_str_buf_print(buffer, "='%s'\n", buf.str);
1351 __kmp_str_buf_free(&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
__kmp_stg_parse_tasking(char const * name,char const * value,void * data)1360 static void __kmp_stg_parse_tasking(char const *name, char const *value,
1361 void *data) {
1362 __kmp_stg_parse_int(name, value, 0, (int)tskm_max,
1363 (int *)&__kmp_tasking_mode);
1364 } // __kmp_stg_parse_tasking
1365
__kmp_stg_print_tasking(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_tasking_mode);
1369 } // __kmp_stg_print_tasking
1370
__kmp_stg_parse_task_stealing(char const * name,char const * value,void * data)1371 static void __kmp_stg_parse_task_stealing(char const *name, char const *value,
1372 void *data) {
1373 __kmp_stg_parse_int(name, value, 0, 1,
1374 (int *)&__kmp_task_stealing_constraint);
1375 } // __kmp_stg_parse_task_stealing
1376
__kmp_stg_print_task_stealing(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_task_stealing_constraint);
1380 } // __kmp_stg_print_task_stealing
1381
__kmp_stg_parse_max_active_levels(char const * name,char const * value,void * data)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(value, &tmp_dflt, &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(tmp_dflt, &(__kmp_dflt_max_active_levels));
1397 __kmp_dflt_max_active_levels_set = true;
1398 }
1399 }
1400 } // __kmp_stg_parse_max_active_levels
1401
__kmp_stg_print_max_active_levels(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_dflt_max_active_levels);
1405 } // __kmp_stg_print_max_active_levels
1406
1407 // -----------------------------------------------------------------------------
1408 // OpenMP 4.0: OMP_DEFAULT_DEVICE
__kmp_stg_parse_default_device(char const * name,char const * value,void * data)1409 static void __kmp_stg_parse_default_device(char const *name, char const *value,
1410 void *data) {
1411 __kmp_stg_parse_int(name, value, 0, KMP_MAX_DEFAULT_DEVICE_LIMIT,
1412 &__kmp_default_device);
1413 } // __kmp_stg_parse_default_device
1414
__kmp_stg_print_default_device(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_default_device);
1418 } // __kmp_stg_print_default_device
1419
1420 // -----------------------------------------------------------------------------
1421 // OpenMP 5.0: OMP_TARGET_OFFLOAD
__kmp_stg_parse_target_offload(char const * name,char const * value,void * data)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("mandatory", scan, 0)) {
1432 __kmp_target_offload = tgt_mandatory;
1433 } else if (!__kmp_strcasecmp_with_sentinel("disabled", scan, 0)) {
1434 __kmp_target_offload = tgt_disabled;
1435 } else if (!__kmp_strcasecmp_with_sentinel("default", scan, 0)) {
1436 __kmp_target_offload = tgt_default;
1437 } else {
1438 KMP_WARNING(SyntaxErrorUsing, name, "DEFAULT");
1439 }
1440 } // __kmp_stg_parse_target_offload
1441
__kmp_stg_print_target_offload(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s", name);
1456 }
1457 __kmp_str_buf_print(buffer, "=%s\n", value);
1458 } // __kmp_stg_print_target_offload
1459
1460 // -----------------------------------------------------------------------------
1461 // OpenMP 4.5: OMP_MAX_TASK_PRIORITY
__kmp_stg_parse_max_task_priority(char const * name,char const * value,void * data)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, 0, KMP_MAX_TASK_PRIORITY_LIMIT,
1465 &__kmp_max_task_priority);
1466 } // __kmp_stg_parse_max_task_priority
1467
__kmp_stg_print_max_task_priority(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_taskloop_min_tasks(char const * name,char const * value,void * data)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, 0, INT_MAX, &tmp);
1479 __kmp_taskloop_min_tasks = tmp;
1480 } // __kmp_stg_parse_taskloop_min_tasks
1481
__kmp_stg_print_taskloop_min_tasks(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_taskloop_min_tasks);
1485 } // __kmp_stg_print_taskloop_min_tasks
1486
1487 // -----------------------------------------------------------------------------
1488 // KMP_DISP_NUM_BUFFERS
__kmp_stg_parse_disp_buffers(char const * name,char const * value,void * data)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 &__kmp_dispatch_num_buffers);
1497 } // __kmp_stg_parse_disp_buffers
1498
__kmp_stg_print_disp_buffers(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_hot_teams_level(char const * name,char const * value,void * data)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, 0, KMP_MAX_ACTIVE_LEVELS_LIMIT,
1515 &__kmp_hot_teams_max_level);
1516 } // __kmp_stg_parse_hot_teams_level
1517
__kmp_stg_print_hot_teams_level(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_hot_teams_max_level);
1521 } // __kmp_stg_print_hot_teams_level
1522
__kmp_stg_parse_hot_teams_mode(char const * name,char const * value,void * data)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, 0, KMP_MAX_ACTIVE_LEVELS_LIMIT,
1530 &__kmp_hot_teams_mode);
1531 } // __kmp_stg_parse_hot_teams_mode
1532
__kmp_stg_print_hot_teams_mode(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_handle_signals(char const * name,char const * value,void * data)1545 static void __kmp_stg_parse_handle_signals(char const *name, char const *value,
1546 void *data) {
1547 __kmp_stg_parse_bool(name, value, &__kmp_handle_signals);
1548 } // __kmp_stg_parse_handle_signals
1549
__kmp_stg_print_handle_signals(kmp_str_buf_t * buffer,char const * name,void * data)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, __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)
KMP_STG_X_DEBUG(b)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, 0, INT_MAX, &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
__kmp_stg_parse_debug_buf(char const * name,char const * value,void * data)1605 static void __kmp_stg_parse_debug_buf(char const *name, char const *value,
1606 void *data) {
1607 __kmp_stg_parse_bool(name, value, &__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
__kmp_stg_print_debug_buf(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_debug_buf);
1628 } // __kmp_stg_print_debug_buf
1629
__kmp_stg_parse_debug_buf_atomic(char const * name,char const * value,void * data)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, &__kmp_debug_buf_atomic);
1633 } // __kmp_stg_parse_debug_buf_atomic
1634
__kmp_stg_print_debug_buf_atomic(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_debug_buf_atomic);
1638 } // __kmp_stg_print_debug_buf_atomic
1639
__kmp_stg_parse_debug_buf_chars(char const * name,char const * value,void * data)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 &__kmp_debug_buf_chars);
1644 } // __kmp_stg_debug_parse_buf_chars
1645
__kmp_stg_print_debug_buf_chars(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_debug_buf_chars);
1649 } // __kmp_stg_print_debug_buf_chars
1650
__kmp_stg_parse_debug_buf_lines(char const * name,char const * value,void * data)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 &__kmp_debug_buf_lines);
1655 } // __kmp_stg_parse_debug_buf_lines
1656
__kmp_stg_print_debug_buf_lines(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_debug_buf_lines);
1660 } // __kmp_stg_print_debug_buf_lines
1661
__kmp_stg_parse_diag(char const * name,char const * value,void * data)1662 static void __kmp_stg_parse_diag(char const *name, char const *value,
1663 void *data) {
1664 __kmp_stg_parse_int(name, value, 0, INT_MAX, &kmp_diag);
1665 } // __kmp_stg_parse_diag
1666
__kmp_stg_print_diag(kmp_str_buf_t * buffer,char const * name,void * data)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, kmp_diag);
1670 } // __kmp_stg_print_diag
1671
1672 #endif // KMP_DEBUG
1673
1674 // -----------------------------------------------------------------------------
1675 // KMP_ALIGN_ALLOC
1676
__kmp_stg_parse_align_alloc(char const * name,char const * value,void * data)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 &__kmp_align_alloc, 1);
1681 } // __kmp_stg_parse_align_alloc
1682
__kmp_stg_print_align_alloc(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_barrier_branch_bit(char const * name,char const * value,void * data)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(var, 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(value, ',');
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(comma + 1, 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
__kmp_stg_print_barrier_branch_bit(kmp_str_buf_t * buffer,char const * name,void * data)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(var, 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, " %s='",
1743 __kmp_barrier_branch_bit_env_name[i]);
1744 }
1745 __kmp_str_buf_print(buffer, "%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
__kmp_stg_parse_barrier_pattern(char const * name,char const * value,void * data)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(var, 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(__kmp_barrier_pattern_name[j], value, 1,
1776 ',')) {
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(__kmp_barrier_pattern_name[j], 1, 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
__kmp_stg_print_barrier_pattern(kmp_str_buf_t * buffer,char const * name,void * data)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(var, 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, " %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, "%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
__kmp_stg_parse_abort_delay(char const * name,char const * value,void * data)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, 0, INT_MAX / 1000, &delay);
1861 __kmp_abort_delay = delay * 1000;
1862 } // __kmp_stg_parse_abort_delay
1863
__kmp_stg_print_abort_delay(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_abort_delay);
1867 } // __kmp_stg_print_abort_delay
1868
1869 // -----------------------------------------------------------------------------
1870 // KMP_CPUINFO_FILE
1871
__kmp_stg_parse_cpuinfo_file(char const * name,char const * value,void * data)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, &__kmp_cpuinfo_file);
1876 K_DIAG(1, ("__kmp_cpuinfo_file == %s\n", __kmp_cpuinfo_file));
1877 #endif
1878 } //__kmp_stg_parse_cpuinfo_file
1879
__kmp_stg_print_cpuinfo_file(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s", name);
1887 }
1888 if (__kmp_cpuinfo_file) {
1889 __kmp_str_buf_print(buffer, "='%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
__kmp_stg_parse_force_reduction(char const * name,char const * value,void * data)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, reduction->rivals);
1905 if (rc) {
1906 return;
1907 }
1908 if (reduction->force) {
1909 if (value != 0) {
1910 if (__kmp_str_match("critical", 0, value))
1911 __kmp_force_reduction_method = critical_reduce_block;
1912 else if (__kmp_str_match("atomic", 0, value))
1913 __kmp_force_reduction_method = atomic_reduce_block;
1914 else if (__kmp_str_match("tree", 0, 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, &__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
__kmp_stg_print_force_reduction(kmp_str_buf_t * buffer,char const * name,void * data)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, "critical");
1939 } else if (__kmp_force_reduction_method == atomic_reduce_block) {
1940 __kmp_stg_print_str(buffer, name, "atomic");
1941 } else if (__kmp_force_reduction_method == tree_reduce_block) {
1942 __kmp_stg_print_str(buffer, name, "tree");
1943 } else {
1944 if (__kmp_env_format) {
1945 KMP_STR_BUF_PRINT_NAME;
1946 } else {
1947 __kmp_str_buf_print(buffer, " %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, __kmp_determ_red);
1953 }
1954
1955 } // __kmp_stg_print_force_reduction
1956
1957 // -----------------------------------------------------------------------------
1958 // KMP_STORAGE_MAP
1959
__kmp_stg_parse_storage_map(char const * name,char const * value,void * data)1960 static void __kmp_stg_parse_storage_map(char const *name, char const *value,
1961 void *data) {
1962 if (__kmp_str_match("verbose", 1, 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, &__kmp_storage_map); // !!!
1970 }
1971 } // __kmp_stg_parse_storage_map
1972
__kmp_stg_print_storage_map(kmp_str_buf_t * buffer,char const * name,void * data)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, "verbose");
1977 } else {
1978 __kmp_stg_print_bool(buffer, name, __kmp_storage_map);
1979 }
1980 } // __kmp_stg_print_storage_map
1981
1982 // -----------------------------------------------------------------------------
1983 // KMP_ALL_THREADPRIVATE
1984
__kmp_stg_parse_all_threadprivate(char const * name,char const * value,void * data)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 __kmp_allThreadsSpecified ? __kmp_max_nth : 1,
1989 __kmp_max_nth, &__kmp_tp_capacity);
1990 } // __kmp_stg_parse_all_threadprivate
1991
__kmp_stg_print_all_threadprivate(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_tp_capacity);
1995 }
1996
1997 // -----------------------------------------------------------------------------
1998 // KMP_FOREIGN_THREADS_THREADPRIVATE
1999
__kmp_stg_parse_foreign_threads_threadprivate(char const * name,char const * value,void * data)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, &__kmp_foreign_tp);
2004 } // __kmp_stg_parse_foreign_threads_threadprivate
2005
__kmp_stg_print_foreign_threads_threadprivate(kmp_str_buf_t * buffer,char const * name,void * data)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, __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 *
__kmp_hw_get_core_type_keyword(kmp_hw_core_type_t type)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.
__kmp_parse_affinity_proc_id_list(const char * var,const char * env,const char ** nextEnv,char ** proclist)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(scan, *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(scan, *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(scan, *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(scan, *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(scan, *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
__kmp_parse_affinity_env(char const * name,char const * value,kmp_affinity_t * out_affinity)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, 0);
2238 return;
2239 }
2240 __kmp_env_toPrint(name, 1);
2241
2242 buffer =
2243 __kmp_str_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("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("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("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("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("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("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("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("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("verbose", buf, CCAST(const char **, &next))) {
2339 set_verbose(TRUE);
2340 buf = next;
2341 } else if (__kmp_match_str("noverbose", buf, CCAST(const char **, &next))) {
2342 set_verbose(FALSE);
2343 buf = next;
2344 } else if (__kmp_match_str("warnings", buf, CCAST(const char **, &next))) {
2345 set_warnings(TRUE);
2346 buf = next;
2347 } else if (__kmp_match_str("nowarnings", buf,
2348 CCAST(const char **, &next))) {
2349 set_warnings(FALSE);
2350 buf = next;
2351 } else if (__kmp_match_str("respect", buf, CCAST(const char **, &next))) {
2352 set_respect(TRUE);
2353 buf = next;
2354 } else if (__kmp_match_str("norespect", buf, CCAST(const char **, &next))) {
2355 set_respect(FALSE);
2356 buf = next;
2357 } else if (__kmp_match_str("reset", buf, CCAST(const char **, &next))) {
2358 set_reset(TRUE);
2359 buf = next;
2360 } else if (__kmp_match_str("noreset", buf, CCAST(const char **, &next))) {
2361 set_reset(FALSE);
2362 buf = next;
2363 } else if (__kmp_match_str("duplicates", buf,
2364 CCAST(const char **, &next)) ||
2365 __kmp_match_str("dups", buf, CCAST(const char **, &next))) {
2366 set_dups(TRUE);
2367 buf = next;
2368 } else if (__kmp_match_str("noduplicates", buf,
2369 CCAST(const char **, &next)) ||
2370 __kmp_match_str("nodups", buf, CCAST(const char **, &next))) {
2371 set_dups(FALSE);
2372 buf = next;
2373 } else if (__kmp_match_str("granularity", buf,
2374 CCAST(const char **, &next)) ||
2375 __kmp_match_str("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("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("core_efficiency", buf,
2394 CCAST(const char **, &next)) ||
2395 __kmp_match_str("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(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("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("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("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(buf, *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("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 name, buf, CCAST(const char **, &next), &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(buf, *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(&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
__kmp_stg_parse_affinity(char const * name,char const * value,void * data)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, &__kmp_affinity);
2624
2625 } // __kmp_stg_parse_affinity
__kmp_stg_parse_hh_affinity(char const * name,char const * value,void * data)2626 static void __kmp_stg_parse_hh_affinity(char const *name, char const *value,
2627 void *data) {
2628 __kmp_parse_affinity_env(name, value, &__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
__kmp_print_affinity_env(kmp_str_buf_t * buffer,char const * name,const kmp_affinity_t & affinity)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, " %s='", name);
2645 }
2646 if (affinity.flags.verbose) {
2647 __kmp_str_buf_print(buffer, "%s,", "verbose");
2648 } else {
2649 __kmp_str_buf_print(buffer, "%s,", "noverbose");
2650 }
2651 if (affinity.flags.warnings) {
2652 __kmp_str_buf_print(buffer, "%s,", "warnings");
2653 } else {
2654 __kmp_str_buf_print(buffer, "%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, "%s,", "respect");
2662 } else {
2663 __kmp_str_buf_print(buffer, "%s,", "norespect");
2664 }
2665 if (affinity.flags.reset) {
2666 __kmp_str_buf_print(buffer, "%s,", "reset");
2667 } else {
2668 __kmp_str_buf_print(buffer, "%s,", "noreset");
2669 }
2670 }
2671 __kmp_str_buf_print(buffer, "granularity=");
2672 if (affinity.flags.core_types_gran)
2673 __kmp_str_buf_print(buffer, "core_type,");
2674 else if (affinity.flags.core_effs_gran) {
2675 __kmp_str_buf_print(buffer, "core_eff,");
2676 } else {
2677 __kmp_str_buf_print(
2678 buffer, "%s,", __kmp_hw_get_keyword(affinity.gran, /*plural=*/false));
2679 }
2680 }
2681 if (!KMP_AFFINITY_CAPABLE()) {
2682 __kmp_str_buf_print(buffer, "%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, "%s", "none");
2689 break;
2690 case affinity_physical:
2691 __kmp_str_buf_print(buffer, "%s,%d", "physical", offset);
2692 break;
2693 case affinity_logical:
2694 __kmp_str_buf_print(buffer, "%s,%d", "logical", offset);
2695 break;
2696 case affinity_compact:
2697 __kmp_str_buf_print(buffer, "%s,%d,%d", "compact", compact, offset);
2698 break;
2699 case affinity_scatter:
2700 __kmp_str_buf_print(buffer, "%s,%d,%d", "scatter", compact, offset);
2701 break;
2702 case affinity_explicit:
2703 __kmp_str_buf_print(buffer, "%s=[%s],%s", "proclist", affinity.proclist,
2704 "explicit");
2705 break;
2706 case affinity_balanced:
2707 __kmp_str_buf_print(buffer, "%s,%d,%d", "balanced", compact, offset);
2708 break;
2709 case affinity_disabled:
2710 __kmp_str_buf_print(buffer, "%s", "disabled");
2711 break;
2712 case affinity_default:
2713 __kmp_str_buf_print(buffer, "%s", "default");
2714 break;
2715 default:
2716 __kmp_str_buf_print(buffer, "%s", "<unknown>");
2717 break;
2718 }
2719 }
2720 __kmp_str_buf_print(buffer, "'\n");
2721 } //__kmp_stg_print_affinity
2722
__kmp_stg_print_affinity(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_affinity);
2726 }
__kmp_stg_print_hh_affinity(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_hh_affinity);
2730 }
2731
2732 #ifdef KMP_GOMP_COMPAT
2733
__kmp_stg_parse_gomp_cpu_affinity(char const * name,char const * value,void * data)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, 0);
2749 return;
2750 }
2751
2752 __kmp_env_toPrint(name, 1);
2753
2754 if (__kmp_parse_affinity_proc_id_list(name, value, &next, &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
__kmp_parse_subplace_list(const char * var,const char ** scan)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(*scan, *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(*scan, *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(*scan, *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
__kmp_parse_place(const char * var,const char ** scan)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(*scan, *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
__kmp_parse_place_list(const char * var,const char * env,char ** place_list)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)) {
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(scan, *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(scan, *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
__kmp_places_set(enum affinity_type type,kmp_hw_t kind)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
__kmp_places_syntax_error_fallback(char const * name,kmp_hw_t kind)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(kind, /*plural=*/true);
3037 KMP_WARNING(SyntaxErrorUsing, name, str);
3038 __kmp_places_set(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
__kmp_stg_parse_places(char const * name,char const * value,void * data)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[] = {{"threads", KMP_HW_THREAD},
3054 {"cores", KMP_HW_CORE},
3055 {"numa_domains", KMP_HW_NUMA},
3056 {"ll_caches", KMP_HW_LLC},
3057 {"sockets", 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(place.name, scan, &next)) {
3070 scan = next;
3071 __kmp_places_set(affinity_compact, 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, 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("intel_core", scan, &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("intel_atom", scan, &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("eff", scan, &next)) {
3095 int eff;
3096 if (!isdigit(*next)) {
3097 __kmp_places_syntax_error_fallback(name, place.type);
3098 return;
3099 }
3100 scan = next;
3101 SKIP_DIGITS(next);
3102 eff = __kmp_str_to_int(scan, *next);
3103 if (eff < 0) {
3104 __kmp_places_syntax_error_fallback(name, 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, 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, true);
3126 if (__kmp_match_str("unknowns", scan, &next))
3127 continue;
3128 if (__kmp_match_str(name, scan, &next)) {
3129 scan = next;
3130 __kmp_places_set(affinity_compact, 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("core_types", scan, &next)) {
3139 scan = next;
3140 if (*scan != '\0') {
3141 KMP_WARNING(ParseExtraCharsWarn, name, scan);
3142 }
3143 __kmp_places_set(affinity_compact, KMP_HW_CORE);
3144 __kmp_affinity.flags.core_types_gran = 1;
3145 set = true;
3146 } else if (__kmp_match_str("core_effs", scan, &next) ||
3147 __kmp_match_str("core_efficiencies", scan, &next)) {
3148 scan = next;
3149 if (*scan != '\0') {
3150 KMP_WARNING(ParseExtraCharsWarn, name, scan);
3151 }
3152 __kmp_places_set(affinity_compact, 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(name, value, &__kmp_affinity.proclist)) {
3164 __kmp_places_set(affinity_explicit, KMP_HW_THREAD);
3165 } else {
3166 // Syntax error fallback
3167 __kmp_places_syntax_error_fallback(name, 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, 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(scan, *next);
3202 KMP_ASSERT(count >= 0);
3203 scan = next;
3204
3205 SKIP_WS(scan);
3206 if (*scan != ')') {
3207 __kmp_places_syntax_error_fallback(name, 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
__kmp_stg_print_places(kmp_str_buf_t * buffer,char const * name,void * data)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, " %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, "='%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, "='%s'\n", "core_types");
3253 return;
3254 }
3255 if (__kmp_affinity.flags.core_effs_gran) {
3256 __kmp_str_buf_print(buffer, "='%s'\n", "core_effs");
3257 return;
3258 }
3259
3260 // threads, cores, sockets, cores:<attribute>, etc.
3261 const char *name = __kmp_hw_get_keyword(gran, true);
3262 __kmp_str_buf_print(buffer, "='%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(ct);
3271 __kmp_str_buf_print(buffer, ":%s", name, ct_name);
3272 } else if (eff >= 0 && eff < KMP_HW_MAX_NUM_CORE_EFFS) {
3273 __kmp_str_buf_print(buffer, ":eff%d", name, eff);
3274 }
3275 }
3276
3277 // Add the '(#)' part if it exists
3278 if (num > 0)
3279 __kmp_str_buf_print(buffer, "(%d)", num);
3280 __kmp_str_buf_print(buffer, "'\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
__kmp_stg_parse_topology_method(char const * name,char const * value,void * data)3289 static void __kmp_stg_parse_topology_method(char const *name, char const *value,
3290 void *data) {
3291 if (__kmp_str_match("all", 1, 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("cpuid_leaf31", 12, value) ||
3301 __kmp_str_match("cpuid 1f", 8, value) ||
3302 __kmp_str_match("cpuid 31", 8, value) ||
3303 __kmp_str_match("cpuid1f", 7, value) ||
3304 __kmp_str_match("cpuid31", 7, value) ||
3305 __kmp_str_match("leaf 1f", 7, value) ||
3306 __kmp_str_match("leaf 31", 7, value) ||
3307 __kmp_str_match("leaf1f", 6, value) ||
3308 __kmp_str_match("leaf31", 6, value)) {
3309 __kmp_affinity_top_method = affinity_top_method_x2apicid_1f;
3310 } else if (__kmp_str_match("x2apic id", 9, value) ||
3311 __kmp_str_match("x2apic_id", 9, value) ||
3312 __kmp_str_match("x2apic-id", 9, value) ||
3313 __kmp_str_match("x2apicid", 8, value) ||
3314 __kmp_str_match("cpuid leaf 11", 13, value) ||
3315 __kmp_str_match("cpuid_leaf_11", 13, value) ||
3316 __kmp_str_match("cpuid-leaf-11", 13, value) ||
3317 __kmp_str_match("cpuid leaf11", 12, value) ||
3318 __kmp_str_match("cpuid_leaf11", 12, value) ||
3319 __kmp_str_match("cpuid-leaf11", 12, value) ||
3320 __kmp_str_match("cpuidleaf 11", 12, value) ||
3321 __kmp_str_match("cpuidleaf_11", 12, value) ||
3322 __kmp_str_match("cpuidleaf-11", 12, value) ||
3323 __kmp_str_match("cpuidleaf11", 11, value) ||
3324 __kmp_str_match("cpuid 11", 8, value) ||
3325 __kmp_str_match("cpuid_11", 8, value) ||
3326 __kmp_str_match("cpuid-11", 8, value) ||
3327 __kmp_str_match("cpuid11", 7, value) ||
3328 __kmp_str_match("leaf 11", 7, value) ||
3329 __kmp_str_match("leaf_11", 7, value) ||
3330 __kmp_str_match("leaf-11", 7, value) ||
3331 __kmp_str_match("leaf11", 6, value)) {
3332 __kmp_affinity_top_method = affinity_top_method_x2apicid;
3333 } else if (__kmp_str_match("apic id", 7, value) ||
3334 __kmp_str_match("apic_id", 7, value) ||
3335 __kmp_str_match("apic-id", 7, value) ||
3336 __kmp_str_match("apicid", 6, value) ||
3337 __kmp_str_match("cpuid leaf 4", 12, value) ||
3338 __kmp_str_match("cpuid_leaf_4", 12, value) ||
3339 __kmp_str_match("cpuid-leaf-4", 12, value) ||
3340 __kmp_str_match("cpuid leaf4", 11, value) ||
3341 __kmp_str_match("cpuid_leaf4", 11, value) ||
3342 __kmp_str_match("cpuid-leaf4", 11, value) ||
3343 __kmp_str_match("cpuidleaf 4", 11, value) ||
3344 __kmp_str_match("cpuidleaf_4", 11, value) ||
3345 __kmp_str_match("cpuidleaf-4", 11, value) ||
3346 __kmp_str_match("cpuidleaf4", 10, value) ||
3347 __kmp_str_match("cpuid 4", 7, value) ||
3348 __kmp_str_match("cpuid_4", 7, value) ||
3349 __kmp_str_match("cpuid-4", 7, value) ||
3350 __kmp_str_match("cpuid4", 6, value) ||
3351 __kmp_str_match("leaf 4", 6, value) ||
3352 __kmp_str_match("leaf_4", 6, value) ||
3353 __kmp_str_match("leaf-4", 6, value) ||
3354 __kmp_str_match("leaf4", 5, 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("/proc/cpuinfo", 2, value) ||
3359 __kmp_str_match("cpuinfo", 5, 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("flat", 1, 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
__kmp_stg_print_topology_method(kmp_str_buf_t * buffer,char const * name,void * data)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 {"spread", proc_bind_spread},
3435 {"true", proc_bind_spread},
3436 {"close", proc_bind_close},
3437 // teams-bind = false means "replicate the primary thread's affinity"
3438 {"false", proc_bind_primary},
3439 {"primary", proc_bind_primary}};
__kmp_stg_parse_teams_proc_bind(char const * name,char const * value,void * data)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(proc_bind_table[i].name, value, &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 }
__kmp_stg_print_teams_proc_bind(kmp_str_buf_t * buffer,char const * name,void * data)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.
__kmp_stg_parse_proc_bind(char const * name,char const * value,void * data)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(buf, *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("disabled", buf, &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("false", buf, &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("true", buf, &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("master", buf, &next) ||
3559 __kmp_match_str("primary", buf, &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("close", buf, &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("spread", buf, &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(buf, *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
__kmp_stg_print_proc_bind(kmp_str_buf_t * buffer,char const * name,void * data)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, " %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, "='", 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, "false");
3624 break;
3625
3626 case proc_bind_true:
3627 __kmp_str_buf_print(buffer, "true");
3628 break;
3629
3630 case proc_bind_primary:
3631 __kmp_str_buf_print(buffer, "primary");
3632 break;
3633
3634 case proc_bind_close:
3635 __kmp_str_buf_print(buffer, "close");
3636 break;
3637
3638 case proc_bind_spread:
3639 __kmp_str_buf_print(buffer, "spread");
3640 break;
3641
3642 case proc_bind_intel:
3643 __kmp_str_buf_print(buffer, "intel");
3644 break;
3645
3646 case proc_bind_default:
3647 __kmp_str_buf_print(buffer, "default");
3648 break;
3649 }
3650 if (i < nelem - 1) {
3651 __kmp_str_buf_print(buffer, ",");
3652 }
3653 }
3654 __kmp_str_buf_print(buffer, "'\n");
3655 }
3656 }
3657
__kmp_stg_parse_display_affinity(char const * name,char const * value,void * data)3658 static void __kmp_stg_parse_display_affinity(char const *name,
3659 char const *value, void *data) {
3660 __kmp_stg_parse_bool(name, value, &__kmp_display_affinity);
3661 }
__kmp_stg_print_display_affinity(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_display_affinity);
3665 }
__kmp_stg_parse_affinity_format(char const * name,char const * value,void * data)3666 static void __kmp_stg_parse_affinity_format(char const *name, char const *value,
3667 void *data) {
3668 size_t length = KMP_STRLEN(value);
3669 __kmp_strncpy_truncate(__kmp_affinity_format, KMP_AFFINITY_FORMAT_SIZE, value,
3670 length);
3671 }
__kmp_stg_print_affinity_format(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s='", name);
3678 }
3679 __kmp_str_buf_print(buffer, "%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
__kmp_stg_parse_allocator(char const * name,char const * value,void * data)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(buf, ':');
3714 const char *predef_mem_space = strstr(buf, "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("fb_data", scan, &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("omp_high_bw_mem_alloc", scan, &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("omp_large_cap_mem_alloc", scan, &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("omp_default_mem_alloc", scan, &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("omp_const_mem_alloc", scan, &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("omp_low_lat_mem_alloc", scan, &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("omp_cgroup_mem_alloc", scan, &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("omp_pteam_mem_alloc", scan, &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("omp_thread_mem_alloc", scan, &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("omp_default_mem_space", scan, &next)) {
3863 SKIP_WS(next);
3864 ms = omp_default_mem_space;
3865 } else if (__kmp_match_str("omp_large_cap_mem_space", scan, &next)) {
3866 SKIP_WS(next);
3867 ms = omp_large_cap_mem_space;
3868 } else if (__kmp_match_str("omp_const_mem_space", scan, &next)) {
3869 SKIP_WS(next);
3870 ms = omp_const_mem_space;
3871 } else if (__kmp_match_str("omp_high_bw_mem_space", scan, &next)) {
3872 SKIP_WS(next);
3873 ms = omp_high_bw_mem_space;
3874 } else if (__kmp_match_str("omp_low_lat_mem_space", scan, &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("sync_hint", scan, &next)) {
3891 GET_NEXT('=');
3892 traits[count].key = omp_atk_sync_hint;
3893 if (__kmp_match_str("contended", scan, &next)) {
3894 traits[count].value = omp_atv_contended;
3895 } else if (__kmp_match_str("uncontended", scan, &next)) {
3896 traits[count].value = omp_atv_uncontended;
3897 } else if (__kmp_match_str("serialized", scan, &next)) {
3898 traits[count].value = omp_atv_serialized;
3899 } else if (__kmp_match_str("private", scan, &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("alignment", scan, &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(scan, ',');
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("access", scan, &next)) {
3923 GET_NEXT('=');
3924 traits[count].key = omp_atk_access;
3925 if (__kmp_match_str("all", scan, &next)) {
3926 traits[count].value = omp_atv_all;
3927 } else if (__kmp_match_str("cgroup", scan, &next)) {
3928 traits[count].value = omp_atv_cgroup;
3929 } else if (__kmp_match_str("pteam", scan, &next)) {
3930 traits[count].value = omp_atv_pteam;
3931 } else if (__kmp_match_str("thread", scan, &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("pool_size", scan, &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(scan, ',');
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("fallback", scan, &next)) {
3955 GET_NEXT('=');
3956 traits[count].key = omp_atk_fallback;
3957 if (__kmp_match_str("default_mem_fb", scan, &next)) {
3958 traits[count].value = omp_atv_default_mem_fb;
3959 } else if (__kmp_match_str("null_fb", scan, &next)) {
3960 traits[count].value = omp_atv_null_fb;
3961 } else if (__kmp_match_str("abort_fb", scan, &next)) {
3962 traits[count].value = omp_atv_abort_fb;
3963 } else if (__kmp_match_str("allocator_fb", scan, &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("pinned", scan, &next)) {
3971 GET_NEXT('=');
3972 traits[count].key = omp_atk_pinned;
3973 if (__kmp_str_match_true(next)) {
3974 traits[count].value = omp_atv_true;
3975 } else if (__kmp_str_match_false(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("partition", scan, &next)) {
3983 GET_NEXT('=');
3984 traits[count].key = omp_atk_partition;
3985 if (__kmp_match_str("environment", scan, &next)) {
3986 traits[count].value = omp_atv_environment;
3987 } else if (__kmp_match_str("nearest", scan, &next)) {
3988 traits[count].value = omp_atv_nearest;
3989 } else if (__kmp_match_str("blocked", scan, &next)) {
3990 traits[count].value = omp_atv_blocked;
3991 } else if (__kmp_match_str("interleaved", scan, &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
__kmp_stg_print_allocator(kmp_str_buf_t * buffer,char const * name,void * data)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, "omp_default_mem_alloc");
4019 } else if (__kmp_def_allocator == omp_high_bw_mem_alloc) {
4020 __kmp_stg_print_str(buffer, name, "omp_high_bw_mem_alloc");
4021 } else if (__kmp_def_allocator == omp_large_cap_mem_alloc) {
4022 __kmp_stg_print_str(buffer, name, "omp_large_cap_mem_alloc");
4023 } else if (__kmp_def_allocator == omp_const_mem_alloc) {
4024 __kmp_stg_print_str(buffer, name, "omp_const_mem_alloc");
4025 } else if (__kmp_def_allocator == omp_low_lat_mem_alloc) {
4026 __kmp_stg_print_str(buffer, name, "omp_low_lat_mem_alloc");
4027 } else if (__kmp_def_allocator == omp_cgroup_mem_alloc) {
4028 __kmp_stg_print_str(buffer, name, "omp_cgroup_mem_alloc");
4029 } else if (__kmp_def_allocator == omp_pteam_mem_alloc) {
4030 __kmp_stg_print_str(buffer, name, "omp_pteam_mem_alloc");
4031 } else if (__kmp_def_allocator == omp_thread_mem_alloc) {
4032 __kmp_stg_print_str(buffer, name, "omp_thread_mem_alloc");
4033 }
4034 }
4035
4036 // -----------------------------------------------------------------------------
4037 // OMP_DYNAMIC
4038
__kmp_stg_parse_omp_dynamic(char const * name,char const * value,void * data)4039 static void __kmp_stg_parse_omp_dynamic(char const *name, char const *value,
4040 void *data) {
4041 __kmp_stg_parse_bool(name, value, &(__kmp_global.g.g_dynamic));
4042 } // __kmp_stg_parse_omp_dynamic
4043
__kmp_stg_print_omp_dynamic(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_global.g.g_dynamic);
4047 } // __kmp_stg_print_omp_dynamic
4048
__kmp_stg_parse_kmp_dynamic_mode(char const * name,char const * value,void * data)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, 0);
4054 return;
4055 }
4056 #ifdef USE_LOAD_BALANCE
4057 else if (__kmp_str_match("load balance", 2, value) ||
4058 __kmp_str_match("load_balance", 2, value) ||
4059 __kmp_str_match("load-balance", 2, value) ||
4060 __kmp_str_match("loadbalance", 2, value) ||
4061 __kmp_str_match("balance", 1, value)) {
4062 __kmp_global.g.g_dynamic_mode = dynamic_load_balance;
4063 }
4064 #endif /* USE_LOAD_BALANCE */
4065 else if (__kmp_str_match("thread limit", 1, value) ||
4066 __kmp_str_match("thread_limit", 1, value) ||
4067 __kmp_str_match("thread-limit", 1, value) ||
4068 __kmp_str_match("threadlimit", 1, value) ||
4069 __kmp_str_match("limit", 2, value)) {
4070 __kmp_global.g.g_dynamic_mode = dynamic_thread_limit;
4071 } else if (__kmp_str_match("random", 1, 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
__kmp_stg_print_kmp_dynamic_mode(kmp_str_buf_t * buffer,char const * name,void * data)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, "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, "thread limit");
4091 } else if (__kmp_global.g.g_dynamic_mode == dynamic_random) {
4092 __kmp_stg_print_str(buffer, name, "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
__kmp_stg_parse_ld_balance_interval(char const * name,char const * value,void * data)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(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
__kmp_stg_print_ld_balance_interval(kmp_str_buf_t * buffer,char const * name,void * data)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, " %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
__kmp_stg_parse_init_at_fork(char const * name,char const * value,void * data)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, &__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
__kmp_stg_print_init_at_fork(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_need_register_atfork_specified);
4138 } // __kmp_stg_print_init_at_fork
4139
4140 // -----------------------------------------------------------------------------
4141 // KMP_SCHEDULE
4142
__kmp_stg_parse_schedule(char const * name,char const * value,void * data)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(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(value, ';');
4158 if (*value && semicolon != value) {
4159 const char *comma = strchr(value, ',');
4160
4161 if (comma) {
4162 ++comma;
4163 sentinel = ',';
4164 } else
4165 sentinel = ';';
4166 if (!__kmp_strcasecmp_with_sentinel("static", value, sentinel)) {
4167 if (!__kmp_strcasecmp_with_sentinel("greedy", comma, ';')) {
4168 __kmp_static = kmp_sch_static_greedy;
4169 continue;
4170 } else if (!__kmp_strcasecmp_with_sentinel("balanced", comma,
4171 ';')) {
4172 __kmp_static = kmp_sch_static_balanced;
4173 continue;
4174 }
4175 } else if (!__kmp_strcasecmp_with_sentinel("guided", value,
4176 sentinel)) {
4177 if (!__kmp_strcasecmp_with_sentinel("iterative", comma, ';')) {
4178 __kmp_guided = kmp_sch_guided_iterative_chunked;
4179 continue;
4180 } else if (!__kmp_strcasecmp_with_sentinel("analytical", comma,
4181 ';')) {
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
__kmp_stg_print_schedule(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s='", name);
4202 }
4203 if (__kmp_static == kmp_sch_static_greedy) {
4204 __kmp_str_buf_print(buffer, "%s", "static,greedy");
4205 } else if (__kmp_static == kmp_sch_static_balanced) {
4206 __kmp_str_buf_print(buffer, "%s", "static,balanced");
4207 }
4208 if (__kmp_guided == kmp_sch_guided_iterative_chunked) {
4209 __kmp_str_buf_print(buffer, ";%s'\n", "guided,iterative");
4210 } else if (__kmp_guided == kmp_sch_guided_analytical_chunked) {
4211 __kmp_str_buf_print(buffer, ";%s'\n", "guided,analytical");
4212 }
4213 } // __kmp_stg_print_schedule
4214
4215 // -----------------------------------------------------------------------------
4216 // OMP_SCHEDULE
4217
__kmp_omp_schedule_restore()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]
__kmp_parse_single_omp_schedule(const char * name,const char * value,bool parse_hier=false)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("monotonic", ptr, *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("nonmonotonic", ptr, *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("dynamic", ptr, *delim))
4291 sched = kmp_sch_dynamic_chunked;
4292 else if (!__kmp_strcasecmp_with_sentinel("guided", ptr, *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("auto", ptr, *delim))
4296 sched = kmp_sch_auto;
4297 else if (!__kmp_strcasecmp_with_sentinel("trapezoidal", ptr, *delim))
4298 sched = kmp_sch_trapezoidal;
4299 else if (!__kmp_strcasecmp_with_sentinel("static", ptr, *delim))
4300 sched = kmp_sch_static;
4301 #if KMP_STATIC_STEAL_ENABLED
4302 else if (!__kmp_strcasecmp_with_sentinel("static_steal", ptr, *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(delim + 1, *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
__kmp_stg_parse_omp_schedule(char const * name,char const * value,void * data)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(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, 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
__kmp_stg_print_omp_schedule(kmp_str_buf_t * buffer,char const * name,void * data)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, " %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, "monotonic:");
4418 } else if (SCHEDULE_HAS_NONMONOTONIC(__kmp_sched)) {
4419 __kmp_str_buf_print(buffer, "nonmonotonic:");
4420 }
4421 if (__kmp_chunk) {
4422 switch (sched) {
4423 case kmp_sch_dynamic_chunked:
4424 __kmp_str_buf_print(buffer, "%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, "%s,%d'\n", "guided", __kmp_chunk);
4429 break;
4430 case kmp_sch_trapezoidal:
4431 __kmp_str_buf_print(buffer, "%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, "%s,%d'\n", "static", __kmp_chunk);
4438 break;
4439 case kmp_sch_static_steal:
4440 __kmp_str_buf_print(buffer, "%s,%d'\n", "static_steal", __kmp_chunk);
4441 break;
4442 case kmp_sch_auto:
4443 __kmp_str_buf_print(buffer, "%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, "%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, "%s'\n", "guided");
4458 break;
4459 case kmp_sch_trapezoidal:
4460 __kmp_str_buf_print(buffer, "%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, "%s'\n", "static");
4467 break;
4468 case kmp_sch_static_steal:
4469 __kmp_str_buf_print(buffer, "%s'\n", "static_steal");
4470 break;
4471 case kmp_sch_auto:
4472 __kmp_str_buf_print(buffer, "%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
__kmp_stg_parse_kmp_hand_thread(char const * name,char const * value,void * data)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
__kmp_stg_print_kmp_hand_thread(kmp_str_buf_t * buffer,char const * name,void * data)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
__kmp_stg_parse_kmp_force_monotonic(char const * name,char const * value,void * data)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, &(__kmp_force_monotonic));
4501 } // __kmp_stg_parse_kmp_force_monotonic
4502
__kmp_stg_print_kmp_force_monotonic(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_force_monotonic);
4506 } // __kmp_stg_print_kmp_force_monotonic
4507
4508 // -----------------------------------------------------------------------------
4509 // KMP_ATOMIC_MODE
4510
__kmp_stg_parse_atomic_mode(char const * name,char const * value,void * data)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, 0, max, &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
__kmp_stg_print_atomic_mode(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_atomic_mode);
4532 } // __kmp_stg_print_atomic_mode
4533
4534 // -----------------------------------------------------------------------------
4535 // KMP_CONSISTENCY_CHECK
4536
__kmp_stg_parse_consistency_check(char const * name,char const * value,void * data)4537 static void __kmp_stg_parse_consistency_check(char const *name,
4538 char const *value, void *data) {
4539 if (!__kmp_strcasecmp_with_sentinel("all", value, 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("none", value, 0)) {
4547 __kmp_env_consistency_check = FALSE;
4548 } else {
4549 KMP_WARNING(StgInvalidValue, name, value);
4550 }
4551 } // __kmp_stg_parse_consistency_check
4552
__kmp_stg_print_consistency_check(kmp_str_buf_t * buffer,char const * name,void * data)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
__kmp_stg_parse_itt_prepare_delay(char const * name,char const * value,void * data)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, 0, INT_MAX, &delay);
4582 __kmp_itt_prepare_delay = delay;
4583 } // __kmp_str_parse_itt_prepare_delay
4584
__kmp_stg_print_itt_prepare_delay(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_malloc_pool_incr(char const * name,char const * value,void * data)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, &__kmp_malloc_pool_incr,
4601 1);
4602 } // __kmp_stg_parse_malloc_pool_incr
4603
__kmp_stg_print_malloc_pool_incr(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_malloc_pool_incr);
4607
4608 } // _kmp_stg_print_malloc_pool_incr
4609
4610 #ifdef KMP_DEBUG
4611
4612 // -----------------------------------------------------------------------------
4613 // KMP_PAR_RANGE
4614
__kmp_stg_parse_par_range_env(char const * name,char const * value,void * data)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, &__kmp_par_range,
4618 __kmp_par_range_routine, __kmp_par_range_filename,
4619 &__kmp_par_range_lb, &__kmp_par_range_ub);
4620 } // __kmp_stg_parse_par_range_env
4621
__kmp_stg_print_par_range_env(kmp_str_buf_t * buffer,char const * name,void * data)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, par_range_to_print);
4626 }
4627 } // __kmp_stg_print_par_range_env
4628
4629 #endif
4630
4631 // -----------------------------------------------------------------------------
4632 // KMP_GTID_MODE
4633
__kmp_stg_parse_gtid_mode(char const * name,char const * value,void * data)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, 0, max, &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
__kmp_stg_print_gtid_mode(kmp_str_buf_t * buffer,char const * name,void * data)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, 0);
4662 } else {
4663 __kmp_stg_print_int(buffer, name, __kmp_gtid_mode);
4664 }
4665 } // __kmp_stg_print_gtid_mode
4666
4667 // -----------------------------------------------------------------------------
4668 // KMP_NUM_LOCKS_IN_BLOCK
4669
__kmp_stg_parse_lock_block(char const * name,char const * value,void * data)4670 static void __kmp_stg_parse_lock_block(char const *name, char const *value,
4671 void *data) {
4672 __kmp_stg_parse_int(name, value, 0, KMP_INT_MAX, &__kmp_num_locks_in_block);
4673 } // __kmp_str_parse_lock_block
4674
__kmp_stg_print_lock_block(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_lock_kind(char const * name,char const * value,void * data)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("tas", 2, value) ||
4697 __kmp_str_match("test and set", 2, value) ||
4698 __kmp_str_match("test_and_set", 2, value) ||
4699 __kmp_str_match("test-and-set", 2, value) ||
4700 __kmp_str_match("test andset", 2, value) ||
4701 __kmp_str_match("test_andset", 2, value) ||
4702 __kmp_str_match("test-andset", 2, value) ||
4703 __kmp_str_match("testand set", 2, value) ||
4704 __kmp_str_match("testand_set", 2, value) ||
4705 __kmp_str_match("testand-set", 2, value) ||
4706 __kmp_str_match("testandset", 2, 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("futex", 1, 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("ticket", 2, value)) {
4721 __kmp_user_lock_kind = lk_ticket;
4722 KMP_STORE_LOCK_SEQ(ticket);
4723 } else if (__kmp_str_match("queuing", 1, value) ||
4724 __kmp_str_match("queue", 1, value)) {
4725 __kmp_user_lock_kind = lk_queuing;
4726 KMP_STORE_LOCK_SEQ(queuing);
4727 } else if (__kmp_str_match("drdpa ticket", 1, value) ||
4728 __kmp_str_match("drdpa_ticket", 1, value) ||
4729 __kmp_str_match("drdpa-ticket", 1, value) ||
4730 __kmp_str_match("drdpaticket", 1, value) ||
4731 __kmp_str_match("drdpa", 1, 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("adaptive", 1, 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("rtm_queuing", 1, 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("rtm_spin", 1, 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("hle", 1, 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
__kmp_stg_print_lock_kind(kmp_str_buf_t * buffer,char const * name,void * data)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)
__kmp_stg_parse_spin_backoff_params(const char * name,const char * value,void * data)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(buf, *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
__kmp_stg_print_spin_backoff_params(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s='", name);
4920 }
4921 __kmp_str_buf_print(buffer, "%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]
__kmp_stg_parse_adaptive_lock_props(const char * name,const char * value,void * data)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(buf, *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
__kmp_stg_print_adaptive_lock_props(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s='", name);
5016 }
5017 __kmp_str_buf_print(buffer, "%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
__kmp_stg_parse_speculative_statsfile(char const * name,char const * value,void * data)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
__kmp_stg_print_speculative_statsfile(kmp_str_buf_t * buffer,char const * name,void * data)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.
__kmp_hw_subset_break_tie(const kmp_hw_t * possible,size_t num_possible)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.
__kmp_stg_parse_hw_subset_name(char const * 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(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, false);
5099 if (index < KMP_STRLEN(s)) {
5100 char c = (char)toupper(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
__kmp_stg_parse_hw_subset(char const * name,char const * value,void * data)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(name, "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(*pos));
5161 if (input[len] == 'X')
5162 input[len] = ','; // unify delimiters of levels
5163 if (input[len] == 'O' && strchr(digits, *(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(pos, ','))) {
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(pos, '&'))) {
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(core_components[j]);
5212 if (num <= 0) {
5213 goto err; // only positive integers are valid for count
5214 }
5215 pos = core_components[j] + strspn(core_components[j], 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(core_components[j], '@');
5225 attr_ptr = strchr(core_components[j], ':');
5226
5227 if (offset_ptr) {
5228 offset = atoi(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("intel_core", -1, attr_ptr + 1)) {
5236 attr.set_core_type(KMP_HW_CORE_TYPE_CORE);
5237 } else if (__kmp_str_match("intel_atom", -1, attr_ptr + 1)) {
5238 attr.set_core_type(KMP_HW_CORE_TYPE_ATOM);
5239 } else
5240 #endif
5241 if (__kmp_str_match("eff", 3, 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(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(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(__kmp_hw_subset);
5276 __kmp_hw_subset = nullptr;
5277 }
5278 return;
5279 }
5280
__kmp_stg_print_hw_subset(kmp_str_buf_t * buffer,char const * name,void * data)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, " %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(i);
5296 if (i > 0)
5297 __kmp_str_buf_print(&buf, "%c", ',');
5298 for (int j = 0; j < item.num_attrs; ++j) {
5299 __kmp_str_buf_print(&buf, "%s%d%s", (j > 0 ? "&" : ""), item.num[j],
5300 __kmp_hw_get_keyword(item.type));
5301 if (item.attr[j].is_core_type_valid())
5302 __kmp_str_buf_print(
5303 &buf, ":%s",
5304 __kmp_hw_get_core_type_keyword(item.attr[j].get_core_type()));
5305 if (item.attr[j].is_core_eff_valid())
5306 __kmp_str_buf_print(&buf, ":eff%d", item.attr[j].get_core_eff());
5307 if (item.offset[j])
5308 __kmp_str_buf_print(&buf, "@%d", item.offset[j]);
5309 }
5310 }
5311 __kmp_str_buf_print(buffer, "%s'\n", buf.str);
5312 __kmp_str_buf_free(&buf);
5313 }
5314
5315 #if USE_ITT_BUILD
5316 // -----------------------------------------------------------------------------
5317 // KMP_FORKJOIN_FRAMES
5318
__kmp_stg_parse_forkjoin_frames(char const * name,char const * value,void * data)5319 static void __kmp_stg_parse_forkjoin_frames(char const *name, char const *value,
5320 void *data) {
5321 __kmp_stg_parse_bool(name, value, &__kmp_forkjoin_frames);
5322 } // __kmp_stg_parse_forkjoin_frames
5323
__kmp_stg_print_forkjoin_frames(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_forkjoin_frames);
5327 } // __kmp_stg_print_forkjoin_frames
5328
5329 // -----------------------------------------------------------------------------
5330 // KMP_FORKJOIN_FRAMES_MODE
5331
__kmp_stg_parse_forkjoin_frames_mode(char const * name,char const * value,void * data)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, 0, 3, &__kmp_forkjoin_frames_mode);
5336 } // __kmp_stg_parse_forkjoin_frames
5337
__kmp_stg_print_forkjoin_frames_mode(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_forkjoin_frames_mode);
5341 } // __kmp_stg_print_forkjoin_frames
5342 #endif /* USE_ITT_BUILD */
5343
5344 // -----------------------------------------------------------------------------
5345 // KMP_ENABLE_TASK_THROTTLING
5346
__kmp_stg_parse_task_throttling(char const * name,char const * value,void * data)5347 static void __kmp_stg_parse_task_throttling(char const *name, char const *value,
5348 void *data) {
5349 __kmp_stg_parse_bool(name, value, &__kmp_enable_task_throttling);
5350 } // __kmp_stg_parse_task_throttling
5351
__kmp_stg_print_task_throttling(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_user_level_mwait(char const * name,char const * value,void * data)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, &__kmp_user_level_mwait);
5364 } // __kmp_stg_parse_user_level_mwait
5365
__kmp_stg_print_user_level_mwait(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_user_level_mwait);
5369 } // __kmp_stg_print_user_level_mwait
5370
5371 // -----------------------------------------------------------------------------
5372 // KMP_MWAIT_HINTS
5373
__kmp_stg_parse_mwait_hints(char const * name,char const * value,void * data)5374 static void __kmp_stg_parse_mwait_hints(char const *name, char const *value,
5375 void *data) {
5376 __kmp_stg_parse_int(name, value, 0, INT_MAX, &__kmp_mwait_hints);
5377 } // __kmp_stg_parse_mwait_hints
5378
__kmp_stg_print_mwait_hints(kmp_str_buf_t * buffer,char const * name,void * data)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, __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
__kmp_stg_parse_tpause(char const * name,char const * value,void * data)5391 static void __kmp_stg_parse_tpause(char const *name, char const *value,
5392 void *data) {
5393 __kmp_stg_parse_int(name, value, 0, INT_MAX, &__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
__kmp_stg_print_tpause(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_tpause_state);
5404 } // __kmp_stg_print_tpause
5405 #endif // KMP_HAVE_UMWAIT
5406
5407 // -----------------------------------------------------------------------------
5408 // OMP_DISPLAY_ENV
5409
__kmp_stg_parse_omp_display_env(char const * name,char const * value,void * data)5410 static void __kmp_stg_parse_omp_display_env(char const *name, char const *value,
5411 void *data) {
5412 if (__kmp_str_match("VERBOSE", 1, value)) {
5413 __kmp_display_env_verbose = TRUE;
5414 } else {
5415 __kmp_stg_parse_bool(name, value, &__kmp_display_env);
5416 }
5417 } // __kmp_stg_parse_omp_display_env
5418
__kmp_stg_print_omp_display_env(kmp_str_buf_t * buffer,char const * name,void * data)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, "VERBOSE");
5423 } else {
5424 __kmp_stg_print_bool(buffer, name, __kmp_display_env);
5425 }
5426 } // __kmp_stg_print_omp_display_env
5427
__kmp_stg_parse_omp_cancellation(char const * name,char const * value,void * data)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, &__kmp_omp_cancellation);
5435 } // __kmp_stg_parse_omp_cancellation
5436
__kmp_stg_print_omp_cancellation(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_omp_cancellation);
5440 } // __kmp_stg_print_omp_cancellation
5441
5442 #if OMPT_SUPPORT
5443 int __kmp_tool = 1;
5444
__kmp_stg_parse_omp_tool(char const * name,char const * value,void * data)5445 static void __kmp_stg_parse_omp_tool(char const *name, char const *value,
5446 void *data) {
5447 __kmp_stg_parse_bool(name, value, &__kmp_tool);
5448 } // __kmp_stg_parse_omp_tool
5449
__kmp_stg_print_omp_tool(kmp_str_buf_t * buffer,char const * name,void * data)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, " %s=%s\n", name,
5456 __kmp_tool ? "enabled" : "disabled");
5457 }
5458 } // __kmp_stg_print_omp_tool
5459
5460 char *__kmp_tool_libraries = NULL;
5461
__kmp_stg_parse_omp_tool_libraries(char const * name,char const * value,void * data)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, &__kmp_tool_libraries);
5465 } // __kmp_stg_parse_omp_tool_libraries
5466
__kmp_stg_print_omp_tool_libraries(kmp_str_buf_t * buffer,char const * name,void * data)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, __kmp_tool_libraries);
5471 else {
5472 if (__kmp_env_format) {
5473 KMP_STR_BUF_PRINT_NAME;
5474 } else {
5475 __kmp_str_buf_print(buffer, " %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
__kmp_stg_parse_omp_tool_verbose_init(char const * name,char const * value,void * data)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, &__kmp_tool_verbose_init);
5487 } // __kmp_stg_parse_omp_tool_libraries
5488
__kmp_stg_print_omp_tool_verbose_init(kmp_str_buf_t * buffer,char const * name,void * data)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, __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, " %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 {"KMP_ALL_THREADS", __kmp_stg_parse_device_thread_limit, NULL, NULL, 0, 0},
5511 {"KMP_BLOCKTIME", __kmp_stg_parse_blocktime, __kmp_stg_print_blocktime,
5512 NULL, 0, 0},
5513 {"KMP_USE_YIELD", __kmp_stg_parse_use_yield, __kmp_stg_print_use_yield,
5514 NULL, 0, 0},
5515 {"KMP_DUPLICATE_LIB_OK", __kmp_stg_parse_duplicate_lib_ok,
5516 __kmp_stg_print_duplicate_lib_ok, NULL, 0, 0},
5517 {"KMP_LIBRARY", __kmp_stg_parse_wait_policy, __kmp_stg_print_wait_policy,
5518 NULL, 0, 0},
5519 {"KMP_DEVICE_THREAD_LIMIT", __kmp_stg_parse_device_thread_limit,
5520 __kmp_stg_print_device_thread_limit, NULL, 0, 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 {"KMP_SETTINGS", __kmp_stg_parse_settings, __kmp_stg_print_settings, NULL,
5526 0, 0},
5527 {"KMP_STACKOFFSET", __kmp_stg_parse_stackoffset,
5528 __kmp_stg_print_stackoffset, NULL, 0, 0},
5529 {"KMP_STACKSIZE", __kmp_stg_parse_stacksize, __kmp_stg_print_stacksize,
5530 NULL, 0, 0},
5531 {"KMP_STACKPAD", __kmp_stg_parse_stackpad, __kmp_stg_print_stackpad, NULL,
5532 0, 0},
5533 {"KMP_VERSION", __kmp_stg_parse_version, __kmp_stg_print_version, NULL, 0,
5534 0},
5535 {"KMP_WARNINGS", __kmp_stg_parse_warnings, __kmp_stg_print_warnings, NULL,
5536 0, 0},
5537
5538 {"KMP_NESTING_MODE", __kmp_stg_parse_nesting_mode,
5539 __kmp_stg_print_nesting_mode, NULL, 0, 0},
5540 {"OMP_NESTED", __kmp_stg_parse_nested, __kmp_stg_print_nested, NULL, 0, 0},
5541 {"OMP_NUM_THREADS", __kmp_stg_parse_num_threads,
5542 __kmp_stg_print_num_threads, NULL, 0, 0},
5543 {"OMP_STACKSIZE", __kmp_stg_parse_stacksize, __kmp_stg_print_stacksize,
5544 NULL, 0, 0},
5545
5546 {"KMP_TASKING", __kmp_stg_parse_tasking, __kmp_stg_print_tasking, NULL, 0,
5547 0},
5548 {"KMP_TASK_STEALING_CONSTRAINT", __kmp_stg_parse_task_stealing,
5549 __kmp_stg_print_task_stealing, NULL, 0, 0},
5550 {"OMP_MAX_ACTIVE_LEVELS", __kmp_stg_parse_max_active_levels,
5551 __kmp_stg_print_max_active_levels, NULL, 0, 0},
5552 {"OMP_DEFAULT_DEVICE", __kmp_stg_parse_default_device,
5553 __kmp_stg_print_default_device, NULL, 0, 0},
5554 {"OMP_TARGET_OFFLOAD", __kmp_stg_parse_target_offload,
5555 __kmp_stg_print_target_offload, NULL, 0, 0},
5556 {"OMP_MAX_TASK_PRIORITY", __kmp_stg_parse_max_task_priority,
5557 __kmp_stg_print_max_task_priority, NULL, 0, 0},
5558 {"KMP_TASKLOOP_MIN_TASKS", __kmp_stg_parse_taskloop_min_tasks,
5559 __kmp_stg_print_taskloop_min_tasks, NULL, 0, 0},
5560 {"OMP_THREAD_LIMIT", __kmp_stg_parse_thread_limit,
5561 __kmp_stg_print_thread_limit, NULL, 0, 0},
5562 {"KMP_TEAMS_THREAD_LIMIT", __kmp_stg_parse_teams_thread_limit,
5563 __kmp_stg_print_teams_thread_limit, NULL, 0, 0},
5564 {"OMP_NUM_TEAMS", __kmp_stg_parse_nteams, __kmp_stg_print_nteams, NULL, 0,
5565 0},
5566 {"OMP_TEAMS_THREAD_LIMIT", __kmp_stg_parse_teams_th_limit,
5567 __kmp_stg_print_teams_th_limit, NULL, 0, 0},
5568 {"OMP_WAIT_POLICY", __kmp_stg_parse_wait_policy,
5569 __kmp_stg_print_wait_policy, NULL, 0, 0},
5570 {"KMP_DISP_NUM_BUFFERS", __kmp_stg_parse_disp_buffers,
5571 __kmp_stg_print_disp_buffers, NULL, 0, 0},
5572 #if KMP_NESTED_HOT_TEAMS
5573 {"KMP_HOT_TEAMS_MAX_LEVEL", __kmp_stg_parse_hot_teams_level,
5574 __kmp_stg_print_hot_teams_level, NULL, 0, 0},
5575 {"KMP_HOT_TEAMS_MODE", __kmp_stg_parse_hot_teams_mode,
5576 __kmp_stg_print_hot_teams_mode, NULL, 0, 0},
5577 #endif // KMP_NESTED_HOT_TEAMS
5578
5579 #if KMP_HANDLE_SIGNALS
5580 {"KMP_HANDLE_SIGNALS", __kmp_stg_parse_handle_signals,
5581 __kmp_stg_print_handle_signals, NULL, 0, 0},
5582 #endif
5583
5584 #if KMP_ARCH_X86 || KMP_ARCH_X86_64
5585 {"KMP_INHERIT_FP_CONTROL", __kmp_stg_parse_inherit_fp_control,
5586 __kmp_stg_print_inherit_fp_control, NULL, 0, 0},
5587 #endif /* KMP_ARCH_X86 || KMP_ARCH_X86_64 */
5588
5589 #ifdef KMP_GOMP_COMPAT
5590 {"GOMP_STACKSIZE", __kmp_stg_parse_stacksize, NULL, NULL, 0, 0},
5591 #endif
5592
5593 #ifdef KMP_DEBUG
5594 {"KMP_A_DEBUG", __kmp_stg_parse_a_debug, __kmp_stg_print_a_debug, NULL, 0,
5595 0},
5596 {"KMP_B_DEBUG", __kmp_stg_parse_b_debug, __kmp_stg_print_b_debug, NULL, 0,
5597 0},
5598 {"KMP_C_DEBUG", __kmp_stg_parse_c_debug, __kmp_stg_print_c_debug, NULL, 0,
5599 0},
5600 {"KMP_D_DEBUG", __kmp_stg_parse_d_debug, __kmp_stg_print_d_debug, NULL, 0,
5601 0},
5602 {"KMP_E_DEBUG", __kmp_stg_parse_e_debug, __kmp_stg_print_e_debug, NULL, 0,
5603 0},
5604 {"KMP_F_DEBUG", __kmp_stg_parse_f_debug, __kmp_stg_print_f_debug, NULL, 0,
5605 0},
5606 {"KMP_DEBUG", __kmp_stg_parse_debug, NULL, /* no print */ NULL, 0, 0},
5607 {"KMP_DEBUG_BUF", __kmp_stg_parse_debug_buf, __kmp_stg_print_debug_buf,
5608 NULL, 0, 0},
5609 {"KMP_DEBUG_BUF_ATOMIC", __kmp_stg_parse_debug_buf_atomic,
5610 __kmp_stg_print_debug_buf_atomic, NULL, 0, 0},
5611 {"KMP_DEBUG_BUF_CHARS", __kmp_stg_parse_debug_buf_chars,
5612 __kmp_stg_print_debug_buf_chars, NULL, 0, 0},
5613 {"KMP_DEBUG_BUF_LINES", __kmp_stg_parse_debug_buf_lines,
5614 __kmp_stg_print_debug_buf_lines, NULL, 0, 0},
5615 {"KMP_DIAG", __kmp_stg_parse_diag, __kmp_stg_print_diag, NULL, 0, 0},
5616
5617 {"KMP_PAR_RANGE", __kmp_stg_parse_par_range_env,
5618 __kmp_stg_print_par_range_env, NULL, 0, 0},
5619 #endif // KMP_DEBUG
5620
5621 {"KMP_ALIGN_ALLOC", __kmp_stg_parse_align_alloc,
5622 __kmp_stg_print_align_alloc, NULL, 0, 0},
5623
5624 {"KMP_PLAIN_BARRIER", __kmp_stg_parse_barrier_branch_bit,
5625 __kmp_stg_print_barrier_branch_bit, NULL, 0, 0},
5626 {"KMP_PLAIN_BARRIER_PATTERN", __kmp_stg_parse_barrier_pattern,
5627 __kmp_stg_print_barrier_pattern, NULL, 0, 0},
5628 {"KMP_FORKJOIN_BARRIER", __kmp_stg_parse_barrier_branch_bit,
5629 __kmp_stg_print_barrier_branch_bit, NULL, 0, 0},
5630 {"KMP_FORKJOIN_BARRIER_PATTERN", __kmp_stg_parse_barrier_pattern,
5631 __kmp_stg_print_barrier_pattern, NULL, 0, 0},
5632 #if KMP_FAST_REDUCTION_BARRIER
5633 {"KMP_REDUCTION_BARRIER", __kmp_stg_parse_barrier_branch_bit,
5634 __kmp_stg_print_barrier_branch_bit, NULL, 0, 0},
5635 {"KMP_REDUCTION_BARRIER_PATTERN", __kmp_stg_parse_barrier_pattern,
5636 __kmp_stg_print_barrier_pattern, NULL, 0, 0},
5637 #endif
5638
5639 {"KMP_ABORT_DELAY", __kmp_stg_parse_abort_delay,
5640 __kmp_stg_print_abort_delay, NULL, 0, 0},
5641 {"KMP_CPUINFO_FILE", __kmp_stg_parse_cpuinfo_file,
5642 __kmp_stg_print_cpuinfo_file, NULL, 0, 0},
5643 {"KMP_FORCE_REDUCTION", __kmp_stg_parse_force_reduction,
5644 __kmp_stg_print_force_reduction, NULL, 0, 0},
5645 {"KMP_DETERMINISTIC_REDUCTION", __kmp_stg_parse_force_reduction,
5646 __kmp_stg_print_force_reduction, NULL, 0, 0},
5647 {"KMP_STORAGE_MAP", __kmp_stg_parse_storage_map,
5648 __kmp_stg_print_storage_map, NULL, 0, 0},
5649 {"KMP_ALL_THREADPRIVATE", __kmp_stg_parse_all_threadprivate,
5650 __kmp_stg_print_all_threadprivate, NULL, 0, 0},
5651 {"KMP_FOREIGN_THREADS_THREADPRIVATE",
5652 __kmp_stg_parse_foreign_threads_threadprivate,
5653 __kmp_stg_print_foreign_threads_threadprivate, NULL, 0, 0},
5654
5655 #if KMP_AFFINITY_SUPPORTED
5656 {"KMP_AFFINITY", __kmp_stg_parse_affinity, __kmp_stg_print_affinity, NULL,
5657 0, 0},
5658 {"KMP_HIDDEN_HELPER_AFFINITY", __kmp_stg_parse_hh_affinity,
5659 __kmp_stg_print_hh_affinity, NULL, 0, 0},
5660 #ifdef KMP_GOMP_COMPAT
5661 {"GOMP_CPU_AFFINITY", __kmp_stg_parse_gomp_cpu_affinity, NULL,
5662 /* no print */ NULL, 0, 0},
5663 #endif /* KMP_GOMP_COMPAT */
5664 {"OMP_PROC_BIND", __kmp_stg_parse_proc_bind, __kmp_stg_print_proc_bind,
5665 NULL, 0, 0},
5666 {"KMP_TEAMS_PROC_BIND", __kmp_stg_parse_teams_proc_bind,
5667 __kmp_stg_print_teams_proc_bind, NULL, 0, 0},
5668 {"OMP_PLACES", __kmp_stg_parse_places, __kmp_stg_print_places, NULL, 0, 0},
5669 {"KMP_TOPOLOGY_METHOD", __kmp_stg_parse_topology_method,
5670 __kmp_stg_print_topology_method, NULL, 0, 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 {"OMP_DISPLAY_AFFINITY", __kmp_stg_parse_display_affinity,
5681 __kmp_stg_print_display_affinity, NULL, 0, 0},
5682 {"OMP_AFFINITY_FORMAT", __kmp_stg_parse_affinity_format,
5683 __kmp_stg_print_affinity_format, NULL, 0, 0},
5684 {"KMP_INIT_AT_FORK", __kmp_stg_parse_init_at_fork,
5685 __kmp_stg_print_init_at_fork, NULL, 0, 0},
5686 {"KMP_SCHEDULE", __kmp_stg_parse_schedule, __kmp_stg_print_schedule, NULL,
5687 0, 0},
5688 {"OMP_SCHEDULE", __kmp_stg_parse_omp_schedule, __kmp_stg_print_omp_schedule,
5689 NULL, 0, 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 {"KMP_FORCE_MONOTONIC_DYNAMIC_SCHEDULE",
5695 __kmp_stg_parse_kmp_force_monotonic, __kmp_stg_print_kmp_force_monotonic,
5696 NULL, 0, 0},
5697 {"KMP_ATOMIC_MODE", __kmp_stg_parse_atomic_mode,
5698 __kmp_stg_print_atomic_mode, NULL, 0, 0},
5699 {"KMP_CONSISTENCY_CHECK", __kmp_stg_parse_consistency_check,
5700 __kmp_stg_print_consistency_check, NULL, 0, 0},
5701
5702 #if USE_ITT_BUILD && USE_ITT_NOTIFY
5703 {"KMP_ITT_PREPARE_DELAY", __kmp_stg_parse_itt_prepare_delay,
5704 __kmp_stg_print_itt_prepare_delay, NULL, 0, 0},
5705 #endif /* USE_ITT_BUILD && USE_ITT_NOTIFY */
5706 {"KMP_MALLOC_POOL_INCR", __kmp_stg_parse_malloc_pool_incr,
5707 __kmp_stg_print_malloc_pool_incr, NULL, 0, 0},
5708 {"KMP_GTID_MODE", __kmp_stg_parse_gtid_mode, __kmp_stg_print_gtid_mode,
5709 NULL, 0, 0},
5710 {"OMP_DYNAMIC", __kmp_stg_parse_omp_dynamic, __kmp_stg_print_omp_dynamic,
5711 NULL, 0, 0},
5712 {"KMP_DYNAMIC_MODE", __kmp_stg_parse_kmp_dynamic_mode,
5713 __kmp_stg_print_kmp_dynamic_mode, NULL, 0, 0},
5714
5715 #ifdef USE_LOAD_BALANCE
5716 {"KMP_LOAD_BALANCE_INTERVAL", __kmp_stg_parse_ld_balance_interval,
5717 __kmp_stg_print_ld_balance_interval, NULL, 0, 0},
5718 #endif
5719
5720 {"KMP_NUM_LOCKS_IN_BLOCK", __kmp_stg_parse_lock_block,
5721 __kmp_stg_print_lock_block, NULL, 0, 0},
5722 {"KMP_LOCK_KIND", __kmp_stg_parse_lock_kind, __kmp_stg_print_lock_kind,
5723 NULL, 0, 0},
5724 {"KMP_SPIN_BACKOFF_PARAMS", __kmp_stg_parse_spin_backoff_params,
5725 __kmp_stg_print_spin_backoff_params, NULL, 0, 0},
5726 #if KMP_USE_ADAPTIVE_LOCKS
5727 {"KMP_ADAPTIVE_LOCK_PROPS", __kmp_stg_parse_adaptive_lock_props,
5728 __kmp_stg_print_adaptive_lock_props, NULL, 0, 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 {"KMP_PLACE_THREADS", __kmp_stg_parse_hw_subset, __kmp_stg_print_hw_subset,
5735 NULL, 0, 0},
5736 {"KMP_HW_SUBSET", __kmp_stg_parse_hw_subset, __kmp_stg_print_hw_subset,
5737 NULL, 0, 0},
5738 #if USE_ITT_BUILD
5739 {"KMP_FORKJOIN_FRAMES", __kmp_stg_parse_forkjoin_frames,
5740 __kmp_stg_print_forkjoin_frames, NULL, 0, 0},
5741 {"KMP_FORKJOIN_FRAMES_MODE", __kmp_stg_parse_forkjoin_frames_mode,
5742 __kmp_stg_print_forkjoin_frames_mode, NULL, 0, 0},
5743 #endif
5744 {"KMP_ENABLE_TASK_THROTTLING", __kmp_stg_parse_task_throttling,
5745 __kmp_stg_print_task_throttling, NULL, 0, 0},
5746
5747 {"OMP_DISPLAY_ENV", __kmp_stg_parse_omp_display_env,
5748 __kmp_stg_print_omp_display_env, NULL, 0, 0},
5749 {"OMP_CANCELLATION", __kmp_stg_parse_omp_cancellation,
5750 __kmp_stg_print_omp_cancellation, NULL, 0, 0},
5751 {"OMP_ALLOCATOR", __kmp_stg_parse_allocator, __kmp_stg_print_allocator,
5752 NULL, 0, 0},
5753 {"LIBOMP_USE_HIDDEN_HELPER_TASK", __kmp_stg_parse_use_hidden_helper,
5754 __kmp_stg_print_use_hidden_helper, NULL, 0, 0},
5755 {"LIBOMP_NUM_HIDDEN_HELPER_THREADS",
5756 __kmp_stg_parse_num_hidden_helper_threads,
5757 __kmp_stg_print_num_hidden_helper_threads, NULL, 0, 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 {"OMP_TOOL", __kmp_stg_parse_omp_tool, __kmp_stg_print_omp_tool, NULL, 0,
5766 0},
5767 {"OMP_TOOL_LIBRARIES", __kmp_stg_parse_omp_tool_libraries,
5768 __kmp_stg_print_omp_tool_libraries, NULL, 0, 0},
5769 {"OMP_TOOL_VERBOSE_INIT", __kmp_stg_parse_omp_tool_verbose_init,
5770 __kmp_stg_print_omp_tool_verbose_init, NULL, 0, 0},
5771 #endif
5772
5773 #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT
5774 {"KMP_USER_LEVEL_MWAIT", __kmp_stg_parse_user_level_mwait,
5775 __kmp_stg_print_user_level_mwait, NULL, 0, 0},
5776 {"KMP_MWAIT_HINTS", __kmp_stg_parse_mwait_hints,
5777 __kmp_stg_print_mwait_hints, NULL, 0, 0},
5778 #endif
5779
5780 #if KMP_HAVE_UMWAIT
5781 {"KMP_TPAUSE", __kmp_stg_parse_tpause, __kmp_stg_print_tpause, NULL, 0, 0},
5782 #endif
5783 {"", NULL, NULL, NULL, 0, 0}}; // settings
5784
5785 static int const __kmp_stg_count =
5786 sizeof(__kmp_stg_table) / sizeof(kmp_setting_t);
5787
__kmp_stg_find(char const * name)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(__kmp_stg_table[i].name, name) == 0) {
5794 return &__kmp_stg_table[i];
5795 }
5796 }
5797 }
5798 return NULL;
5799
5800 } // __kmp_stg_find
5801
__kmp_stg_cmp(void const * _a,void const * _b)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(a->name, "KMP_AFFINITY") == 0) {
5809 if (strcmp(b->name, "KMP_AFFINITY") == 0) {
5810 return 0;
5811 }
5812 return 1;
5813 } else if (strcmp(b->name, "KMP_AFFINITY") == 0) {
5814 return -1;
5815 }
5816 return strcmp(a->name, b->name);
5817 } // __kmp_stg_cmp
5818
__kmp_stg_init(void)5819 static void __kmp_stg_init(void) {
5820
5821 static int initialized = 0;
5822
5823 if (!initialized) {
5824
5825 // Sort table.
5826 qsort(__kmp_stg_table, __kmp_stg_count - 1, sizeof(kmp_setting_t),
5827 __kmp_stg_cmp);
5828
5829 { // Initialize *_STACKSIZE data.
5830 kmp_setting_t *kmp_stacksize =
5831 __kmp_stg_find("KMP_STACKSIZE"); // 1st priority.
5832 #ifdef KMP_GOMP_COMPAT
5833 kmp_setting_t *gomp_stacksize =
5834 __kmp_stg_find("GOMP_STACKSIZE"); // 2nd priority.
5835 #endif
5836 kmp_setting_t *omp_stacksize =
5837 __kmp_stg_find("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 = {1, CCAST(kmp_setting_t **, rivals)};
5845 #ifdef KMP_GOMP_COMPAT
5846 static kmp_stg_ss_data_t gomp_data = {1024,
5847 CCAST(kmp_setting_t **, rivals)};
5848 #endif
5849 static kmp_stg_ss_data_t omp_data = {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("KMP_LIBRARY"); // 1st priority.
5874 kmp_setting_t *omp_wait_policy =
5875 __kmp_stg_find("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 = {0, CCAST(kmp_setting_t **, rivals)};
5880 static kmp_stg_wp_data_t omp_data = {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("KMP_DEVICE_THREAD_LIMIT"); // 1st priority.
5898 kmp_setting_t *kmp_all_threads =
5899 __kmp_stg_find("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("KMP_HW_SUBSET");
5916 // 2nd priority
5917 kmp_setting_t *kmp_place_threads = __kmp_stg_find("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("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("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("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("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("KMP_FORCE_REDUCTION"); // 1st priority.
5984 kmp_setting_t *kmp_determ_red =
5985 __kmp_stg_find("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 = {1,
5990 CCAST(kmp_setting_t **, rivals)};
5991 static kmp_stg_fr_data_t determ_data = {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
__kmp_stg_parse(char const * name,char const * value)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
__kmp_stg_check_rivals(char const * name,char const * value,kmp_setting_t ** rivals)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(rivals[i]->name, 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
__kmp_env_toPrint(char const * name,int flag)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
__kmp_print_affinity_settings(const kmp_affinity_t * affinity)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
__kmp_aux_env_initialize(kmp_env_blk_t * block)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, "OMP_NUM_THREADS");
6104 if (value) {
6105 ompc_set_num_threads(__kmp_dflt_team_nth);
6106 }
6107
6108 /* KMP_BLOCKTIME */
6109 value = __kmp_env_blk_var(block, "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(__kmp_dflt_blocktime, thread, tid);
6118 }
6119
6120 /* OMP_NESTED */
6121 value = __kmp_env_blk_var(block, "OMP_NESTED");
6122 if (value) {
6123 ompc_set_nested(__kmp_dflt_max_active_levels > 1);
6124 }
6125
6126 /* OMP_DYNAMIC */
6127 value = __kmp_env_blk_var(block, "OMP_DYNAMIC");
6128 if (value) {
6129 ompc_set_dynamic(__kmp_global.g.g_dynamic);
6130 }
6131 }
6132
__kmp_env_initialize(char const * string)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(__kmp_dflt_team_nth_ub);
6145 }
6146 __kmp_env_blk_init(&block, 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(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, "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, 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, "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(aff_str, "none") == NULL) &&
6199 (FIND(aff_str, "physical") == NULL) &&
6200 (FIND(aff_str, "logical") == NULL) &&
6201 (FIND(aff_str, "compact") == NULL) &&
6202 (FIND(aff_str, "scatter") == NULL) &&
6203 (FIND(aff_str, "explicit") == NULL) &&
6204 (FIND(aff_str, "balanced") == NULL) &&
6205 (FIND(aff_str, "disabled") == NULL)) {
6206 __kmp_affinity_notype = __kmp_stg_find("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, "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(&m.str);
6259
6260 // Now process all of the settings.
6261 for (i = 0; i < block.count; ++i) {
6262 __kmp_stg_parse(block.vars[i].name, 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(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_nested_proc_bind.bind_types[0] == proc_bind_false)
6424 __kmp_affinity.type = affinity_none;
6425 if (__kmp_affinity.type == affinity_default) {
6426 __kmp_affinity.type = affinity_compact;
6427 __kmp_affinity.flags.dups = FALSE;
6428 }
6429 } else if (__kmp_affinity.type == affinity_default) {
6430 #if KMP_MIC_SUPPORTED
6431 if (__kmp_mic_type != non_mic) {
6432 __kmp_nested_proc_bind.bind_types[0] = proc_bind_intel;
6433 } else
6434 #endif
6435 {
6436 __kmp_nested_proc_bind.bind_types[0] = proc_bind_false;
6437 }
6438 #if KMP_MIC_SUPPORTED
6439 if (__kmp_mic_type != non_mic) {
6440 __kmp_affinity.type = affinity_scatter;
6441 } else
6442 #endif
6443 {
6444 __kmp_affinity.type = affinity_none;
6445 }
6446 }
6447 if (__kmp_hh_affinity.type == affinity_default)
6448 __kmp_hh_affinity.type = affinity_none;
6449 if ((__kmp_affinity.gran == KMP_HW_UNKNOWN) &&
6450 (__kmp_affinity.gran_levels < 0)) {
6451 #if KMP_MIC_SUPPORTED
6452 if (__kmp_mic_type != non_mic) {
6453 __kmp_affinity.gran = KMP_HW_THREAD;
6454 } else
6455 #endif
6456 {
6457 __kmp_affinity.gran = KMP_HW_CORE;
6458 }
6459 }
6460 if ((__kmp_hh_affinity.gran == KMP_HW_UNKNOWN) &&
6461 (__kmp_hh_affinity.gran_levels < 0)) {
6462 #if KMP_MIC_SUPPORTED
6463 if (__kmp_mic_type != non_mic) {
6464 __kmp_hh_affinity.gran = KMP_HW_THREAD;
6465 } else
6466 #endif
6467 {
6468 __kmp_hh_affinity.gran = KMP_HW_CORE;
6469 }
6470 }
6471 if (__kmp_affinity_top_method == affinity_top_method_default) {
6472 __kmp_affinity_top_method = affinity_top_method_all;
6473 }
6474 }
6475 } else {
6476 // If affinity is disabled, then still need to assign topology method
6477 // to attempt machine detection and affinity types
6478 if (__kmp_affinity_top_method == affinity_top_method_default)
6479 __kmp_affinity_top_method = affinity_top_method_all;
6480 if (__kmp_affinity.type == affinity_default)
6481 __kmp_affinity.type = affinity_disabled;
6482 if (__kmp_hh_affinity.type == affinity_default)
6483 __kmp_hh_affinity.type = affinity_disabled;
6484 }
6485
6486 #ifdef KMP_DEBUG
6487 for (const kmp_affinity_t *affinity : __kmp_affinities)
6488 __kmp_print_affinity_settings(affinity);
6489 KMP_DEBUG_ASSERT(__kmp_nested_proc_bind.bind_types[0] != proc_bind_default);
6490 K_DIAG(1, ("__kmp_nested_proc_bind.bind_types[0] == %d\n",
6491 __kmp_nested_proc_bind.bind_types[0]));
6492 #endif
6493 }
6494
6495 #endif /* KMP_AFFINITY_SUPPORTED */
6496
6497 // Post-initialization step: some env. vars need their value's further
6498 // processing
6499 if (string != NULL) { // kmp_set_defaults() was called
6500 __kmp_aux_env_initialize(&block);
6501 }
6502
6503 __kmp_env_blk_free(&block);
6504
6505 KMP_MB();
6506
6507 } // __kmp_env_initialize
6508
__kmp_env_print()6509 void __kmp_env_print() {
6510
6511 kmp_env_blk_t block;
6512 int i;
6513 kmp_str_buf_t buffer;
6514
6515 __kmp_stg_init();
6516 __kmp_str_buf_init(&buffer);
6517
6518 __kmp_env_blk_init(&block, NULL);
6519 __kmp_env_blk_sort(&block);
6520
6521 // Print real environment values.
6522 __kmp_str_buf_print(&buffer, "\n%s\n\n", KMP_I18N_STR(UserSettings));
6523 for (i = 0; i < block.count; ++i) {
6524 char const *name = block.vars[i].name;
6525 char const *value = block.vars[i].value;
6526 if ((KMP_STRLEN(name) > 4 && strncmp(name, "KMP_", 4) == 0) ||
6527 strncmp(name, "OMP_", 4) == 0
6528 #ifdef KMP_GOMP_COMPAT
6529 || strncmp(name, "GOMP_", 5) == 0
6530 #endif // KMP_GOMP_COMPAT
6531 ) {
6532 __kmp_str_buf_print(&buffer, " %s=%s\n", name, value);
6533 }
6534 }
6535 __kmp_str_buf_print(&buffer, "\n");
6536
6537 // Print internal (effective) settings.
6538 __kmp_str_buf_print(&buffer, "%s\n\n", KMP_I18N_STR(EffectiveSettings));
6539 for (int i = 0; i < __kmp_stg_count; ++i) {
6540 if (__kmp_stg_table[i].print != NULL) {
6541 __kmp_stg_table[i].print(&buffer, __kmp_stg_table[i].name,
6542 __kmp_stg_table[i].data);
6543 }
6544 }
6545
6546 __kmp_printf("%s", buffer.str);
6547
6548 __kmp_env_blk_free(&block);
6549 __kmp_str_buf_free(&buffer);
6550
6551 __kmp_printf("\n");
6552
6553 } // __kmp_env_print
6554
__kmp_env_print_2()6555 void __kmp_env_print_2() {
6556 __kmp_display_env_impl(__kmp_display_env, __kmp_display_env_verbose);
6557 } // __kmp_env_print_2
6558
__kmp_display_env_impl(int display_env,int display_env_verbose)6559 void __kmp_display_env_impl(int display_env, int display_env_verbose) {
6560 kmp_env_blk_t block;
6561 kmp_str_buf_t buffer;
6562
6563 __kmp_env_format = 1;
6564
6565 __kmp_stg_init();
6566 __kmp_str_buf_init(&buffer);
6567
6568 __kmp_env_blk_init(&block, NULL);
6569 __kmp_env_blk_sort(&block);
6570
6571 __kmp_str_buf_print(&buffer, "\n%s\n", KMP_I18N_STR(DisplayEnvBegin));
6572 __kmp_str_buf_print(&buffer, " _OPENMP='%d'\n", __kmp_openmp_version);
6573
6574 for (int i = 0; i < __kmp_stg_count; ++i) {
6575 if (__kmp_stg_table[i].print != NULL &&
6576 ((display_env && strncmp(__kmp_stg_table[i].name, "OMP_", 4) == 0) ||
6577 display_env_verbose)) {
6578 __kmp_stg_table[i].print(&buffer, __kmp_stg_table[i].name,
6579 __kmp_stg_table[i].data);
6580 }
6581 }
6582
6583 __kmp_str_buf_print(&buffer, "%s\n", KMP_I18N_STR(DisplayEnvEnd));
6584 __kmp_str_buf_print(&buffer, "\n");
6585
6586 __kmp_printf("%s", buffer.str);
6587
6588 __kmp_env_blk_free(&block);
6589 __kmp_str_buf_free(&buffer);
6590
6591 __kmp_printf("\n");
6592 }
6593
6594 #if OMPD_SUPPORT
6595 // Dump environment variables for OMPD
__kmp_env_dump()6596 void __kmp_env_dump() {
6597
6598 kmp_env_blk_t block;
6599 kmp_str_buf_t buffer, env, notdefined;
6600
6601 __kmp_stg_init();
6602 __kmp_str_buf_init(&buffer);
6603 __kmp_str_buf_init(&env);
6604 __kmp_str_buf_init(¬defined);
6605
6606 __kmp_env_blk_init(&block, NULL);
6607 __kmp_env_blk_sort(&block);
6608
6609 __kmp_str_buf_print(¬defined, ": %s", KMP_I18N_STR(NotDefined));
6610
6611 for (int i = 0; i < __kmp_stg_count; ++i) {
6612 if (__kmp_stg_table[i].print == NULL)
6613 continue;
6614 __kmp_str_buf_clear(&env);
6615 __kmp_stg_table[i].print(&env, __kmp_stg_table[i].name,
6616 __kmp_stg_table[i].data);
6617 if (env.used < 4) // valid definition must have indents (3) and a new line
6618 continue;
6619 if (strstr(env.str, notdefined.str))
6620 // normalize the string
6621 __kmp_str_buf_print(&buffer, "%s=undefined\n", __kmp_stg_table[i].name);
6622 else
6623 __kmp_str_buf_cat(&buffer, env.str + 3, env.used - 3);
6624 }
6625
6626 ompd_env_block = (char *)__kmp_allocate(buffer.used + 1);
6627 KMP_MEMCPY(ompd_env_block, buffer.str, buffer.used + 1);
6628 ompd_env_block_size = (ompd_size_t)KMP_STRLEN(ompd_env_block);
6629
6630 __kmp_env_blk_free(&block);
6631 __kmp_str_buf_free(&buffer);
6632 __kmp_str_buf_free(&env);
6633 __kmp_str_buf_free(¬defined);
6634 }
6635 #endif // OMPD_SUPPORT
6636
6637 // end of file
6638