|
145
|
1 /* Copyright (C) 2018-2020 Free Software Foundation, Inc.
|
|
|
2 Contributed by Jakub Jelinek <jakub@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 handles the host TEAMS construct. */
|
|
|
27
|
|
|
28 #include "libgomp.h"
|
|
|
29 #include <limits.h>
|
|
|
30
|
|
|
31 static unsigned gomp_num_teams = 1, gomp_team_num = 0;
|
|
|
32
|
|
|
33 void
|
|
|
34 GOMP_teams_reg (void (*fn) (void *), void *data, unsigned int num_teams,
|
|
|
35 unsigned int thread_limit, unsigned int flags)
|
|
|
36 {
|
|
|
37 (void) flags;
|
|
|
38 (void) num_teams;
|
|
|
39 unsigned old_thread_limit_var = 0;
|
|
|
40 if (thread_limit)
|
|
|
41 {
|
|
|
42 struct gomp_task_icv *icv = gomp_icv (true);
|
|
|
43 old_thread_limit_var = icv->thread_limit_var;
|
|
|
44 icv->thread_limit_var
|
|
|
45 = thread_limit > INT_MAX ? UINT_MAX : thread_limit;
|
|
|
46 }
|
|
|
47 if (num_teams == 0)
|
|
|
48 num_teams = 3;
|
|
|
49 gomp_num_teams = num_teams;
|
|
|
50 for (gomp_team_num = 0; gomp_team_num < num_teams; gomp_team_num++)
|
|
|
51 fn (data);
|
|
|
52 gomp_num_teams = 1;
|
|
|
53 gomp_team_num = 0;
|
|
|
54 if (thread_limit)
|
|
|
55 {
|
|
|
56 struct gomp_task_icv *icv = gomp_icv (true);
|
|
|
57 icv->thread_limit_var = old_thread_limit_var;
|
|
|
58 }
|
|
|
59 }
|
|
|
60
|
|
|
61 int
|
|
|
62 omp_get_num_teams (void)
|
|
|
63 {
|
|
|
64 return gomp_num_teams;
|
|
|
65 }
|
|
|
66
|
|
|
67 int
|
|
|
68 omp_get_team_num (void)
|
|
|
69 {
|
|
|
70 return gomp_team_num;
|
|
|
71 }
|
|
|
72
|
|
|
73 ialias (omp_get_num_teams)
|
|
|
74 ialias (omp_get_team_num)
|
