1/* Copyright (C) 2005-2020 Free Software Foundation, Inc.
2   Contributed by Richard Henderson <rth@redhat.com>.
3
4   This file is part of the GNU Offloading and Multi Processing Library
5   (libgomp).
6
7   Libgomp is free software; you can redistribute it and/or modify it
8   under the terms of the GNU General Public License as published by
9   the Free Software Foundation; either version 3, or (at your option)
10   any later version.
11
12   Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
13   WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
14   FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
15   more details.
16
17   Under Section 7 of GPL version 3, you are granted additional
18   permissions described in the GCC Runtime Library Exception, version
19   3.1, as published by the Free Software Foundation.
20
21   You should have received a copy of the GNU General Public License and
22   a copy of the GCC Runtime Library Exception along with this program;
23   see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
24   <http://www.gnu.org/licenses/>.  */
25
26/* This file defines the OpenMP API entry points that operate on internal
27   control variables.  */
28
29#include "libgomp.h"
30#include "gomp-constants.h"
31#include <limits.h>
32
33void
34omp_set_num_threads (int n)
35{
36  struct gomp_task_icv *icv = gomp_icv (true);
37  icv->nthreads_var = (n > 0 ? n : 1);
38}
39
40void
41omp_set_dynamic (int val)
42{
43  struct gomp_task_icv *icv = gomp_icv (true);
44  icv->dyn_var = val;
45}
46
47int
48omp_get_dynamic (void)
49{
50  struct gomp_task_icv *icv = gomp_icv (false);
51  return icv->dyn_var;
52}
53
54void
55omp_set_nested (int val)
56{
57  struct gomp_task_icv *icv = gomp_icv (true);
58  icv->nest_var = val;
59}
60
61int
62omp_get_nested (void)
63{
64  struct gomp_task_icv *icv = gomp_icv (false);
65  return icv->nest_var;
66}
67
68void
69omp_set_schedule (omp_sched_t kind, int chunk_size)
70{
71  struct gomp_task_icv *icv = gomp_icv (true);
72  switch (kind & ~omp_sched_monotonic)
73    {
74    case omp_sched_static:
75      if (chunk_size < 1)
76	chunk_size = 0;
77      icv->run_sched_chunk_size = chunk_size;
78      break;
79    case omp_sched_dynamic:
80    case omp_sched_guided:
81      if (chunk_size < 1)
82	chunk_size = 1;
83      icv->run_sched_chunk_size = chunk_size;
84      break;
85    case omp_sched_auto:
86      break;
87    default:
88      return;
89    }
90  icv->run_sched_var = kind;
91}
92
93void
94omp_get_schedule (omp_sched_t *kind, int *chunk_size)
95{
96  struct gomp_task_icv *icv = gomp_icv (false);
97  *kind = icv->run_sched_var;
98  *chunk_size = icv->run_sched_chunk_size;
99}
100
101int
102omp_get_max_threads (void)
103{
104  struct gomp_task_icv *icv = gomp_icv (false);
105  return icv->nthreads_var;
106}
107
108int
109omp_get_thread_limit (void)
110{
111  struct gomp_task_icv *icv = gomp_icv (false);
112  return icv->thread_limit_var > INT_MAX ? INT_MAX : icv->thread_limit_var;
113}
114
115void
116omp_set_max_active_levels (int max_levels)
117{
118  if (max_levels >= 0)
119    gomp_max_active_levels_var = max_levels;
120}
121
122int
123omp_get_max_active_levels (void)
124{
125  return gomp_max_active_levels_var;
126}
127
128int
129omp_get_cancellation (void)
130{
131  return gomp_cancel_var;
132}
133
134int
135omp_get_max_task_priority (void)
136{
137  return gomp_max_task_priority_var;
138}
139
140omp_proc_bind_t
141omp_get_proc_bind (void)
142{
143  struct gomp_task_icv *icv = gomp_icv (false);
144  return icv->bind_var;
145}
146
147int
148omp_get_initial_device (void)
149{
150  return GOMP_DEVICE_HOST_FALLBACK;
151}
152
153int
154omp_get_num_places (void)
155{
156  return gomp_places_list_len;
157}
158
159int
160omp_get_place_num (void)
161{
162  if (gomp_places_list == NULL)
163    return -1;
164
165  struct gomp_thread *thr = gomp_thread ();
166  if (thr->place == 0)
167    gomp_init_affinity ();
168
169  return (int) thr->place - 1;
170}
171
172int
173omp_get_partition_num_places (void)
174{
175  if (gomp_places_list == NULL)
176    return 0;
177
178  struct gomp_thread *thr = gomp_thread ();
179  if (thr->place == 0)
180    gomp_init_affinity ();
181
182  return thr->ts.place_partition_len;
183}
184
185void
186omp_get_partition_place_nums (int *place_nums)
187{
188  if (gomp_places_list == NULL)
189    return;
190
191  struct gomp_thread *thr = gomp_thread ();
192  if (thr->place == 0)
193    gomp_init_affinity ();
194
195  unsigned int i;
196  for (i = 0; i < thr->ts.place_partition_len; i++)
197    *place_nums++ = thr->ts.place_partition_off + i;
198}
199
200ialias (omp_set_dynamic)
201ialias (omp_set_nested)
202ialias (omp_set_num_threads)
203ialias (omp_get_dynamic)
204ialias (omp_get_nested)
205ialias (omp_set_schedule)
206ialias (omp_get_schedule)
207ialias (omp_get_max_threads)
208ialias (omp_get_thread_limit)
209ialias (omp_set_max_active_levels)
210ialias (omp_get_max_active_levels)
211ialias (omp_get_cancellation)
212ialias (omp_get_proc_bind)
213ialias (omp_get_initial_device)
214ialias (omp_get_max_task_priority)
215ialias (omp_get_num_places)
216ialias (omp_get_place_num)
217ialias (omp_get_partition_num_places)
218ialias (omp_get_partition_place_nums)
219