libgomp/icv.c

   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
  33 void
  34 omp_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
  40 void
  41 omp_set_dynamic (int val)
  42 {
  43   struct gomp_task_icv *icv = gomp_icv (true);
  44   icv->dyn_var = val;
  45 }
  46
  47 int
  48 omp_get_dynamic (void)
  49 {
  50   struct gomp_task_icv *icv = gomp_icv (false);
  51   return icv->dyn_var;
  52 }
  53
  54 void
  55 omp_set_nested (int val)
  56 {
  57   struct gomp_task_icv *icv = gomp_icv (true);
  58   icv->nest_var = val;
  59 }
  60
  61 int
  62 omp_get_nested (void)
  63 {
  64   struct gomp_task_icv *icv = gomp_icv (false);
  65   return icv->nest_var;
  66 }
  67
  68 void
  69 omp_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
  93 void
  94 omp_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
 101 int
 102 omp_get_max_threads (void)
 103 {
 104   struct gomp_task_icv *icv = gomp_icv (false);
 105   return icv->nthreads_var;
 106 }
 107
 108 int
 109 omp_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
 115 void
 116 omp_set_max_active_levels (int max_levels)
 117 {
 118   if (max_levels >= 0)
 119     gomp_max_active_levels_var = max_levels;
 120 }
 121
 122 int
 123 omp_get_max_active_levels (void)
 124 {
 125   return gomp_max_active_levels_var;
 126 }
 127
 128 int
 129 omp_get_cancellation (void)
 130 {
 131   return gomp_cancel_var;
 132 }
 133
 134 int
 135 omp_get_max_task_priority (void)
 136 {
 137   return gomp_max_task_priority_var;
 138 }
 139
 140 omp_proc_bind_t
 141 omp_get_proc_bind (void)
 142 {
 143   struct gomp_task_icv *icv = gomp_icv (false);
 144   return icv->bind_var;
 145 }
 146
 147 int
 148 omp_get_initial_device (void)
 149 {
 150   return GOMP_DEVICE_HOST_FALLBACK;
 151 }
 152
 153 int
 154 omp_get_num_places (void)
 155 {
 156   return gomp_places_list_len;
 157 }
 158
 159 int
 160 omp_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
 172 int
 173 omp_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
 185 void
 186 omp_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
 200 void
 201 omp_set_default_allocator (omp_allocator_handle_t allocator)
 202 {
 203   struct gomp_thread *thr = gomp_thread ();
 204   if (allocator == omp_null_allocator)
 205     allocator = omp_default_mem_alloc;
 206   thr->ts.def_allocator = (uintptr_t) allocator;
 207 }
 208
 209 omp_allocator_handle_t
 210 omp_get_default_allocator (void)
 211 {
 212   struct gomp_thread *thr = gomp_thread ();
 213   if (thr->ts.def_allocator == omp_null_allocator)
 214     return (omp_allocator_handle_t) gomp_def_allocator;
 215   else
 216     return (omp_allocator_handle_t) thr->ts.def_allocator;
 217 }
 218
 219 ialias (omp_set_dynamic)
 220 ialias (omp_set_nested)
 221 ialias (omp_set_num_threads)
 222 ialias (omp_get_dynamic)
 223 ialias (omp_get_nested)
 224 ialias (omp_set_schedule)
 225 ialias (omp_get_schedule)
 226 ialias (omp_get_max_threads)
 227 ialias (omp_get_thread_limit)
 228 ialias (omp_set_max_active_levels)
 229 ialias (omp_get_max_active_levels)
 230 ialias (omp_get_cancellation)
 231 ialias (omp_get_proc_bind)
 232 ialias (omp_get_initial_device)
 233 ialias (omp_get_max_task_priority)
 234 ialias (omp_get_num_places)
 235 ialias (omp_get_place_num)
 236 ialias (omp_get_partition_num_places)
 237 ialias (omp_get_partition_place_nums)
 238 ialias (omp_set_default_allocator)
 239 ialias (omp_get_default_allocator)