]> git.ipfire.org Git - thirdparty/gcc.git/blame - libgomp/env.c
[multiple changes]
[thirdparty/gcc.git] / libgomp / env.c
CommitLineData
953ff289
DN
1/* Copyright (C) 2005 Free Software Foundation, Inc.
2 Contributed by Richard Henderson <rth@redhat.com>.
3
4 This file is part of the GNU OpenMP Library (libgomp).
5
6 Libgomp is free software; you can redistribute it and/or modify it
7 under the terms of the GNU Lesser General Public License as published by
8 the Free Software Foundation; either version 2.1 of the License, or
9 (at your option) any later version.
10
11 Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
12 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
13 FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for
14 more details.
15
16 You should have received a copy of the GNU Lesser General Public License
17 along with libgomp; see the file COPYING.LIB. If not, write to the
18 Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
19 MA 02110-1301, USA. */
20
21/* As a special exception, if you link this library with other files, some
22 of which are compiled with GCC, to produce an executable, this library
23 does not by itself cause the resulting executable to be covered by the
24 GNU General Public License. This exception does not however invalidate
25 any other reasons why the executable file might be covered by the GNU
26 General Public License. */
27
28/* This file defines the OpenMP internal control variables, and arranges
29 for them to be initialized from environment variables at startup. */
30
31#include "libgomp.h"
32#include "libgomp_f.h"
33#include <stdlib.h>
34#include <string.h>
35
36
37unsigned gomp_nthreads_var = 1;
38bool gomp_dyn_var = false;
39bool gomp_nest_var = false;
40enum gomp_schedule_type gomp_run_sched_var = GFS_DYNAMIC;
41unsigned gomp_run_sched_chunk = 1;
42
43/* Parse the OMP_SCHEDULE environment variable. */
44
45static void
46parse_schedule (void)
47{
48 char *env, *end;
49
50 env = getenv ("OMP_SCHEDULE");
51 if (env == NULL)
52 return;
53
54 if (strncmp (env, "static", 6) == 0)
55 {
56 gomp_run_sched_var = GFS_STATIC;
57 env += 6;
58 }
59 else if (strncmp (env, "dynamic", 7) == 0)
60 {
61 gomp_run_sched_var = GFS_DYNAMIC;
62 env += 7;
63 }
64 else if (strncmp (env, "guided", 6) == 0)
65 {
66 gomp_run_sched_var = GFS_GUIDED;
67 env += 6;
68 }
69 else
70 goto unknown;
71
72 if (*env == '\0')
73 return;
74 if (*env != ' ' && *env != ',')
75 goto unknown;
76 while (*env == ' ')
77 env++;
78 if (*env == '\0')
79 return;
80 if (*env != ',')
81 goto unknown;
82 if (*++env == '\0')
83 goto invalid;
84
85 gomp_run_sched_chunk = strtoul (env, &end, 10);
86 if (*end != '\0')
87 goto invalid;
88 return;
89
90 unknown:
91 gomp_error ("Unknown value for environment variable OMP_SCHEDULE");
92 return;
93
94 invalid:
95 gomp_error ("Invalid value for chunk size in "
96 "environment variable OMP_SCHEDULE");
97 gomp_run_sched_chunk = 1;
98 return;
99}
100
101/* Parse the OMP_NUM_THREADS environment varible. Return true if one was
102 present and it was successfully parsed. */
103
104static bool
105parse_num_threads (void)
106{
107 char *env, *end;
108
109 env = getenv ("OMP_NUM_THREADS");
110 if (env == NULL)
111 return false;
112
113 if (*env == '\0')
114 goto invalid;
115
116 gomp_nthreads_var = strtoul (env, &end, 10);
117 if (*end != '\0')
118 goto invalid;
119 return true;
120
121 invalid:
122 gomp_error ("Invalid value for enviroment variable OMP_NUM_THREADS");
123 gomp_nthreads_var = 1;
124 return false;
125}
126
127/* Parse a boolean value for environement variable NAME and store the
128 result in VALUE. */
129
130static void
131parse_boolean (const char *name, bool *value)
132{
133 const char *env;
134
135 env = getenv (name);
136 if (env == NULL)
137 return;
138
139 if (strcmp (env, "true") == 0)
140 *value = true;
141 else if (strcmp (env, "false") == 0)
142 *value = false;
143 else
144 gomp_error ("Invalid value for environement variable %s", name);
145}
146
147static void __attribute__((constructor))
148initialize_env (void)
149{
150 /* Do a compile time check that mkomp_h.pl did good job. */
151 omp_check_defines ();
152
153 parse_schedule ();
154 parse_boolean ("OMP_DYNAMIC", &gomp_dyn_var);
155 parse_boolean ("OMP_NESTED", &gomp_nest_var);
156 if (!parse_num_threads ())
157 gomp_init_num_threads ();
158}
159
160\f
161/* The public OpenMP API routines that access these variables. */
162
163void
164omp_set_num_threads (int n)
165{
166 gomp_nthreads_var = n;
167}
168
169void
170omp_set_dynamic (int val)
171{
172 gomp_dyn_var = val;
173}
174
175int
176omp_get_dynamic (void)
177{
178 return gomp_dyn_var;
179}
180
181void
182omp_set_nested (int val)
183{
184 gomp_nest_var = val;
185}
186
187int
188omp_get_nested (void)
189{
190 return gomp_nest_var;
191}
192
193ialias (omp_set_dynamic)
194ialias (omp_set_nested)
195ialias (omp_set_num_threads)
196ialias (omp_get_dynamic)
197ialias (omp_get_nested)