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     {
 120       if (max_levels <= gomp_supported_active_levels)
 121         gomp_max_active_levels_var = max_levels;
 122       else
 123         gomp_max_active_levels_var = gomp_supported_active_levels;
 124     }
 125 }
 126
 127 int
 128 omp_get_max_active_levels (void)
 129 {
 130   return gomp_max_active_levels_var;
 131 }
 132
 133 int
 134 omp_get_supported_active_levels (void)
 135 {
 136   return gomp_supported_active_levels;
 137 }
 138
 139 int
 140 omp_get_cancellation (void)
 141 {
 142   return gomp_cancel_var;
 143 }
 144
 145 int
 146 omp_get_max_task_priority (void)
 147 {
 148   return gomp_max_task_priority_var;
 149 }
 150
 151 omp_proc_bind_t
 152 omp_get_proc_bind (void)
 153 {
 154   struct gomp_task_icv *icv = gomp_icv (false);
 155   return icv->bind_var;
 156 }
 157
 158 int
 159 omp_get_num_places (void)
 160 {
 161   return gomp_places_list_len;
 162 }
 163
 164 int
 165 omp_get_place_num (void)
 166 {
 167   if (gomp_places_list == NULL)
 168     return -1;
 169
 170   struct gomp_thread *thr = gomp_thread ();
 171   if (thr->place == 0)
 172     gomp_init_affinity ();
 173
 174   return (int) thr->place - 1;
 175 }
 176
 177 int
 178 omp_get_partition_num_places (void)
 179 {
 180   if (gomp_places_list == NULL)
 181     return 0;
 182
 183   struct gomp_thread *thr = gomp_thread ();
 184   if (thr->place == 0)
 185     gomp_init_affinity ();
 186
 187   return thr->ts.place_partition_len;
 188 }
 189
 190 void
 191 omp_get_partition_place_nums (int *place_nums)
 192 {
 193   if (gomp_places_list == NULL)
 194     return;
 195
 196   struct gomp_thread *thr = gomp_thread ();
 197   if (thr->place == 0)
 198     gomp_init_affinity ();
 199
 200   unsigned int i;
 201   for (i = 0; i < thr->ts.place_partition_len; i++)
 202     *place_nums++ = thr->ts.place_partition_off + i;
 203 }
 204
 205 void
 206 omp_set_default_allocator (omp_allocator_handle_t allocator)
 207 {
 208   struct gomp_thread *thr = gomp_thread ();
 209   if (allocator == omp_null_allocator)
 210     allocator = omp_default_mem_alloc;
 211   thr->ts.def_allocator = (uintptr_t) allocator;
 212 }
 213
 214 omp_allocator_handle_t
 215 omp_get_default_allocator (void)
 216 {
 217   struct gomp_thread *thr = gomp_thread ();
 218   if (thr->ts.def_allocator == omp_null_allocator)
 219     return (omp_allocator_handle_t) gomp_def_allocator;
 220   else
 221     return (omp_allocator_handle_t) thr->ts.def_allocator;
 222 }
 223
 224 ialias (omp_set_dynamic)
 225 ialias (omp_set_nested)
 226 ialias (omp_set_num_threads)
 227 ialias (omp_get_dynamic)
 228 ialias (omp_get_nested)
 229 ialias (omp_set_schedule)
 230 ialias (omp_get_schedule)
 231 ialias (omp_get_max_threads)
 232 ialias (omp_get_thread_limit)
 233 ialias (omp_set_max_active_levels)
 234 ialias (omp_get_max_active_levels)
 235 ialias (omp_get_supported_active_levels)
 236 ialias (omp_get_cancellation)
 237 ialias (omp_get_proc_bind)
 238 ialias (omp_get_max_task_priority)
 239 ialias (omp_get_num_places)
 240 ialias (omp_get_place_num)
 241 ialias (omp_get_partition_num_places)
 242 ialias (omp_get_partition_place_nums)
 243 ialias (omp_set_default_allocator)
 244 ialias (omp_get_default_allocator)