(root)/
gcc-13.2.0/
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)