URL
https://opencores.org/ocsvn/openrisc_me/openrisc_me/trunk
Subversion Repositories openrisc_me
[/] [openrisc/] [trunk/] [gnu-src/] [gcc-4.5.1/] [libgomp/] [env.c] - Rev 281
Go to most recent revision | Compare with Previous | Blame | View Log
/* Copyright (C) 2005, 2006, 2007, 2008, 2009, 2010 Free Software Foundation, Inc. Contributed by Richard Henderson <rth@redhat.com>. This file is part of the GNU OpenMP Library (libgomp). Libgomp is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 3, or (at your option) any later version. Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. Under Section 7 of GPL version 3, you are granted additional permissions described in the GCC Runtime Library Exception, version 3.1, as published by the Free Software Foundation. You should have received a copy of the GNU General Public License and a copy of the GCC Runtime Library Exception along with this program; see the files COPYING3 and COPYING.RUNTIME respectively. If not, see <http://www.gnu.org/licenses/>. */ /* This file defines the OpenMP internal control variables, and arranges for them to be initialized from environment variables at startup. */ #include "libgomp.h" #include "libgomp_f.h" #include <ctype.h> #include <stdlib.h> #ifdef STRING_WITH_STRINGS # include <string.h> # include <strings.h> #else # ifdef HAVE_STRING_H # include <string.h> # else # ifdef HAVE_STRINGS_H # include <strings.h> # endif # endif #endif #include <limits.h> #include <errno.h> #ifndef HAVE_STRTOULL # define strtoull(ptr, eptr, base) strtoul (ptr, eptr, base) #endif struct gomp_task_icv gomp_global_icv = { .nthreads_var = 1, .run_sched_var = GFS_DYNAMIC, .run_sched_modifier = 1, .dyn_var = false, .nest_var = false }; unsigned short *gomp_cpu_affinity; size_t gomp_cpu_affinity_len; unsigned long gomp_max_active_levels_var = INT_MAX; unsigned long gomp_thread_limit_var = ULONG_MAX; unsigned long gomp_remaining_threads_count; #ifndef HAVE_SYNC_BUILTINS gomp_mutex_t gomp_remaining_threads_lock; #endif unsigned long gomp_available_cpus = 1, gomp_managed_threads = 1; unsigned long long gomp_spin_count_var, gomp_throttled_spin_count_var; /* Parse the OMP_SCHEDULE environment variable. */ static void parse_schedule (void) { char *env, *end; unsigned long value; env = getenv ("OMP_SCHEDULE"); if (env == NULL) return; while (isspace ((unsigned char) *env)) ++env; if (strncasecmp (env, "static", 6) == 0) { gomp_global_icv.run_sched_var = GFS_STATIC; env += 6; } else if (strncasecmp (env, "dynamic", 7) == 0) { gomp_global_icv.run_sched_var = GFS_DYNAMIC; env += 7; } else if (strncasecmp (env, "guided", 6) == 0) { gomp_global_icv.run_sched_var = GFS_GUIDED; env += 6; } else if (strncasecmp (env, "auto", 4) == 0) { gomp_global_icv.run_sched_var = GFS_AUTO; env += 4; } else goto unknown; while (isspace ((unsigned char) *env)) ++env; if (*env == '\0') return; if (*env++ != ',') goto unknown; while (isspace ((unsigned char) *env)) ++env; if (*env == '\0') goto invalid; errno = 0; value = strtoul (env, &end, 10); if (errno) goto invalid; while (isspace ((unsigned char) *end)) ++end; if (*end != '\0') goto invalid; if ((int)value != value) goto invalid; gomp_global_icv.run_sched_modifier = value; return; unknown: gomp_error ("Unknown value for environment variable OMP_SCHEDULE"); return; invalid: gomp_error ("Invalid value for chunk size in " "environment variable OMP_SCHEDULE"); return; } /* Parse an unsigned long environment variable. Return true if one was present and it was successfully parsed. */ static bool parse_unsigned_long (const char *name, unsigned long *pvalue, bool allow_zero) { char *env, *end; unsigned long value; env = getenv (name); if (env == NULL) return false; while (isspace ((unsigned char) *env)) ++env; if (*env == '\0') goto invalid; errno = 0; value = strtoul (env, &end, 10); if (errno || (long) value <= 0 - allow_zero) goto invalid; while (isspace ((unsigned char) *end)) ++end; if (*end != '\0') goto invalid; *pvalue = value; return true; invalid: gomp_error ("Invalid value for environment variable %s", name); return false; } /* Parse the OMP_STACKSIZE environment varible. Return true if one was present and it was successfully parsed. */ static bool parse_stacksize (const char *name, unsigned long *pvalue) { char *env, *end; unsigned long value, shift = 10; env = getenv (name); if (env == NULL) return false; while (isspace ((unsigned char) *env)) ++env; if (*env == '\0') goto invalid; errno = 0; value = strtoul (env, &end, 10); if (errno) goto invalid; while (isspace ((unsigned char) *end)) ++end; if (*end != '\0') { switch (tolower ((unsigned char) *end)) { case 'b': shift = 0; break; case 'k': break; case 'm': shift = 20; break; case 'g': shift = 30; break; default: goto invalid; } ++end; while (isspace ((unsigned char) *end)) ++end; if (*end != '\0') goto invalid; } if (((value << shift) >> shift) != value) goto invalid; *pvalue = value << shift; return true; invalid: gomp_error ("Invalid value for environment variable %s", name); return false; } /* Parse the GOMP_SPINCOUNT environment varible. Return true if one was present and it was successfully parsed. */ static bool parse_spincount (const char *name, unsigned long long *pvalue) { char *env, *end; unsigned long long value, mult = 1; env = getenv (name); if (env == NULL) return false; while (isspace ((unsigned char) *env)) ++env; if (*env == '\0') goto invalid; if (strncasecmp (env, "infinite", 8) == 0 || strncasecmp (env, "infinity", 8) == 0) { value = ~0ULL; end = env + 8; goto check_tail; } errno = 0; value = strtoull (env, &end, 10); if (errno) goto invalid; while (isspace ((unsigned char) *end)) ++end; if (*end != '\0') { switch (tolower ((unsigned char) *end)) { case 'k': mult = 1000LL; break; case 'm': mult = 1000LL * 1000LL; break; case 'g': mult = 1000LL * 1000LL * 1000LL; break; case 't': mult = 1000LL * 1000LL * 1000LL * 1000LL; break; default: goto invalid; } ++end; check_tail: while (isspace ((unsigned char) *end)) ++end; if (*end != '\0') goto invalid; } if (value > ~0ULL / mult) value = ~0ULL; else value *= mult; *pvalue = value; return true; invalid: gomp_error ("Invalid value for environment variable %s", name); return false; } /* Parse a boolean value for environment variable NAME and store the result in VALUE. */ static void parse_boolean (const char *name, bool *value) { const char *env; env = getenv (name); if (env == NULL) return; while (isspace ((unsigned char) *env)) ++env; if (strncasecmp (env, "true", 4) == 0) { *value = true; env += 4; } else if (strncasecmp (env, "false", 5) == 0) { *value = false; env += 5; } else env = "X"; while (isspace ((unsigned char) *env)) ++env; if (*env != '\0') gomp_error ("Invalid value for environment variable %s", name); } /* Parse the OMP_WAIT_POLICY environment variable and store the result in gomp_active_wait_policy. */ static int parse_wait_policy (void) { const char *env; int ret = -1; env = getenv ("OMP_WAIT_POLICY"); if (env == NULL) return -1; while (isspace ((unsigned char) *env)) ++env; if (strncasecmp (env, "active", 6) == 0) { ret = 1; env += 6; } else if (strncasecmp (env, "passive", 7) == 0) { ret = 0; env += 7; } else env = "X"; while (isspace ((unsigned char) *env)) ++env; if (*env == '\0') return ret; gomp_error ("Invalid value for environment variable OMP_WAIT_POLICY"); return -1; } /* Parse the GOMP_CPU_AFFINITY environment varible. Return true if one was present and it was successfully parsed. */ static bool parse_affinity (void) { char *env, *end; unsigned long cpu_beg, cpu_end, cpu_stride; unsigned short *cpus = NULL; size_t allocated = 0, used = 0, needed; env = getenv ("GOMP_CPU_AFFINITY"); if (env == NULL) return false; do { while (*env == ' ' || *env == '\t') env++; cpu_beg = strtoul (env, &end, 0); cpu_end = cpu_beg; cpu_stride = 1; if (env == end || cpu_beg >= 65536) goto invalid; env = end; if (*env == '-') { cpu_end = strtoul (++env, &end, 0); if (env == end || cpu_end >= 65536 || cpu_end < cpu_beg) goto invalid; env = end; if (*env == ':') { cpu_stride = strtoul (++env, &end, 0); if (env == end || cpu_stride == 0 || cpu_stride >= 65536) goto invalid; env = end; } } needed = (cpu_end - cpu_beg) / cpu_stride + 1; if (used + needed >= allocated) { unsigned short *new_cpus; if (allocated < 64) allocated = 64; if (allocated > needed) allocated <<= 1; else allocated += 2 * needed; new_cpus = realloc (cpus, allocated * sizeof (unsigned short)); if (new_cpus == NULL) { free (cpus); gomp_error ("not enough memory to store GOMP_CPU_AFFINITY list"); return false; } cpus = new_cpus; } while (needed--) { cpus[used++] = cpu_beg; cpu_beg += cpu_stride; } while (*env == ' ' || *env == '\t') env++; if (*env == ',') env++; else if (*env == '\0') break; } while (1); gomp_cpu_affinity = cpus; gomp_cpu_affinity_len = used; return true; invalid: gomp_error ("Invalid value for enviroment variable GOMP_CPU_AFFINITY"); return false; } static void __attribute__((constructor)) initialize_env (void) { unsigned long stacksize; int wait_policy; /* Do a compile time check that mkomp_h.pl did good job. */ omp_check_defines (); parse_schedule (); parse_boolean ("OMP_DYNAMIC", &gomp_global_icv.dyn_var); parse_boolean ("OMP_NESTED", &gomp_global_icv.nest_var); parse_unsigned_long ("OMP_MAX_ACTIVE_LEVELS", &gomp_max_active_levels_var, true); parse_unsigned_long ("OMP_THREAD_LIMIT", &gomp_thread_limit_var, false); if (gomp_thread_limit_var != ULONG_MAX) gomp_remaining_threads_count = gomp_thread_limit_var - 1; #ifndef HAVE_SYNC_BUILTINS gomp_mutex_init (&gomp_remaining_threads_lock); #endif gomp_init_num_threads (); gomp_available_cpus = gomp_global_icv.nthreads_var; if (!parse_unsigned_long ("OMP_NUM_THREADS", &gomp_global_icv.nthreads_var, false)) gomp_global_icv.nthreads_var = gomp_available_cpus; if (parse_affinity ()) gomp_init_affinity (); wait_policy = parse_wait_policy (); if (!parse_spincount ("GOMP_SPINCOUNT", &gomp_spin_count_var)) { /* Using a rough estimation of 100000 spins per msec, use 5 min blocking for OMP_WAIT_POLICY=active, 200 msec blocking when OMP_WAIT_POLICY is not specificed and 0 when OMP_WAIT_POLICY=passive. Depending on the CPU speed, this can be e.g. 5 times longer or 5 times shorter. */ if (wait_policy > 0) gomp_spin_count_var = 30000000000LL; else if (wait_policy < 0) gomp_spin_count_var = 20000000LL; } /* gomp_throttled_spin_count_var is used when there are more libgomp managed threads than available CPUs. Use very short spinning. */ if (wait_policy > 0) gomp_throttled_spin_count_var = 1000LL; else if (wait_policy < 0) gomp_throttled_spin_count_var = 100LL; if (gomp_throttled_spin_count_var > gomp_spin_count_var) gomp_throttled_spin_count_var = gomp_spin_count_var; /* Not strictly environment related, but ordering constructors is tricky. */ pthread_attr_init (&gomp_thread_attr); pthread_attr_setdetachstate (&gomp_thread_attr, PTHREAD_CREATE_DETACHED); if (parse_stacksize ("OMP_STACKSIZE", &stacksize) || parse_stacksize ("GOMP_STACKSIZE", &stacksize)) { int err; err = pthread_attr_setstacksize (&gomp_thread_attr, stacksize); #ifdef PTHREAD_STACK_MIN if (err == EINVAL) { if (stacksize < PTHREAD_STACK_MIN) gomp_error ("Stack size less than minimum of %luk", PTHREAD_STACK_MIN / 1024ul + (PTHREAD_STACK_MIN % 1024 != 0)); else gomp_error ("Stack size larger than system limit"); } else #endif if (err != 0) gomp_error ("Stack size change failed: %s", strerror (err)); } } /* The public OpenMP API routines that access these variables. */ void omp_set_num_threads (int n) { struct gomp_task_icv *icv = gomp_icv (true); icv->nthreads_var = (n > 0 ? n : 1); } void omp_set_dynamic (int val) { struct gomp_task_icv *icv = gomp_icv (true); icv->dyn_var = val; } int omp_get_dynamic (void) { struct gomp_task_icv *icv = gomp_icv (false); return icv->dyn_var; } void omp_set_nested (int val) { struct gomp_task_icv *icv = gomp_icv (true); icv->nest_var = val; } int omp_get_nested (void) { struct gomp_task_icv *icv = gomp_icv (false); return icv->nest_var; } void omp_set_schedule (omp_sched_t kind, int modifier) { struct gomp_task_icv *icv = gomp_icv (true); switch (kind) { case omp_sched_static: if (modifier < 1) modifier = 0; icv->run_sched_modifier = modifier; break; case omp_sched_dynamic: case omp_sched_guided: if (modifier < 1) modifier = 1; icv->run_sched_modifier = modifier; break; case omp_sched_auto: break; default: return; } icv->run_sched_var = kind; } void omp_get_schedule (omp_sched_t *kind, int *modifier) { struct gomp_task_icv *icv = gomp_icv (false); *kind = icv->run_sched_var; *modifier = icv->run_sched_modifier; } int omp_get_max_threads (void) { struct gomp_task_icv *icv = gomp_icv (false); return icv->nthreads_var; } int omp_get_thread_limit (void) { return gomp_thread_limit_var > INT_MAX ? INT_MAX : gomp_thread_limit_var; } void omp_set_max_active_levels (int max_levels) { if (max_levels >= 0) gomp_max_active_levels_var = max_levels; } int omp_get_max_active_levels (void) { return gomp_max_active_levels_var; } ialias (omp_set_dynamic) ialias (omp_set_nested) ialias (omp_set_num_threads) ialias (omp_get_dynamic) ialias (omp_get_nested) ialias (omp_set_schedule) ialias (omp_get_schedule) ialias (omp_get_max_threads) ialias (omp_get_thread_limit) ialias (omp_set_max_active_levels) ialias (omp_get_max_active_levels)
Go to most recent revision | Compare with Previous | Blame | View Log