OSDN Git Service

PR ada/60703
[pf3gnuchains/gcc-fork.git] / libgomp / sections.c
1 /* Copyright (C) 2005, 2007, 2008, 2009, 2010 Free Software Foundation, Inc.
2    Contributed by Richard Henderson <rth@redhat.com>.
3
4    This file is part of the GNU OpenMP Library (libgomp).
5
6    Libgomp is free software; you can redistribute it and/or modify it
7    under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 3, or (at your option)
9    any later version.
10
11    Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
12    WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
13    FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
14    more details.
15
16    Under Section 7 of GPL version 3, you are granted additional
17    permissions described in the GCC Runtime Library Exception, version
18    3.1, as published by the Free Software Foundation.
19
20    You should have received a copy of the GNU General Public License and
21    a copy of the GCC Runtime Library Exception along with this program;
22    see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
23    <http://www.gnu.org/licenses/>.  */
24
25 /* This file handles the SECTIONS construct.  */
26
27 #include "libgomp.h"
28
29
30 /* Initialize the given work share construct from the given arguments.  */
31
32 static inline void
33 gomp_sections_init (struct gomp_work_share *ws, unsigned count)
34 {
35   ws->sched = GFS_DYNAMIC;
36   ws->chunk_size = 1;
37   ws->end = count + 1L;
38   ws->incr = 1;
39   ws->next = 1;
40 #ifdef HAVE_SYNC_BUILTINS
41   /* Prepare things to make each iteration faster.  */
42   if (sizeof (long) > sizeof (unsigned))
43     ws->mode = 1;
44   else
45     {
46       struct gomp_thread *thr = gomp_thread ();
47       struct gomp_team *team = thr->ts.team;
48       long nthreads = team ? team->nthreads : 1;
49
50       ws->mode = ((nthreads | ws->end)
51                   < 1UL << (sizeof (long) * __CHAR_BIT__ / 2 - 1));
52     }
53 #else
54   ws->mode = 0;
55 #endif
56 }
57
58 /* This routine is called when first encountering a sections construct
59    that is not bound directly to a parallel construct.  The first thread 
60    that arrives will create the work-share construct; subsequent threads
61    will see the construct exists and allocate work from it.
62
63    COUNT is the number of sections in this construct.
64
65    Returns the 1-based section number for this thread to perform, or 0 if
66    all work was assigned to other threads prior to this thread's arrival.  */
67
68 unsigned
69 GOMP_sections_start (unsigned count)
70 {
71   struct gomp_thread *thr = gomp_thread ();
72   long s, e, ret;
73
74   if (gomp_work_share_start (false))
75     {
76       gomp_sections_init (thr->ts.work_share, count);
77       gomp_work_share_init_done ();
78     }
79
80 #ifdef HAVE_SYNC_BUILTINS
81   if (gomp_iter_dynamic_next (&s, &e))
82     ret = s;
83   else
84     ret = 0;
85 #else
86   gomp_mutex_lock (&thr->ts.work_share->lock);
87   if (gomp_iter_dynamic_next_locked (&s, &e))
88     ret = s;
89   else
90     ret = 0;
91   gomp_mutex_unlock (&thr->ts.work_share->lock);
92 #endif
93
94   return ret;
95 }
96
97 /* This routine is called when the thread completes processing of the
98    section currently assigned to it.  If the work-share construct is
99    bound directly to a parallel construct, then the construct may have
100    been set up before the parallel.  In which case, this may be the
101    first iteration for the thread.
102
103    Returns the 1-based section number for this thread to perform, or 0 if
104    all work was assigned to other threads prior to this thread's arrival.  */
105
106 unsigned
107 GOMP_sections_next (void)
108 {
109   long s, e, ret;
110
111 #ifdef HAVE_SYNC_BUILTINS
112   if (gomp_iter_dynamic_next (&s, &e))
113     ret = s;
114   else
115     ret = 0;
116 #else
117   struct gomp_thread *thr = gomp_thread ();
118
119   gomp_mutex_lock (&thr->ts.work_share->lock);
120   if (gomp_iter_dynamic_next_locked (&s, &e))
121     ret = s;
122   else
123     ret = 0;
124   gomp_mutex_unlock (&thr->ts.work_share->lock);
125 #endif
126
127   return ret;
128 }
129
130 /* This routine pre-initializes a work-share construct to avoid one
131    synchronization once we get into the loop.  */
132
133 void
134 GOMP_parallel_sections_start (void (*fn) (void *), void *data,
135                               unsigned num_threads, unsigned count)
136 {
137   struct gomp_team *team;
138
139   num_threads = gomp_resolve_num_threads (num_threads, count);
140   team = gomp_new_team (num_threads);
141   gomp_sections_init (&team->work_shares[0], count);
142   gomp_team_start (fn, data, num_threads, team);
143 }
144
145 /* The GOMP_section_end* routines are called after the thread is told
146    that all sections are complete.  This first version synchronizes
147    all threads; the nowait version does not.  */
148
149 void
150 GOMP_sections_end (void)
151 {
152   gomp_work_share_end ();
153 }
154
155 void
156 GOMP_sections_end_nowait (void)
157 {
158   gomp_work_share_end_nowait ();
159 }