libgomp/icv-device.c

   1 /* Copyright (C) 2005-2023 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 OpenMP API entry points that accelerator targets are
  27    expected to replace.  */
  28
  29 #include "libgomp.h"
  30 #include <limits.h>
  31
  32 void
  33 omp_set_default_device (int device_num)
  34 {
  35   struct gomp_task_icv *icv = gomp_icv (true);
  36   icv->default_device_var = device_num;
  37 }
  38
  39 ialias (omp_set_default_device)
  40
  41 int
  42 omp_get_default_device (void)
  43 {
  44   struct gomp_task_icv *icv = gomp_icv (false);
  45   if (icv->default_device_var == INT_MIN)
  46     /* This implies OMP_TARGET_OFFLOAD=mandatory.  */
  47     gomp_init_targets_once ();
  48   return icv->default_device_var;
  49 }
  50
  51 ialias (omp_get_default_device)
  52
  53 int
  54 omp_get_initial_device (void)
  55 {
  56   return gomp_get_num_devices ();
  57 }
  58
  59 ialias (omp_get_initial_device)
  60
  61 int
  62 omp_get_num_devices (void)
  63 {
  64   return gomp_get_num_devices ();
  65 }
  66
  67 ialias (omp_get_num_devices)
  68
  69 int
  70 omp_is_initial_device (void)
  71 {
  72   /* Hardcoded to 1 on host, should be 0 on MIC, HSAIL, PTX.  */
  73   return 1;
  74 }
  75
  76 ialias (omp_is_initial_device)
  77
  78 int
  79 omp_get_device_num (void)
  80 {
  81   /* By specification, this is equivalent to omp_get_initial_device
  82      on the host.  */
  83   return ialias_call (omp_get_initial_device) ();
  84 }
  85
  86 ialias (omp_get_device_num)
  87
  88 int
  89 omp_get_max_teams (void)
  90 {
  91   return gomp_nteams_var;
  92 }
  93
  94 ialias (omp_get_max_teams)
  95
  96 void
  97 omp_set_num_teams (int num_teams)
  98 {
  99   if (num_teams >= 0)
 100     gomp_nteams_var = num_teams;
 101 }
 102
 103 ialias (omp_set_num_teams)
 104
 105 int
 106 omp_get_teams_thread_limit (void)
 107 {
 108   return gomp_teams_thread_limit_var;
 109 }
 110
 111 ialias (omp_get_teams_thread_limit)
 112
 113 void
 114 omp_set_teams_thread_limit (int thread_limit)
 115 {
 116   if (thread_limit >= 0)
 117     gomp_teams_thread_limit_var = thread_limit;
 118 }
 119
 120 ialias (omp_set_teams_thread_limit)