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