Branch data Line data Source code
1 : : /* General types and functions that are useful for processing of OpenMP,
2 : : OpenACC and similar directives at various stages of compilation.
3 : :
4 : : Copyright (C) 2005-2024 Free Software Foundation, Inc.
5 : :
6 : : This file is part of GCC.
7 : :
8 : : GCC is free software; you can redistribute it and/or modify it under
9 : : the terms of the GNU General Public License as published by the Free
10 : : Software Foundation; either version 3, or (at your option) any later
11 : : version.
12 : :
13 : : GCC is distributed in the hope that it will be useful, but WITHOUT ANY
14 : : WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 : : FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16 : : for more details.
17 : :
18 : : You should have received a copy of the GNU General Public License
19 : : along with GCC; see the file COPYING3. If not see
20 : : <http://www.gnu.org/licenses/>. */
21 : :
22 : : #include "config.h"
23 : : #include "system.h"
24 : : #include "coretypes.h"
25 : : #include "backend.h"
26 : : #include "target.h"
27 : : #include "tree.h"
28 : : #include "gimple.h"
29 : : #include "ssa.h"
30 : : #include "diagnostic-core.h"
31 : : #include "fold-const.h"
32 : : #include "langhooks.h"
33 : : #include "omp-general.h"
34 : : #include "stringpool.h"
35 : : #include "attribs.h"
36 : : #include "gimplify.h"
37 : : #include "cgraph.h"
38 : : #include "alloc-pool.h"
39 : : #include "symbol-summary.h"
40 : : #include "tree-pass.h"
41 : : #include "omp-device-properties.h"
42 : : #include "tree-iterator.h"
43 : : #include "data-streamer.h"
44 : : #include "streamer-hooks.h"
45 : : #include "opts.h"
46 : : #include "tree-pretty-print.h"
47 : :
48 : : enum omp_requires omp_requires_mask;
49 : :
50 : : /* Find an OMP clause of type KIND within CLAUSES. */
51 : : tree
52 : 1174654 : omp_find_clause (tree clauses, enum omp_clause_code kind)
53 : : {
54 : 4333812 : for (; clauses ; clauses = OMP_CLAUSE_CHAIN (clauses))
55 : 3410262 : if (OMP_CLAUSE_CODE (clauses) == kind)
56 : : return clauses;
57 : :
58 : : return NULL_TREE;
59 : : }
60 : :
61 : : /* True if OpenMP should regard this DECL as being a scalar which has Fortran's
62 : : allocatable or pointer attribute. */
63 : : bool
64 : 15346 : omp_is_allocatable_or_ptr (tree decl)
65 : : {
66 : 15346 : return lang_hooks.decls.omp_is_allocatable_or_ptr (decl);
67 : : }
68 : :
69 : : /* Check whether this DECL belongs to a Fortran optional argument.
70 : : With 'for_present_check' set to false, decls which are optional parameters
71 : : themselve are returned as tree - or a NULL_TREE otherwise. Those decls are
72 : : always pointers. With 'for_present_check' set to true, the decl for checking
73 : : whether an argument is present is returned; for arguments with value
74 : : attribute this is the hidden argument and of BOOLEAN_TYPE. If the decl is
75 : : unrelated to optional arguments, NULL_TREE is returned. */
76 : :
77 : : tree
78 : 12884 : omp_check_optional_argument (tree decl, bool for_present_check)
79 : : {
80 : 12884 : return lang_hooks.decls.omp_check_optional_argument (decl, for_present_check);
81 : : }
82 : :
83 : : /* Return true if TYPE is an OpenMP mappable type. */
84 : :
85 : : bool
86 : 44538 : omp_mappable_type (tree type)
87 : : {
88 : : /* Mappable type has to be complete. */
89 : 44538 : if (type == error_mark_node || !COMPLETE_TYPE_P (type))
90 : 188 : return false;
91 : : return true;
92 : : }
93 : :
94 : : /* True if OpenMP should privatize what this DECL points to rather
95 : : than the DECL itself. */
96 : :
97 : : bool
98 : 4904228 : omp_privatize_by_reference (tree decl)
99 : : {
100 : 4904228 : return lang_hooks.decls.omp_privatize_by_reference (decl);
101 : : }
102 : :
103 : : /* Adjust *COND_CODE and *N2 so that the former is either LT_EXPR or GT_EXPR,
104 : : given that V is the loop index variable and STEP is loop step. */
105 : :
106 : : void
107 : 192049 : omp_adjust_for_condition (location_t loc, enum tree_code *cond_code, tree *n2,
108 : : tree v, tree step)
109 : : {
110 : 192049 : switch (*cond_code)
111 : : {
112 : : case LT_EXPR:
113 : : case GT_EXPR:
114 : : break;
115 : :
116 : 29865 : case NE_EXPR:
117 : 29865 : gcc_assert (TREE_CODE (step) == INTEGER_CST);
118 : 29865 : if (TREE_CODE (TREE_TYPE (v)) == INTEGER_TYPE
119 : 29865 : || TREE_CODE (TREE_TYPE (v)) == BITINT_TYPE)
120 : : {
121 : 25021 : if (integer_onep (step))
122 : 17737 : *cond_code = LT_EXPR;
123 : : else
124 : : {
125 : 7284 : gcc_assert (integer_minus_onep (step));
126 : 7284 : *cond_code = GT_EXPR;
127 : : }
128 : : }
129 : : else
130 : : {
131 : 4844 : tree unit = TYPE_SIZE_UNIT (TREE_TYPE (TREE_TYPE (v)));
132 : 4844 : gcc_assert (TREE_CODE (unit) == INTEGER_CST);
133 : 4844 : if (tree_int_cst_equal (unit, step))
134 : 3358 : *cond_code = LT_EXPR;
135 : : else
136 : : {
137 : 1486 : gcc_assert (wi::neg (wi::to_widest (unit))
138 : : == wi::to_widest (step));
139 : 1486 : *cond_code = GT_EXPR;
140 : : }
141 : : }
142 : :
143 : : break;
144 : :
145 : 27255 : case LE_EXPR:
146 : 27255 : if (POINTER_TYPE_P (TREE_TYPE (*n2)))
147 : : {
148 : 123 : tree unit = TYPE_SIZE_UNIT (TREE_TYPE (TREE_TYPE (v)));
149 : 123 : gcc_assert (TREE_CODE (unit) == INTEGER_CST);
150 : 123 : *n2 = fold_build_pointer_plus_loc (loc, *n2, unit);
151 : : }
152 : : else
153 : 27132 : *n2 = fold_build2_loc (loc, PLUS_EXPR, TREE_TYPE (*n2), *n2,
154 : 27132 : build_int_cst (TREE_TYPE (*n2), 1));
155 : 27255 : *cond_code = LT_EXPR;
156 : 27255 : break;
157 : 1808 : case GE_EXPR:
158 : 1808 : if (POINTER_TYPE_P (TREE_TYPE (*n2)))
159 : : {
160 : 137 : tree unit = TYPE_SIZE_UNIT (TREE_TYPE (TREE_TYPE (v)));
161 : 137 : gcc_assert (TREE_CODE (unit) == INTEGER_CST);
162 : 137 : unit = convert_to_ptrofftype_loc (loc, unit);
163 : 137 : unit = fold_build1_loc (loc, NEGATE_EXPR, TREE_TYPE (unit),
164 : : unit);
165 : 137 : *n2 = fold_build_pointer_plus_loc (loc, *n2, unit);
166 : : }
167 : : else
168 : 1671 : *n2 = fold_build2_loc (loc, MINUS_EXPR, TREE_TYPE (*n2), *n2,
169 : 1671 : build_int_cst (TREE_TYPE (*n2), 1));
170 : 1808 : *cond_code = GT_EXPR;
171 : 1808 : break;
172 : 0 : default:
173 : 0 : gcc_unreachable ();
174 : : }
175 : 192049 : }
176 : :
177 : : /* Return the looping step from INCR, extracted from the step of a gimple omp
178 : : for statement. */
179 : :
180 : : tree
181 : 190885 : omp_get_for_step_from_incr (location_t loc, tree incr)
182 : : {
183 : 190885 : tree step;
184 : 190885 : switch (TREE_CODE (incr))
185 : : {
186 : 168654 : case PLUS_EXPR:
187 : 168654 : step = TREE_OPERAND (incr, 1);
188 : 168654 : break;
189 : 16136 : case POINTER_PLUS_EXPR:
190 : 16136 : step = fold_convert (ssizetype, TREE_OPERAND (incr, 1));
191 : 16136 : break;
192 : 6095 : case MINUS_EXPR:
193 : 6095 : step = TREE_OPERAND (incr, 1);
194 : 6095 : step = fold_build1_loc (loc, NEGATE_EXPR, TREE_TYPE (step), step);
195 : 6095 : break;
196 : 0 : default:
197 : 0 : gcc_unreachable ();
198 : : }
199 : 190885 : return step;
200 : : }
201 : :
202 : : /* Extract the header elements of parallel loop FOR_STMT and store
203 : : them into *FD. */
204 : :
205 : : void
206 : 125082 : omp_extract_for_data (gomp_for *for_stmt, struct omp_for_data *fd,
207 : : struct omp_for_data_loop *loops)
208 : : {
209 : 125082 : tree t, var, *collapse_iter, *collapse_count;
210 : 125082 : tree count = NULL_TREE, iter_type = long_integer_type_node;
211 : 125082 : struct omp_for_data_loop *loop;
212 : 125082 : int i;
213 : 125082 : struct omp_for_data_loop dummy_loop;
214 : 125082 : location_t loc = gimple_location (for_stmt);
215 : 125082 : bool simd = gimple_omp_for_kind (for_stmt) == GF_OMP_FOR_KIND_SIMD;
216 : 125082 : bool distribute = gimple_omp_for_kind (for_stmt)
217 : 125082 : == GF_OMP_FOR_KIND_DISTRIBUTE;
218 : 125082 : bool taskloop = gimple_omp_for_kind (for_stmt)
219 : 125082 : == GF_OMP_FOR_KIND_TASKLOOP;
220 : 125082 : bool order_reproducible = false;
221 : 125082 : tree iterv, countv;
222 : :
223 : 125082 : fd->for_stmt = for_stmt;
224 : 125082 : fd->pre = NULL;
225 : 125082 : fd->have_nowait = distribute || simd;
226 : 125082 : fd->have_ordered = false;
227 : 125082 : fd->have_reductemp = false;
228 : 125082 : fd->have_pointer_condtemp = false;
229 : 125082 : fd->have_scantemp = false;
230 : 125082 : fd->have_nonctrl_scantemp = false;
231 : 125082 : fd->non_rect = false;
232 : 125082 : fd->lastprivate_conditional = 0;
233 : 125082 : fd->tiling = NULL_TREE;
234 : 125082 : fd->collapse = 1;
235 : 125082 : fd->ordered = 0;
236 : 125082 : fd->first_nonrect = -1;
237 : 125082 : fd->last_nonrect = -1;
238 : 125082 : fd->sched_kind = OMP_CLAUSE_SCHEDULE_STATIC;
239 : 125082 : fd->sched_modifiers = 0;
240 : 125082 : fd->chunk_size = NULL_TREE;
241 : 125082 : fd->simd_schedule = false;
242 : 125082 : fd->first_inner_iterations = NULL_TREE;
243 : 125082 : fd->factor = NULL_TREE;
244 : 125082 : fd->adjn1 = NULL_TREE;
245 : 125082 : collapse_iter = NULL;
246 : 125082 : collapse_count = NULL;
247 : :
248 : 635003 : for (t = gimple_omp_for_clauses (for_stmt); t ; t = OMP_CLAUSE_CHAIN (t))
249 : 509921 : switch (OMP_CLAUSE_CODE (t))
250 : : {
251 : 44430 : case OMP_CLAUSE_NOWAIT:
252 : 44430 : fd->have_nowait = true;
253 : 44430 : break;
254 : 1389 : case OMP_CLAUSE_ORDERED:
255 : 1389 : fd->have_ordered = true;
256 : 1389 : if (OMP_CLAUSE_ORDERED_DOACROSS (t))
257 : : {
258 : 692 : if (OMP_CLAUSE_ORDERED_EXPR (t))
259 : 672 : fd->ordered = tree_to_shwi (OMP_CLAUSE_ORDERED_EXPR (t));
260 : : else
261 : 20 : fd->ordered = -1;
262 : : }
263 : : break;
264 : 30367 : case OMP_CLAUSE_SCHEDULE:
265 : 30367 : gcc_assert (!distribute && !taskloop);
266 : 30367 : fd->sched_kind
267 : 30367 : = (enum omp_clause_schedule_kind)
268 : 30367 : (OMP_CLAUSE_SCHEDULE_KIND (t) & OMP_CLAUSE_SCHEDULE_MASK);
269 : 30367 : fd->sched_modifiers = (OMP_CLAUSE_SCHEDULE_KIND (t)
270 : 30367 : & ~OMP_CLAUSE_SCHEDULE_MASK);
271 : 30367 : fd->chunk_size = OMP_CLAUSE_SCHEDULE_CHUNK_EXPR (t);
272 : 30367 : fd->simd_schedule = OMP_CLAUSE_SCHEDULE_SIMD (t);
273 : 30367 : break;
274 : 7004 : case OMP_CLAUSE_DIST_SCHEDULE:
275 : 7004 : gcc_assert (distribute);
276 : 7004 : fd->chunk_size = OMP_CLAUSE_DIST_SCHEDULE_CHUNK_EXPR (t);
277 : 7004 : break;
278 : 41249 : case OMP_CLAUSE_COLLAPSE:
279 : 41249 : fd->collapse = tree_to_shwi (OMP_CLAUSE_COLLAPSE_EXPR (t));
280 : 41249 : if (fd->collapse > 1)
281 : : {
282 : 33684 : collapse_iter = &OMP_CLAUSE_COLLAPSE_ITERVAR (t);
283 : 33684 : collapse_count = &OMP_CLAUSE_COLLAPSE_COUNT (t);
284 : : }
285 : : break;
286 : 338 : case OMP_CLAUSE_TILE:
287 : 338 : fd->tiling = OMP_CLAUSE_TILE_LIST (t);
288 : 338 : fd->collapse = list_length (fd->tiling);
289 : 338 : gcc_assert (fd->collapse);
290 : 338 : collapse_iter = &OMP_CLAUSE_TILE_ITERVAR (t);
291 : 338 : collapse_count = &OMP_CLAUSE_TILE_COUNT (t);
292 : 338 : break;
293 : 482 : case OMP_CLAUSE__REDUCTEMP_:
294 : 482 : fd->have_reductemp = true;
295 : 482 : break;
296 : 40522 : case OMP_CLAUSE_LASTPRIVATE:
297 : 40522 : if (OMP_CLAUSE_LASTPRIVATE_CONDITIONAL (t))
298 : 1504 : fd->lastprivate_conditional++;
299 : : break;
300 : 1556 : case OMP_CLAUSE__CONDTEMP_:
301 : 1556 : if (POINTER_TYPE_P (TREE_TYPE (OMP_CLAUSE_DECL (t))))
302 : 185 : fd->have_pointer_condtemp = true;
303 : : break;
304 : 1102 : case OMP_CLAUSE__SCANTEMP_:
305 : 1102 : fd->have_scantemp = true;
306 : 1102 : if (!OMP_CLAUSE__SCANTEMP__ALLOC (t)
307 : 1102 : && !OMP_CLAUSE__SCANTEMP__CONTROL (t))
308 : 205 : fd->have_nonctrl_scantemp = true;
309 : : break;
310 : 7544 : case OMP_CLAUSE_ORDER:
311 : : /* FIXME: For OpenMP 5.2 this should change to
312 : : if (OMP_CLAUSE_ORDER_REPRODUCIBLE (t))
313 : : (with the exception of loop construct but that lowers to
314 : : no schedule/dist_schedule clauses currently). */
315 : 7544 : if (!OMP_CLAUSE_ORDER_UNCONSTRAINED (t))
316 : 509921 : order_reproducible = true;
317 : : default:
318 : : break;
319 : : }
320 : :
321 : 125082 : if (fd->ordered == -1)
322 : 20 : fd->ordered = fd->collapse;
323 : :
324 : : /* For order(reproducible:concurrent) schedule ({dynamic,guided,runtime})
325 : : we have either the option to expensively remember at runtime how we've
326 : : distributed work from first loop and reuse that in following loops with
327 : : the same number of iterations and schedule, or just force static schedule.
328 : : OpenMP API calls etc. aren't allowed in order(concurrent) bodies so
329 : : users can't observe it easily anyway. */
330 : 125082 : if (order_reproducible)
331 : 7249 : fd->sched_kind = OMP_CLAUSE_SCHEDULE_STATIC;
332 : 125082 : if (fd->collapse > 1 || fd->tiling)
333 : 34018 : fd->loops = loops;
334 : : else
335 : 91064 : fd->loops = &fd->loop;
336 : :
337 : 125082 : if (fd->ordered && fd->collapse == 1 && loops != NULL)
338 : : {
339 : 184 : fd->loops = loops;
340 : 184 : iterv = NULL_TREE;
341 : 184 : countv = NULL_TREE;
342 : 184 : collapse_iter = &iterv;
343 : 184 : collapse_count = &countv;
344 : : }
345 : :
346 : : /* FIXME: for now map schedule(auto) to schedule(static).
347 : : There should be analysis to determine whether all iterations
348 : : are approximately the same amount of work (then schedule(static)
349 : : is best) or if it varies (then schedule(dynamic,N) is better). */
350 : 125082 : if (fd->sched_kind == OMP_CLAUSE_SCHEDULE_AUTO)
351 : : {
352 : 5747 : fd->sched_kind = OMP_CLAUSE_SCHEDULE_STATIC;
353 : 5747 : gcc_assert (fd->chunk_size == NULL);
354 : : }
355 : 125082 : gcc_assert ((fd->collapse == 1 && !fd->tiling) || collapse_iter != NULL);
356 : 125082 : if (taskloop)
357 : 9638 : fd->sched_kind = OMP_CLAUSE_SCHEDULE_RUNTIME;
358 : 125082 : if (fd->sched_kind == OMP_CLAUSE_SCHEDULE_RUNTIME)
359 : 16639 : gcc_assert (fd->chunk_size == NULL);
360 : 108443 : else if (fd->chunk_size == NULL)
361 : : {
362 : : /* We only need to compute a default chunk size for ordered
363 : : static loops and dynamic loops. */
364 : 85374 : if (fd->sched_kind != OMP_CLAUSE_SCHEDULE_STATIC
365 : 84815 : || fd->have_ordered)
366 : 1270 : fd->chunk_size = (fd->sched_kind == OMP_CLAUSE_SCHEDULE_STATIC)
367 : 1270 : ? integer_zero_node : integer_one_node;
368 : : }
369 : :
370 : 125082 : int cnt = fd->ordered ? fd->ordered : fd->collapse;
371 : 125082 : int single_nonrect = -1;
372 : 125082 : tree single_nonrect_count = NULL_TREE;
373 : 125082 : enum tree_code single_nonrect_cond_code = ERROR_MARK;
374 : 189027 : for (i = 1; i < cnt; i++)
375 : : {
376 : 64073 : tree n1 = gimple_omp_for_initial (for_stmt, i);
377 : 64073 : tree n2 = gimple_omp_for_final (for_stmt, i);
378 : 64073 : if (TREE_CODE (n1) == TREE_VEC)
379 : : {
380 : 1786 : if (fd->non_rect)
381 : : {
382 : : single_nonrect = -1;
383 : : break;
384 : : }
385 : 2152 : for (int j = i - 1; j >= 0; j--)
386 : 2152 : if (TREE_VEC_ELT (n1, 0) == gimple_omp_for_index (for_stmt, j))
387 : : {
388 : : single_nonrect = j;
389 : : break;
390 : : }
391 : 1664 : fd->non_rect = true;
392 : : }
393 : 62287 : else if (TREE_CODE (n2) == TREE_VEC)
394 : : {
395 : 531 : if (fd->non_rect)
396 : : {
397 : : single_nonrect = -1;
398 : : break;
399 : : }
400 : 681 : for (int j = i - 1; j >= 0; j--)
401 : 681 : if (TREE_VEC_ELT (n2, 0) == gimple_omp_for_index (for_stmt, j))
402 : : {
403 : : single_nonrect = j;
404 : : break;
405 : : }
406 : 525 : fd->non_rect = true;
407 : : }
408 : : }
409 : 314321 : for (i = 0; i < cnt; i++)
410 : : {
411 : 189239 : if (i == 0
412 : 125082 : && fd->collapse == 1
413 : 91236 : && !fd->tiling
414 : 91064 : && (fd->ordered == 0 || loops == NULL))
415 : 90880 : loop = &fd->loop;
416 : 98175 : else if (loops != NULL)
417 : 32574 : loop = loops + i;
418 : : else
419 : : loop = &dummy_loop;
420 : :
421 : 189239 : loop->v = gimple_omp_for_index (for_stmt, i);
422 : 189239 : gcc_assert (SSA_VAR_P (loop->v));
423 : 189239 : gcc_assert (TREE_CODE (TREE_TYPE (loop->v)) == INTEGER_TYPE
424 : : || TREE_CODE (TREE_TYPE (loop->v)) == BITINT_TYPE
425 : : || TREE_CODE (TREE_TYPE (loop->v)) == POINTER_TYPE);
426 : 189239 : var = TREE_CODE (loop->v) == SSA_NAME ? SSA_NAME_VAR (loop->v) : loop->v;
427 : 189239 : loop->n1 = gimple_omp_for_initial (for_stmt, i);
428 : 189239 : loop->m1 = NULL_TREE;
429 : 189239 : loop->m2 = NULL_TREE;
430 : 189239 : loop->outer = 0;
431 : 189239 : loop->non_rect_referenced = false;
432 : 189239 : if (TREE_CODE (loop->n1) == TREE_VEC)
433 : : {
434 : 2313 : for (int j = i - 1; j >= 0; j--)
435 : 2313 : if (TREE_VEC_ELT (loop->n1, 0) == gimple_omp_for_index (for_stmt, j))
436 : : {
437 : 1786 : loop->outer = i - j;
438 : 1786 : if (loops != NULL)
439 : 639 : loops[j].non_rect_referenced = true;
440 : 1786 : if (fd->first_nonrect == -1 || fd->first_nonrect > j)
441 : 1664 : fd->first_nonrect = j;
442 : : break;
443 : : }
444 : 1786 : gcc_assert (loop->outer);
445 : 1786 : loop->m1 = TREE_VEC_ELT (loop->n1, 1);
446 : 1786 : loop->n1 = TREE_VEC_ELT (loop->n1, 2);
447 : 1786 : fd->non_rect = true;
448 : 1786 : fd->last_nonrect = i;
449 : : }
450 : :
451 : 189239 : loop->cond_code = gimple_omp_for_cond (for_stmt, i);
452 : 189239 : loop->n2 = gimple_omp_for_final (for_stmt, i);
453 : 189239 : gcc_assert (loop->cond_code != NE_EXPR
454 : : || (gimple_omp_for_kind (for_stmt)
455 : : != GF_OMP_FOR_KIND_OACC_LOOP));
456 : 189239 : if (TREE_CODE (loop->n2) == TREE_VEC)
457 : : {
458 : 1273 : if (loop->outer)
459 : 742 : gcc_assert (TREE_VEC_ELT (loop->n2, 0)
460 : : == gimple_omp_for_index (for_stmt, i - loop->outer));
461 : : else
462 : 690 : for (int j = i - 1; j >= 0; j--)
463 : 690 : if (TREE_VEC_ELT (loop->n2, 0) == gimple_omp_for_index (for_stmt, j))
464 : : {
465 : 531 : loop->outer = i - j;
466 : 531 : if (loops != NULL)
467 : 241 : loops[j].non_rect_referenced = true;
468 : 531 : if (fd->first_nonrect == -1 || fd->first_nonrect > j)
469 : 525 : fd->first_nonrect = j;
470 : : break;
471 : : }
472 : 1273 : gcc_assert (loop->outer);
473 : 1273 : loop->m2 = TREE_VEC_ELT (loop->n2, 1);
474 : 1273 : loop->n2 = TREE_VEC_ELT (loop->n2, 2);
475 : 1273 : fd->non_rect = true;
476 : 1273 : fd->last_nonrect = i;
477 : : }
478 : :
479 : 189239 : t = gimple_omp_for_incr (for_stmt, i);
480 : 189239 : gcc_assert (TREE_OPERAND (t, 0) == var);
481 : 189239 : loop->step = omp_get_for_step_from_incr (loc, t);
482 : :
483 : 189239 : omp_adjust_for_condition (loc, &loop->cond_code, &loop->n2, loop->v,
484 : : loop->step);
485 : :
486 : 189239 : if (simd
487 : 149496 : || (fd->sched_kind == OMP_CLAUSE_SCHEDULE_STATIC
488 : 110031 : && !fd->have_ordered))
489 : : {
490 : 148063 : if (fd->collapse == 1 && !fd->tiling)
491 : 73253 : iter_type = TREE_TYPE (loop->v);
492 : 74810 : else if (i == 0
493 : 74810 : || TYPE_PRECISION (iter_type)
494 : 48313 : < TYPE_PRECISION (TREE_TYPE (loop->v)))
495 : : {
496 : 37374 : if (TREE_CODE (iter_type) == BITINT_TYPE
497 : 37374 : || TREE_CODE (TREE_TYPE (loop->v)) == BITINT_TYPE)
498 : 2 : iter_type
499 : 2 : = build_bitint_type (TYPE_PRECISION (TREE_TYPE (loop->v)),
500 : : 1);
501 : : else
502 : 37372 : iter_type
503 : : = build_nonstandard_integer_type
504 : 37372 : (TYPE_PRECISION (TREE_TYPE (loop->v)), 1);
505 : : }
506 : : }
507 : 41176 : else if (iter_type != long_long_unsigned_type_node)
508 : : {
509 : 37716 : if (POINTER_TYPE_P (TREE_TYPE (loop->v)))
510 : : iter_type = long_long_unsigned_type_node;
511 : 35754 : else if (TYPE_UNSIGNED (TREE_TYPE (loop->v))
512 : 35754 : && TYPE_PRECISION (TREE_TYPE (loop->v))
513 : 6284 : >= TYPE_PRECISION (iter_type))
514 : : {
515 : 2318 : tree n;
516 : :
517 : 2318 : if (loop->cond_code == LT_EXPR)
518 : 450 : n = fold_build2_loc (loc, PLUS_EXPR, TREE_TYPE (loop->v),
519 : : loop->n2, loop->step);
520 : : else
521 : 1868 : n = loop->n1;
522 : 2318 : if (loop->m1
523 : 2318 : || loop->m2
524 : 2318 : || TREE_CODE (n) != INTEGER_CST
525 : 4282 : || tree_int_cst_lt (TYPE_MAX_VALUE (iter_type), n))
526 : 2242 : iter_type = long_long_unsigned_type_node;
527 : : }
528 : 33436 : else if (TYPE_PRECISION (TREE_TYPE (loop->v))
529 : 33436 : > TYPE_PRECISION (iter_type))
530 : : {
531 : 0 : tree n1, n2;
532 : :
533 : 0 : if (loop->cond_code == LT_EXPR)
534 : : {
535 : 0 : n1 = loop->n1;
536 : 0 : n2 = fold_build2_loc (loc, PLUS_EXPR, TREE_TYPE (loop->v),
537 : : loop->n2, loop->step);
538 : : }
539 : : else
540 : : {
541 : 0 : n1 = fold_build2_loc (loc, MINUS_EXPR, TREE_TYPE (loop->v),
542 : : loop->n2, loop->step);
543 : 0 : n2 = loop->n1;
544 : : }
545 : 0 : if (loop->m1
546 : 0 : || loop->m2
547 : 0 : || TREE_CODE (n1) != INTEGER_CST
548 : 0 : || TREE_CODE (n2) != INTEGER_CST
549 : 0 : || !tree_int_cst_lt (TYPE_MIN_VALUE (iter_type), n1)
550 : 0 : || !tree_int_cst_lt (n2, TYPE_MAX_VALUE (iter_type)))
551 : 0 : iter_type = long_long_unsigned_type_node;
552 : : }
553 : : }
554 : :
555 : 189239 : if (i >= fd->collapse)
556 : 1802 : continue;
557 : :
558 : 187437 : if (collapse_count && *collapse_count == NULL)
559 : : {
560 : 31749 : if (count && integer_zerop (count))
561 : 2240 : continue;
562 : 29509 : tree n1first = NULL_TREE, n2first = NULL_TREE;
563 : 29509 : tree n1last = NULL_TREE, n2last = NULL_TREE;
564 : 29509 : tree ostep = NULL_TREE;
565 : 29509 : if (loop->m1 || loop->m2)
566 : : {
567 : 911 : if (count == NULL_TREE)
568 : 645 : continue;
569 : 266 : if (single_nonrect == -1
570 : 247 : || (loop->m1 && TREE_CODE (loop->m1) != INTEGER_CST)
571 : 206 : || (loop->m2 && TREE_CODE (loop->m2) != INTEGER_CST)
572 : 190 : || TREE_CODE (loop->n1) != INTEGER_CST
573 : 186 : || TREE_CODE (loop->n2) != INTEGER_CST
574 : 182 : || TREE_CODE (loop->step) != INTEGER_CST)
575 : : {
576 : 84 : count = NULL_TREE;
577 : 84 : continue;
578 : : }
579 : 182 : tree var = gimple_omp_for_initial (for_stmt, single_nonrect);
580 : 182 : tree itype = TREE_TYPE (var);
581 : 182 : tree first = gimple_omp_for_initial (for_stmt, single_nonrect);
582 : 182 : t = gimple_omp_for_incr (for_stmt, single_nonrect);
583 : 182 : ostep = omp_get_for_step_from_incr (loc, t);
584 : 182 : t = fold_binary (MINUS_EXPR, long_long_unsigned_type_node,
585 : : single_nonrect_count,
586 : : build_one_cst (long_long_unsigned_type_node));
587 : 182 : t = fold_convert (itype, t);
588 : 182 : first = fold_convert (itype, first);
589 : 182 : ostep = fold_convert (itype, ostep);
590 : 182 : tree last = fold_binary (PLUS_EXPR, itype, first,
591 : : fold_binary (MULT_EXPR, itype, t,
592 : : ostep));
593 : 182 : if (TREE_CODE (first) != INTEGER_CST
594 : 182 : || TREE_CODE (last) != INTEGER_CST)
595 : : {
596 : 0 : count = NULL_TREE;
597 : 0 : continue;
598 : : }
599 : 182 : if (loop->m1)
600 : : {
601 : 115 : tree m1 = fold_convert (itype, loop->m1);
602 : 115 : tree n1 = fold_convert (itype, loop->n1);
603 : 115 : n1first = fold_binary (PLUS_EXPR, itype,
604 : : fold_binary (MULT_EXPR, itype,
605 : : first, m1), n1);
606 : 115 : n1last = fold_binary (PLUS_EXPR, itype,
607 : : fold_binary (MULT_EXPR, itype,
608 : : last, m1), n1);
609 : : }
610 : : else
611 : 67 : n1first = n1last = loop->n1;
612 : 182 : if (loop->m2)
613 : : {
614 : 136 : tree n2 = fold_convert (itype, loop->n2);
615 : 136 : tree m2 = fold_convert (itype, loop->m2);
616 : 136 : n2first = fold_binary (PLUS_EXPR, itype,
617 : : fold_binary (MULT_EXPR, itype,
618 : : first, m2), n2);
619 : 136 : n2last = fold_binary (PLUS_EXPR, itype,
620 : : fold_binary (MULT_EXPR, itype,
621 : : last, m2), n2);
622 : : }
623 : : else
624 : 46 : n2first = n2last = loop->n2;
625 : 182 : n1first = fold_convert (TREE_TYPE (loop->v), n1first);
626 : 182 : n2first = fold_convert (TREE_TYPE (loop->v), n2first);
627 : 182 : n1last = fold_convert (TREE_TYPE (loop->v), n1last);
628 : 182 : n2last = fold_convert (TREE_TYPE (loop->v), n2last);
629 : 182 : t = fold_binary (loop->cond_code, boolean_type_node,
630 : : n1first, n2first);
631 : 182 : tree t2 = fold_binary (loop->cond_code, boolean_type_node,
632 : : n1last, n2last);
633 : 182 : if (t && t2 && integer_nonzerop (t) && integer_nonzerop (t2))
634 : : /* All outer loop iterators have at least one inner loop
635 : : iteration. Try to compute the count at compile time. */
636 : : t = NULL_TREE;
637 : 103 : else if (t && t2 && integer_zerop (t) && integer_zerop (t2))
638 : : /* No iterations of the inner loop. count will be set to
639 : : zero cst below. */;
640 : 95 : else if (TYPE_UNSIGNED (itype)
641 : 95 : || t == NULL_TREE
642 : 95 : || t2 == NULL_TREE
643 : 95 : || TREE_CODE (t) != INTEGER_CST
644 : 190 : || TREE_CODE (t2) != INTEGER_CST)
645 : : {
646 : : /* Punt (for now). */
647 : 0 : count = NULL_TREE;
648 : 0 : continue;
649 : : }
650 : : else
651 : : {
652 : : /* Some iterations of the outer loop have zero iterations
653 : : of the inner loop, while others have at least one.
654 : : In this case, we need to adjust one of those outer
655 : : loop bounds. If ADJ_FIRST, we need to adjust outer n1
656 : : (first), otherwise outer n2 (last). */
657 : 95 : bool adj_first = integer_zerop (t);
658 : 95 : tree n1 = fold_convert (itype, loop->n1);
659 : 95 : tree n2 = fold_convert (itype, loop->n2);
660 : 95 : tree m1 = loop->m1 ? fold_convert (itype, loop->m1)
661 : 25 : : build_zero_cst (itype);
662 : 95 : tree m2 = loop->m2 ? fold_convert (itype, loop->m2)
663 : 11 : : build_zero_cst (itype);
664 : 95 : t = fold_binary (MINUS_EXPR, itype, n1, n2);
665 : 95 : t2 = fold_binary (MINUS_EXPR, itype, m2, m1);
666 : 95 : t = fold_binary (TRUNC_DIV_EXPR, itype, t, t2);
667 : 95 : t2 = fold_binary (MINUS_EXPR, itype, t, first);
668 : 95 : t2 = fold_binary (TRUNC_MOD_EXPR, itype, t2, ostep);
669 : 95 : t = fold_binary (MINUS_EXPR, itype, t, t2);
670 : 95 : tree n1cur
671 : 95 : = fold_binary (PLUS_EXPR, itype, n1,
672 : : fold_binary (MULT_EXPR, itype, m1, t));
673 : 95 : tree n2cur
674 : 95 : = fold_binary (PLUS_EXPR, itype, n2,
675 : : fold_binary (MULT_EXPR, itype, m2, t));
676 : 95 : t2 = fold_binary (loop->cond_code, boolean_type_node,
677 : : n1cur, n2cur);
678 : 95 : tree t3 = fold_binary (MULT_EXPR, itype, m1, ostep);
679 : 95 : tree t4 = fold_binary (MULT_EXPR, itype, m2, ostep);
680 : 95 : tree diff;
681 : 95 : if (adj_first)
682 : : {
683 : 67 : tree new_first;
684 : 67 : if (integer_nonzerop (t2))
685 : : {
686 : 6 : new_first = t;
687 : 6 : n1first = n1cur;
688 : 6 : n2first = n2cur;
689 : 6 : if (flag_checking)
690 : : {
691 : 6 : t3 = fold_binary (MINUS_EXPR, itype, n1cur, t3);
692 : 6 : t4 = fold_binary (MINUS_EXPR, itype, n2cur, t4);
693 : 6 : t3 = fold_binary (loop->cond_code,
694 : : boolean_type_node, t3, t4);
695 : 6 : gcc_assert (integer_zerop (t3));
696 : : }
697 : : }
698 : : else
699 : : {
700 : 61 : t3 = fold_binary (PLUS_EXPR, itype, n1cur, t3);
701 : 61 : t4 = fold_binary (PLUS_EXPR, itype, n2cur, t4);
702 : 61 : new_first = fold_binary (PLUS_EXPR, itype, t, ostep);
703 : 61 : n1first = t3;
704 : 61 : n2first = t4;
705 : 61 : if (flag_checking)
706 : : {
707 : 61 : t3 = fold_binary (loop->cond_code,
708 : : boolean_type_node, t3, t4);
709 : 61 : gcc_assert (integer_nonzerop (t3));
710 : : }
711 : : }
712 : 67 : diff = fold_binary (MINUS_EXPR, itype, new_first, first);
713 : 67 : first = new_first;
714 : 67 : fd->adjn1 = first;
715 : : }
716 : : else
717 : : {
718 : 28 : tree new_last;
719 : 28 : if (integer_zerop (t2))
720 : : {
721 : 11 : t3 = fold_binary (MINUS_EXPR, itype, n1cur, t3);
722 : 11 : t4 = fold_binary (MINUS_EXPR, itype, n2cur, t4);
723 : 11 : new_last = fold_binary (MINUS_EXPR, itype, t, ostep);
724 : 11 : n1last = t3;
725 : 11 : n2last = t4;
726 : 11 : if (flag_checking)
727 : : {
728 : 11 : t3 = fold_binary (loop->cond_code,
729 : : boolean_type_node, t3, t4);
730 : 11 : gcc_assert (integer_nonzerop (t3));
731 : : }
732 : : }
733 : : else
734 : : {
735 : 17 : new_last = t;
736 : 17 : n1last = n1cur;
737 : 17 : n2last = n2cur;
738 : 17 : if (flag_checking)
739 : : {
740 : 17 : t3 = fold_binary (PLUS_EXPR, itype, n1cur, t3);
741 : 17 : t4 = fold_binary (PLUS_EXPR, itype, n2cur, t4);
742 : 17 : t3 = fold_binary (loop->cond_code,
743 : : boolean_type_node, t3, t4);
744 : 17 : gcc_assert (integer_zerop (t3));
745 : : }
746 : : }
747 : 28 : diff = fold_binary (MINUS_EXPR, itype, last, new_last);
748 : : }
749 : 95 : if (TYPE_UNSIGNED (itype)
750 : 95 : && single_nonrect_cond_code == GT_EXPR)
751 : 0 : diff = fold_binary (TRUNC_DIV_EXPR, itype,
752 : : fold_unary (NEGATE_EXPR, itype, diff),
753 : : fold_unary (NEGATE_EXPR, itype,
754 : : ostep));
755 : : else
756 : 95 : diff = fold_binary (TRUNC_DIV_EXPR, itype, diff, ostep);
757 : 95 : diff = fold_convert (long_long_unsigned_type_node, diff);
758 : 95 : single_nonrect_count
759 : 95 : = fold_binary (MINUS_EXPR, long_long_unsigned_type_node,
760 : : single_nonrect_count, diff);
761 : 95 : t = NULL_TREE;
762 : : }
763 : : }
764 : : else
765 : 28598 : t = fold_binary (loop->cond_code, boolean_type_node,
766 : : fold_convert (TREE_TYPE (loop->v), loop->n1),
767 : : fold_convert (TREE_TYPE (loop->v), loop->n2));
768 : 28701 : if (t && integer_zerop (t))
769 : 2248 : count = build_zero_cst (long_long_unsigned_type_node);
770 : 26532 : else if ((i == 0 || count != NULL_TREE)
771 : 16843 : && (TREE_CODE (TREE_TYPE (loop->v)) == INTEGER_TYPE
772 : 1202 : || TREE_CODE (TREE_TYPE (loop->v)) == BITINT_TYPE)
773 : 15642 : && TREE_CONSTANT (loop->n1)
774 : 11641 : && TREE_CONSTANT (loop->n2)
775 : 37188 : && TREE_CODE (loop->step) == INTEGER_CST)
776 : : {
777 : 10634 : tree itype = TREE_TYPE (loop->v);
778 : :
779 : 10634 : if (POINTER_TYPE_P (itype))
780 : 0 : itype = signed_type_for (itype);
781 : 13002 : t = build_int_cst (itype, (loop->cond_code == LT_EXPR ? -1 : 1));
782 : 10634 : t = fold_build2 (PLUS_EXPR, itype,
783 : : fold_convert (itype, loop->step), t);
784 : 10634 : tree n1 = loop->n1;
785 : 10634 : tree n2 = loop->n2;
786 : 10634 : if (loop->m1 || loop->m2)
787 : : {
788 : 174 : gcc_assert (single_nonrect != -1);
789 : : n1 = n1first;
790 : : n2 = n2first;
791 : : }
792 : 10634 : t = fold_build2 (PLUS_EXPR, itype, t, fold_convert (itype, n2));
793 : 10634 : t = fold_build2 (MINUS_EXPR, itype, t, fold_convert (itype, n1));
794 : 10634 : tree step = fold_convert_loc (loc, itype, loop->step);
795 : 10634 : if (TYPE_UNSIGNED (itype) && loop->cond_code == GT_EXPR)
796 : 2254 : t = fold_build2 (TRUNC_DIV_EXPR, itype,
797 : : fold_build1 (NEGATE_EXPR, itype, t),
798 : : fold_build1 (NEGATE_EXPR, itype, step));
799 : : else
800 : 8380 : t = fold_build2 (TRUNC_DIV_EXPR, itype, t, step);
801 : 10634 : tree llutype = long_long_unsigned_type_node;
802 : 10634 : t = fold_convert (llutype, t);
803 : 10634 : if (loop->m1 || loop->m2)
804 : : {
805 : : /* t is number of iterations of inner loop at either first
806 : : or last value of the outer iterator (the one with fewer
807 : : iterations).
808 : : Compute t2 = ((m2 - m1) * ostep) / step
809 : : and niters = outer_count * t
810 : : + t2 * ((outer_count - 1) * outer_count / 2)
811 : : */
812 : 174 : tree m1 = loop->m1 ? loop->m1 : integer_zero_node;
813 : 174 : tree m2 = loop->m2 ? loop->m2 : integer_zero_node;
814 : 174 : m1 = fold_convert (itype, m1);
815 : 174 : m2 = fold_convert (itype, m2);
816 : 174 : tree t2 = fold_build2 (MINUS_EXPR, itype, m2, m1);
817 : 174 : t2 = fold_build2 (MULT_EXPR, itype, t2, ostep);
818 : 174 : if (TYPE_UNSIGNED (itype) && loop->cond_code == GT_EXPR)
819 : 0 : t2 = fold_build2 (TRUNC_DIV_EXPR, itype,
820 : : fold_build1 (NEGATE_EXPR, itype, t2),
821 : : fold_build1 (NEGATE_EXPR, itype, step));
822 : : else
823 : 174 : t2 = fold_build2 (TRUNC_DIV_EXPR, itype, t2, step);
824 : 174 : t2 = fold_convert (llutype, t2);
825 : 174 : fd->first_inner_iterations = t;
826 : 174 : fd->factor = t2;
827 : 174 : t = fold_build2 (MULT_EXPR, llutype, t,
828 : : single_nonrect_count);
829 : 174 : tree t3 = fold_build2 (MINUS_EXPR, llutype,
830 : : single_nonrect_count,
831 : : build_one_cst (llutype));
832 : 174 : t3 = fold_build2 (MULT_EXPR, llutype, t3,
833 : : single_nonrect_count);
834 : 174 : t3 = fold_build2 (TRUNC_DIV_EXPR, llutype, t3,
835 : : build_int_cst (llutype, 2));
836 : 174 : t2 = fold_build2 (MULT_EXPR, llutype, t2, t3);
837 : 174 : t = fold_build2 (PLUS_EXPR, llutype, t, t2);
838 : : }
839 : 10634 : if (i == single_nonrect)
840 : : {
841 : 247 : if (integer_zerop (t) || TREE_CODE (t) != INTEGER_CST)
842 : : count = t;
843 : : else
844 : : {
845 : 247 : single_nonrect_count = t;
846 : 247 : single_nonrect_cond_code = loop->cond_code;
847 : 247 : if (count == NULL_TREE)
848 : 246 : count = build_one_cst (llutype);
849 : : }
850 : : }
851 : 10387 : else if (count != NULL_TREE)
852 : 4144 : count = fold_build2 (MULT_EXPR, llutype, count, t);
853 : : else
854 : : count = t;
855 : 10634 : if (TREE_CODE (count) != INTEGER_CST)
856 : 0 : count = NULL_TREE;
857 : : }
858 : 15898 : else if (count && !integer_zerop (count))
859 : : count = NULL_TREE;
860 : : }
861 : : }
862 : :
863 : 125082 : if (count
864 : 125082 : && !simd
865 : 3989 : && (fd->sched_kind != OMP_CLAUSE_SCHEDULE_STATIC
866 : 3230 : || fd->have_ordered))
867 : : {
868 : 855 : if (!tree_int_cst_lt (count, TYPE_MAX_VALUE (long_integer_type_node)))
869 : 0 : iter_type = long_long_unsigned_type_node;
870 : : else
871 : 855 : iter_type = long_integer_type_node;
872 : : }
873 : 124227 : else if (collapse_iter && *collapse_iter != NULL)
874 : 22753 : iter_type = TREE_TYPE (*collapse_iter);
875 : 125082 : fd->iter_type = iter_type;
876 : 125082 : if (collapse_iter && *collapse_iter == NULL)
877 : 11453 : *collapse_iter = create_tmp_var (iter_type, ".iter");
878 : 125082 : if (collapse_count && *collapse_count == NULL)
879 : : {
880 : 11453 : if (count)
881 : : {
882 : 5160 : *collapse_count = fold_convert_loc (loc, iter_type, count);
883 : 5160 : if (fd->first_inner_iterations && fd->factor)
884 : : {
885 : 174 : t = make_tree_vec (4);
886 : 174 : TREE_VEC_ELT (t, 0) = *collapse_count;
887 : 174 : TREE_VEC_ELT (t, 1) = fd->first_inner_iterations;
888 : 174 : TREE_VEC_ELT (t, 2) = fd->factor;
889 : 174 : TREE_VEC_ELT (t, 3) = fd->adjn1;
890 : 174 : *collapse_count = t;
891 : : }
892 : : }
893 : : else
894 : 6293 : *collapse_count = create_tmp_var (iter_type, ".count");
895 : : }
896 : :
897 : 125082 : if (fd->collapse > 1 || fd->tiling || (fd->ordered && loops))
898 : : {
899 : 34202 : fd->loop.v = *collapse_iter;
900 : 34202 : fd->loop.n1 = build_int_cst (TREE_TYPE (fd->loop.v), 0);
901 : 34202 : fd->loop.n2 = *collapse_count;
902 : 34202 : if (TREE_CODE (fd->loop.n2) == TREE_VEC)
903 : : {
904 : 377 : gcc_assert (fd->non_rect);
905 : 377 : fd->first_inner_iterations = TREE_VEC_ELT (fd->loop.n2, 1);
906 : 377 : fd->factor = TREE_VEC_ELT (fd->loop.n2, 2);
907 : 377 : fd->adjn1 = TREE_VEC_ELT (fd->loop.n2, 3);
908 : 377 : fd->loop.n2 = TREE_VEC_ELT (fd->loop.n2, 0);
909 : : }
910 : 34202 : fd->loop.step = build_int_cst (TREE_TYPE (fd->loop.v), 1);
911 : 34202 : fd->loop.m1 = NULL_TREE;
912 : 34202 : fd->loop.m2 = NULL_TREE;
913 : 34202 : fd->loop.outer = 0;
914 : 34202 : fd->loop.cond_code = LT_EXPR;
915 : : }
916 : 90668 : else if (loops)
917 : 34147 : loops[0] = fd->loop;
918 : 125082 : }
919 : :
920 : : /* Build a call to GOMP_barrier. */
921 : :
922 : : gimple *
923 : 4701 : omp_build_barrier (tree lhs)
924 : : {
925 : 9344 : tree fndecl = builtin_decl_explicit (lhs ? BUILT_IN_GOMP_BARRIER_CANCEL
926 : : : BUILT_IN_GOMP_BARRIER);
927 : 4701 : gcall *g = gimple_build_call (fndecl, 0);
928 : 4701 : if (lhs)
929 : 58 : gimple_call_set_lhs (g, lhs);
930 : 4701 : return g;
931 : : }
932 : :
933 : : /* Find OMP_FOR resp. OMP_SIMD with non-NULL OMP_FOR_INIT. Also, fill in pdata
934 : : array, pdata[0] non-NULL if there is anything non-trivial in between,
935 : : pdata[1] is address of OMP_PARALLEL in between if any, pdata[2] is address
936 : : of OMP_FOR in between if any and pdata[3] is address of the inner
937 : : OMP_FOR/OMP_SIMD. */
938 : :
939 : : tree
940 : 109277 : find_combined_omp_for (tree *tp, int *walk_subtrees, void *data)
941 : : {
942 : 109277 : tree **pdata = (tree **) data;
943 : 109277 : *walk_subtrees = 0;
944 : 109277 : switch (TREE_CODE (*tp))
945 : : {
946 : 11806 : case OMP_FOR:
947 : 11806 : if (OMP_FOR_INIT (*tp) != NULL_TREE)
948 : : {
949 : 5996 : pdata[3] = tp;
950 : 5996 : return *tp;
951 : : }
952 : 5810 : pdata[2] = tp;
953 : 5810 : *walk_subtrees = 1;
954 : 5810 : break;
955 : 15279 : case OMP_SIMD:
956 : 15279 : if (OMP_FOR_INIT (*tp) != NULL_TREE)
957 : : {
958 : 15279 : pdata[3] = tp;
959 : 15279 : return *tp;
960 : : }
961 : : break;
962 : 46271 : case BIND_EXPR:
963 : 46271 : if (BIND_EXPR_VARS (*tp)
964 : 46271 : || (BIND_EXPR_BLOCK (*tp)
965 : 40735 : && BLOCK_VARS (BIND_EXPR_BLOCK (*tp))))
966 : 5098 : pdata[0] = tp;
967 : 46271 : *walk_subtrees = 1;
968 : 46271 : break;
969 : 8516 : case STATEMENT_LIST:
970 : 8516 : if (!tsi_one_before_end_p (tsi_start (*tp)))
971 : 242 : pdata[0] = tp;
972 : 8516 : *walk_subtrees = 1;
973 : 8516 : break;
974 : 218 : case TRY_FINALLY_EXPR:
975 : 218 : case CLEANUP_POINT_EXPR:
976 : 218 : pdata[0] = tp;
977 : 218 : *walk_subtrees = 1;
978 : 218 : break;
979 : 11834 : case OMP_PARALLEL:
980 : 11834 : pdata[1] = tp;
981 : 11834 : *walk_subtrees = 1;
982 : 11834 : break;
983 : : default:
984 : : break;
985 : : }
986 : : return NULL_TREE;
987 : : }
988 : :
989 : : /* Return maximum possible vectorization factor for the target, or for
990 : : the OpenMP offload target if one exists. */
991 : :
992 : : poly_uint64
993 : 30128 : omp_max_vf (bool offload)
994 : : {
995 : 30128 : if (!optimize
996 : 28801 : || optimize_debug
997 : 28801 : || !flag_tree_loop_optimize
998 : 28800 : || (!flag_tree_loop_vectorize
999 : 577 : && OPTION_SET_P (flag_tree_loop_vectorize)))
1000 : 1332 : return 1;
1001 : :
1002 : 28796 : if (ENABLE_OFFLOADING && offload)
1003 : : {
1004 : : for (const char *c = getenv ("OFFLOAD_TARGET_NAMES"); c;)
1005 : : {
1006 : : if (startswith (c, "amdgcn"))
1007 : : return ordered_max (poly_uint64 (64), omp_max_vf (false));
1008 : : else if ((c = strchr (c, ':')))
1009 : : c++;
1010 : : }
1011 : : /* Otherwise, fall through to host VF. */
1012 : : }
1013 : :
1014 : 28796 : auto_vector_modes modes;
1015 : 28796 : targetm.vectorize.autovectorize_vector_modes (&modes, true);
1016 : 28796 : if (!modes.is_empty ())
1017 : : {
1018 : : poly_uint64 vf = 0;
1019 : 120037 : for (unsigned int i = 0; i < modes.length (); ++i)
1020 : : /* The returned modes use the smallest element size (and thus
1021 : : the largest nunits) for the vectorization approach that they
1022 : : represent. */
1023 : 182490 : vf = ordered_max (vf, GET_MODE_NUNITS (modes[i]));
1024 : 28792 : return vf;
1025 : : }
1026 : :
1027 : 4 : machine_mode vqimode = targetm.vectorize.preferred_simd_mode (QImode);
1028 : 4 : if (GET_MODE_CLASS (vqimode) == MODE_VECTOR_INT)
1029 : 0 : return GET_MODE_NUNITS (vqimode);
1030 : :
1031 : 4 : return 1;
1032 : 28796 : }
1033 : :
1034 : : /* Return maximum SIMT width if offloading may target SIMT hardware. */
1035 : :
1036 : : int
1037 : 3627 : omp_max_simt_vf (void)
1038 : : {
1039 : 3627 : if (!optimize)
1040 : : return 0;
1041 : : if (ENABLE_OFFLOADING)
1042 : : for (const char *c = getenv ("OFFLOAD_TARGET_NAMES"); c;)
1043 : : {
1044 : : if (startswith (c, "nvptx"))
1045 : : return 32;
1046 : : else if ((c = strchr (c, ':')))
1047 : : c++;
1048 : : }
1049 : : return 0;
1050 : : }
1051 : :
1052 : : /* Store the construct selectors as tree codes from last to first.
1053 : : CTX is a list of trait selectors, nconstructs must be equal to its
1054 : : length, and the array CONSTRUCTS holds the output. */
1055 : :
1056 : : void
1057 : 1006 : omp_construct_traits_to_codes (tree ctx, int nconstructs,
1058 : : enum tree_code *constructs)
1059 : : {
1060 : 1006 : int i = nconstructs - 1;
1061 : :
1062 : : /* Order must match the OMP_TRAIT_CONSTRUCT_* enumerators in
1063 : : enum omp_ts_code. */
1064 : 1006 : static enum tree_code code_map[]
1065 : : = { OMP_TARGET, OMP_TEAMS, OMP_PARALLEL, OMP_FOR, OMP_SIMD, OMP_DISPATCH };
1066 : :
1067 : 2317 : for (tree ts = ctx; ts; ts = TREE_CHAIN (ts), i--)
1068 : : {
1069 : 1311 : enum omp_ts_code sel = OMP_TS_CODE (ts);
1070 : 1311 : int j = (int)sel - (int)OMP_TRAIT_CONSTRUCT_TARGET;
1071 : 1311 : gcc_assert (j >= 0 && (unsigned int) j < ARRAY_SIZE (code_map));
1072 : 1311 : constructs[i] = code_map[j];
1073 : : }
1074 : 1006 : gcc_assert (i == -1);
1075 : 1006 : }
1076 : :
1077 : : /* Return true if PROP is possibly present in one of the offloading target's
1078 : : OpenMP contexts. The format of PROPS string is always offloading target's
1079 : : name terminated by '\0', followed by properties for that offloading
1080 : : target separated by '\0' and terminated by another '\0'. The strings
1081 : : are created from omp-device-properties installed files of all configured
1082 : : offloading targets. */
1083 : :
1084 : : static bool
1085 : 0 : omp_offload_device_kind_arch_isa (const char *props, const char *prop)
1086 : : {
1087 : 0 : const char *names = getenv ("OFFLOAD_TARGET_NAMES");
1088 : 0 : if (names == NULL || *names == '\0')
1089 : : return false;
1090 : 0 : while (*props != '\0')
1091 : : {
1092 : 0 : size_t name_len = strlen (props);
1093 : 0 : bool matches = false;
1094 : 0 : for (const char *c = names; c; )
1095 : : {
1096 : 0 : if (strncmp (props, c, name_len) == 0
1097 : 0 : && (c[name_len] == '\0'
1098 : : || c[name_len] == ':'
1099 : : || c[name_len] == '='))
1100 : : {
1101 : : matches = true;
1102 : : break;
1103 : : }
1104 : 0 : else if ((c = strchr (c, ':')))
1105 : 0 : c++;
1106 : : }
1107 : 0 : props = props + name_len + 1;
1108 : 0 : while (*props != '\0')
1109 : : {
1110 : 0 : if (matches && strcmp (props, prop) == 0)
1111 : : return true;
1112 : 0 : props = strchr (props, '\0') + 1;
1113 : : }
1114 : 0 : props++;
1115 : : }
1116 : : return false;
1117 : : }
1118 : :
1119 : : /* Return true if the current code location is or might be offloaded.
1120 : : Return true in declare target functions, or when nested in a target
1121 : : region or when unsure, return false otherwise. */
1122 : :
1123 : : static bool
1124 : 0 : omp_maybe_offloaded (void)
1125 : : {
1126 : 0 : if (!ENABLE_OFFLOADING)
1127 : 0 : return false;
1128 : : const char *names = getenv ("OFFLOAD_TARGET_NAMES");
1129 : : if (names == NULL || *names == '\0')
1130 : : return false;
1131 : :
1132 : : if (symtab->state == PARSING)
1133 : : /* Maybe. */
1134 : : return true;
1135 : : if (cfun && cfun->after_inlining)
1136 : : return false;
1137 : : if (current_function_decl
1138 : : && lookup_attribute ("omp declare target",
1139 : : DECL_ATTRIBUTES (current_function_decl)))
1140 : : return true;
1141 : : if (cfun && (cfun->curr_properties & PROP_gimple_any) == 0)
1142 : : {
1143 : : enum tree_code construct = OMP_TARGET;
1144 : : if (omp_construct_selector_matches (&construct, 1, NULL))
1145 : : return true;
1146 : : }
1147 : : return false;
1148 : : }
1149 : :
1150 : : /* Lookup tables for context selectors. */
1151 : : const char *omp_tss_map[] =
1152 : : {
1153 : : "construct",
1154 : : "device",
1155 : : "target_device",
1156 : : "implementation",
1157 : : "user",
1158 : : NULL
1159 : : };
1160 : :
1161 : : /* Arrays of property candidates must be null-terminated. */
1162 : : static const char *const kind_properties[] =
1163 : : { "host", "nohost", "cpu", "gpu", "fpga", "any", NULL };
1164 : : static const char *const vendor_properties[] =
1165 : : { "amd", "arm", "bsc", "cray", "fujitsu", "gnu", "hpe", "ibm", "intel",
1166 : : "llvm", "nec", "nvidia", "pgi", "ti", "unknown", NULL };
1167 : : static const char *const extension_properties[] =
1168 : : { NULL };
1169 : : static const char *const atomic_default_mem_order_properties[] =
1170 : : { "seq_cst", "relaxed", "acq_rel", "acquire", "release", NULL };
1171 : :
1172 : : struct omp_ts_info omp_ts_map[] =
1173 : : {
1174 : : { "kind",
1175 : : (1 << OMP_TRAIT_SET_DEVICE) | (1 << OMP_TRAIT_SET_TARGET_DEVICE),
1176 : : OMP_TRAIT_PROPERTY_NAME_LIST, false,
1177 : : kind_properties
1178 : : },
1179 : : { "isa",
1180 : : (1 << OMP_TRAIT_SET_DEVICE) | (1 << OMP_TRAIT_SET_TARGET_DEVICE),
1181 : : OMP_TRAIT_PROPERTY_NAME_LIST, false,
1182 : : NULL
1183 : : },
1184 : : { "arch",
1185 : : (1 << OMP_TRAIT_SET_DEVICE) | (1 << OMP_TRAIT_SET_TARGET_DEVICE),
1186 : : OMP_TRAIT_PROPERTY_NAME_LIST, false,
1187 : : NULL
1188 : : },
1189 : : { "device_num",
1190 : : (1 << OMP_TRAIT_SET_TARGET_DEVICE),
1191 : : OMP_TRAIT_PROPERTY_DEV_NUM_EXPR, false,
1192 : : NULL
1193 : : },
1194 : : { "vendor",
1195 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1196 : : OMP_TRAIT_PROPERTY_NAME_LIST, true,
1197 : : vendor_properties,
1198 : : },
1199 : : { "extension",
1200 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1201 : : OMP_TRAIT_PROPERTY_NAME_LIST, true,
1202 : : extension_properties,
1203 : : },
1204 : : { "atomic_default_mem_order",
1205 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1206 : : OMP_TRAIT_PROPERTY_ID, true,
1207 : : atomic_default_mem_order_properties,
1208 : : },
1209 : : { "requires",
1210 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1211 : : OMP_TRAIT_PROPERTY_CLAUSE_LIST, true,
1212 : : NULL
1213 : : },
1214 : : { "unified_address",
1215 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1216 : : OMP_TRAIT_PROPERTY_NONE, true,
1217 : : NULL
1218 : : },
1219 : : { "unified_shared_memory",
1220 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1221 : : OMP_TRAIT_PROPERTY_NONE, true,
1222 : : NULL
1223 : : },
1224 : : { "self_maps",
1225 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1226 : : OMP_TRAIT_PROPERTY_NONE, true,
1227 : : NULL
1228 : : },
1229 : : { "dynamic_allocators",
1230 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1231 : : OMP_TRAIT_PROPERTY_NONE, true,
1232 : : NULL
1233 : : },
1234 : : { "reverse_offload",
1235 : : (1 << OMP_TRAIT_SET_IMPLEMENTATION),
1236 : : OMP_TRAIT_PROPERTY_NONE, true,
1237 : : NULL
1238 : : },
1239 : : { "condition",
1240 : : (1 << OMP_TRAIT_SET_USER),
1241 : : OMP_TRAIT_PROPERTY_BOOL_EXPR, true,
1242 : : NULL
1243 : : },
1244 : : { "target",
1245 : : (1 << OMP_TRAIT_SET_CONSTRUCT),
1246 : : OMP_TRAIT_PROPERTY_NONE, false,
1247 : : NULL
1248 : : },
1249 : : { "teams",
1250 : : (1 << OMP_TRAIT_SET_CONSTRUCT),
1251 : : OMP_TRAIT_PROPERTY_NONE, false,
1252 : : NULL
1253 : : },
1254 : : { "parallel",
1255 : : (1 << OMP_TRAIT_SET_CONSTRUCT),
1256 : : OMP_TRAIT_PROPERTY_NONE, false,
1257 : : NULL
1258 : : },
1259 : : { "for",
1260 : : (1 << OMP_TRAIT_SET_CONSTRUCT),
1261 : : OMP_TRAIT_PROPERTY_NONE, false,
1262 : : NULL
1263 : : },
1264 : : { "simd",
1265 : : (1 << OMP_TRAIT_SET_CONSTRUCT),
1266 : : OMP_TRAIT_PROPERTY_CLAUSE_LIST, false,
1267 : : NULL
1268 : : },
1269 : : { "dispatch",
1270 : : (1 << OMP_TRAIT_SET_CONSTRUCT),
1271 : : OMP_TRAIT_PROPERTY_NONE, false,
1272 : : NULL
1273 : : },
1274 : : { NULL, 0, OMP_TRAIT_PROPERTY_NONE, false, NULL } /* OMP_TRAIT_LAST */
1275 : : };
1276 : :
1277 : : /* Return a name from PROP, a property in selectors accepting
1278 : : name lists. */
1279 : :
1280 : : const char *
1281 : 6773 : omp_context_name_list_prop (tree prop)
1282 : : {
1283 : 6773 : gcc_assert (OMP_TP_NAME (prop) == OMP_TP_NAMELIST_NODE);
1284 : 6773 : tree val = OMP_TP_VALUE (prop);
1285 : 6773 : switch (TREE_CODE (val))
1286 : : {
1287 : 5316 : case IDENTIFIER_NODE:
1288 : 5316 : return IDENTIFIER_POINTER (val);
1289 : 1457 : case STRING_CST:
1290 : 1457 : {
1291 : 1457 : const char *ret = TREE_STRING_POINTER (val);
1292 : 2914 : if ((size_t) TREE_STRING_LENGTH (val)
1293 : 2605 : == strlen (ret) + (lang_GNU_Fortran () ? 0 : 1))
1294 : : return ret;
1295 : : return NULL;
1296 : : }
1297 : : default:
1298 : : return NULL;
1299 : : }
1300 : : }
1301 : :
1302 : : /* Diagnose errors in an OpenMP context selector, return CTX if
1303 : : it is correct or error_mark_node otherwise. */
1304 : :
1305 : : tree
1306 : 1812 : omp_check_context_selector (location_t loc, tree ctx)
1307 : : {
1308 : 1812 : bool tss_seen[OMP_TRAIT_SET_LAST], ts_seen[OMP_TRAIT_LAST];
1309 : :
1310 : 1812 : memset (tss_seen, 0, sizeof (tss_seen));
1311 : 3768 : for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
1312 : : {
1313 : 2056 : enum omp_tss_code tss_code = OMP_TSS_CODE (tss);
1314 : 2056 : bool saw_any_prop = false;
1315 : 2056 : bool saw_other_prop = false;
1316 : :
1317 : : /* We can parse this, but not handle it yet. */
1318 : 2056 : if (tss_code == OMP_TRAIT_SET_TARGET_DEVICE)
1319 : 0 : sorry_at (loc, "%<target_device%> selector set is not supported yet");
1320 : :
1321 : : /* Each trait-set-selector-name can only be specified once. */
1322 : 2056 : if (tss_seen[tss_code])
1323 : : {
1324 : 30 : error_at (loc, "selector set %qs specified more than once",
1325 : 30 : OMP_TSS_NAME (tss));
1326 : 30 : return error_mark_node;
1327 : : }
1328 : : else
1329 : 2026 : tss_seen[tss_code] = true;
1330 : :
1331 : 2026 : memset (ts_seen, 0, sizeof (ts_seen));
1332 : 4524 : for (tree ts = OMP_TSS_TRAIT_SELECTORS (tss); ts; ts = TREE_CHAIN (ts))
1333 : : {
1334 : 2548 : enum omp_ts_code ts_code = OMP_TS_CODE (ts);
1335 : :
1336 : : /* Ignore unknown traits. */
1337 : 2548 : if (ts_code == OMP_TRAIT_INVALID)
1338 : 73 : continue;
1339 : :
1340 : : /* Each trait-selector-name can only be specified once. */
1341 : 2475 : if (ts_seen[ts_code])
1342 : : {
1343 : 15 : error_at (loc,
1344 : : "selector %qs specified more than once in set %qs",
1345 : 15 : OMP_TS_NAME (ts),
1346 : 15 : OMP_TSS_NAME (tss));
1347 : 15 : return error_mark_node;
1348 : : }
1349 : : else
1350 : 2460 : ts_seen[ts_code] = true;
1351 : :
1352 : : /* If trait-property "any" is specified in the "kind"
1353 : : trait-selector of the "device" selector set or the
1354 : : "target_device" selector sets, no other trait-property
1355 : : may be specified in the same selector set. */
1356 : 2460 : if (ts_code == OMP_TRAIT_DEVICE_KIND)
1357 : 676 : for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
1358 : : {
1359 : 268 : const char *prop = omp_context_name_list_prop (p);
1360 : 268 : if (!prop)
1361 : 1 : continue;
1362 : 267 : else if (strcmp (prop, "any") == 0)
1363 : : saw_any_prop = true;
1364 : : else
1365 : 205 : saw_other_prop = true;
1366 : : }
1367 : : /* It seems slightly suspicious that the spec's language covers
1368 : : the device_num selector too, but
1369 : : target_device={device_num(whatever),kind(any)}
1370 : : is probably not terribly useful anyway. */
1371 : 2256 : else if (ts_code == OMP_TRAIT_DEVICE_ARCH
1372 : : || ts_code == OMP_TRAIT_DEVICE_ISA
1373 : 2256 : || ts_code == OMP_TRAIT_DEVICE_NUM)
1374 : 344 : saw_other_prop = true;
1375 : :
1376 : : /* Each trait-property can only be specified once in a trait-selector
1377 : : other than the construct selector set. FIXME: only handles
1378 : : name-list properties, not clause-list properties, since the
1379 : : "requires" selector is not implemented yet (PR 113067). */
1380 : 2460 : if (tss_code != OMP_TRAIT_SET_CONSTRUCT)
1381 : 3970 : for (tree p1 = OMP_TS_PROPERTIES (ts); p1; p1 = TREE_CHAIN (p1))
1382 : : {
1383 : 1622 : if (OMP_TP_NAME (p1) != OMP_TP_NAMELIST_NODE)
1384 : : break;
1385 : 1094 : const char *n1 = omp_context_name_list_prop (p1);
1386 : 1094 : if (!n1)
1387 : 2 : continue;
1388 : 1450 : for (tree p2 = TREE_CHAIN (p1); p2; p2 = TREE_CHAIN (p2))
1389 : : {
1390 : 388 : const char *n2 = omp_context_name_list_prop (p2);
1391 : 388 : if (!n2)
1392 : 0 : continue;
1393 : 388 : if (!strcmp (n1, n2))
1394 : : {
1395 : 30 : error_at (loc,
1396 : : "trait-property %qs specified more "
1397 : : "than once in %qs selector",
1398 : 30 : n1, OMP_TS_NAME (ts));
1399 : 30 : return error_mark_node;
1400 : : }
1401 : : }
1402 : : }
1403 : :
1404 : : /* Check for unknown properties. */
1405 : 2430 : if (omp_ts_map[ts_code].valid_properties == NULL)
1406 : 1825 : continue;
1407 : 1899 : for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
1408 : 2816 : for (unsigned j = 0; ; j++)
1409 : : {
1410 : 3510 : const char *candidate
1411 : 3510 : = omp_ts_map[ts_code].valid_properties[j];
1412 : 3510 : if (candidate == NULL)
1413 : : {
1414 : : /* We've reached the end of the candidate array. */
1415 : 69 : if (ts_code == OMP_TRAIT_IMPLEMENTATION_ADMO)
1416 : : /* FIXME: not sure why this is an error vs warnings
1417 : : for the others, + incorrect/unknown wording? */
1418 : : {
1419 : 5 : error_at (loc,
1420 : : "incorrect property %qs of %qs selector",
1421 : 5 : IDENTIFIER_POINTER (OMP_TP_NAME (p)),
1422 : : "atomic_default_mem_order");
1423 : 5 : return error_mark_node;
1424 : : }
1425 : 64 : if (OMP_TP_NAME (p) == OMP_TP_NAMELIST_NODE
1426 : 64 : && (TREE_CODE (OMP_TP_VALUE (p)) == STRING_CST))
1427 : 18 : warning_at (loc, OPT_Wopenmp,
1428 : : "unknown property %qE of %qs selector",
1429 : 9 : OMP_TP_VALUE (p),
1430 : 9 : OMP_TS_NAME (ts));
1431 : 55 : else if (OMP_TP_NAME (p) == OMP_TP_NAMELIST_NODE)
1432 : 110 : warning_at (loc, OPT_Wopenmp,
1433 : : "unknown property %qs of %qs selector",
1434 : : omp_context_name_list_prop (p),
1435 : 55 : OMP_TS_NAME (ts));
1436 : 0 : else if (OMP_TP_NAME (p))
1437 : 0 : warning_at (loc, OPT_Wopenmp,
1438 : : "unknown property %qs of %qs selector",
1439 : 0 : IDENTIFIER_POINTER (OMP_TP_NAME (p)),
1440 : 0 : OMP_TS_NAME (ts));
1441 : : break;
1442 : : }
1443 : 3441 : else if (OMP_TP_NAME (p) == OMP_TP_NAMELIST_NODE)
1444 : : /* Property-list traits. */
1445 : : {
1446 : 3205 : const char *str = omp_context_name_list_prop (p);
1447 : 3205 : if (str && !strcmp (str, candidate))
1448 : : break;
1449 : : }
1450 : 236 : else if (!strcmp (IDENTIFIER_POINTER (OMP_TP_NAME (p)),
1451 : : candidate))
1452 : : /* Identifier traits. */
1453 : : break;
1454 : 2816 : }
1455 : : }
1456 : :
1457 : 1976 : if (saw_any_prop && saw_other_prop)
1458 : : {
1459 : 20 : error_at (loc,
1460 : : "no other trait-property may be specified "
1461 : : "in the same selector set with %<kind(\"any\")%>");
1462 : 20 : return error_mark_node;
1463 : : }
1464 : : }
1465 : : return ctx;
1466 : : }
1467 : :
1468 : :
1469 : : /* Register VARIANT as variant of some base function marked with
1470 : : #pragma omp declare variant. CONSTRUCT is corresponding list of
1471 : : trait-selectors for the construct selector set. This is stashed as the
1472 : : value of the "omp declare variant variant" attribute on VARIANT. */
1473 : : void
1474 : 1537 : omp_mark_declare_variant (location_t loc, tree variant, tree construct)
1475 : : {
1476 : : /* Ignore this variant if it contains unknown construct selectors.
1477 : : It will never match, and the front ends have already issued a warning
1478 : : about it. */
1479 : 2369 : for (tree c = construct; c; c = TREE_CHAIN (c))
1480 : 888 : if (OMP_TS_CODE (c) == OMP_TRAIT_INVALID)
1481 : : return;
1482 : :
1483 : 1481 : tree attr = lookup_attribute ("omp declare variant variant",
1484 : 1481 : DECL_ATTRIBUTES (variant));
1485 : 1481 : if (attr == NULL_TREE)
1486 : : {
1487 : 963 : attr = tree_cons (get_identifier ("omp declare variant variant"),
1488 : : unshare_expr (construct),
1489 : 963 : DECL_ATTRIBUTES (variant));
1490 : 963 : DECL_ATTRIBUTES (variant) = attr;
1491 : 963 : return;
1492 : : }
1493 : 518 : if ((TREE_VALUE (attr) != NULL_TREE) != (construct != NULL_TREE)
1494 : 518 : || (construct != NULL_TREE
1495 : 103 : && omp_context_selector_set_compare (OMP_TRAIT_SET_CONSTRUCT,
1496 : 103 : TREE_VALUE (attr),
1497 : : construct)))
1498 : 53 : error_at (loc, "%qD used as a variant with incompatible %<construct%> "
1499 : : "selector sets", variant);
1500 : : }
1501 : :
1502 : :
1503 : : /* Constructors for context selectors. */
1504 : :
1505 : : tree
1506 : 2060 : make_trait_set_selector (enum omp_tss_code code, tree selectors, tree chain)
1507 : : {
1508 : 2060 : return tree_cons (build_int_cst (integer_type_node, code),
1509 : 2060 : selectors, chain);
1510 : : }
1511 : :
1512 : : tree
1513 : 2690 : make_trait_selector (enum omp_ts_code code, tree score, tree properties,
1514 : : tree chain)
1515 : : {
1516 : 2690 : if (score == NULL_TREE)
1517 : 2362 : return tree_cons (build_int_cst (integer_type_node, code),
1518 : : properties, chain);
1519 : : else
1520 : 328 : return tree_cons (build_int_cst (integer_type_node, code),
1521 : : tree_cons (OMP_TS_SCORE_NODE, score, properties),
1522 : : chain);
1523 : : }
1524 : :
1525 : : tree
1526 : 1691 : make_trait_property (tree name, tree value, tree chain)
1527 : : {
1528 : 1691 : return tree_cons (name, value, chain);
1529 : : }
1530 : :
1531 : : /* Return 1 if context selector matches the current OpenMP context, 0
1532 : : if it does not and -1 if it is unknown and need to be determined later.
1533 : : Some properties can be checked right away during parsing (this routine),
1534 : : others need to wait until the whole TU is parsed, others need to wait until
1535 : : IPA, others until vectorization. */
1536 : :
1537 : : int
1538 : 2951 : omp_context_selector_matches (tree ctx)
1539 : : {
1540 : 2951 : int ret = 1;
1541 : 5277 : for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
1542 : : {
1543 : 3239 : enum omp_tss_code set = OMP_TSS_CODE (tss);
1544 : 3239 : tree selectors = OMP_TSS_TRAIT_SELECTORS (tss);
1545 : :
1546 : : /* Immediately reject the match if there are any ignored
1547 : : selectors present. */
1548 : 7103 : for (tree ts = selectors; ts; ts = TREE_CHAIN (ts))
1549 : 3932 : if (OMP_TS_CODE (ts) == OMP_TRAIT_INVALID)
1550 : : return 0;
1551 : :
1552 : 3171 : if (set == OMP_TRAIT_SET_CONSTRUCT)
1553 : : {
1554 : : /* For now, ignore the construct set. While something can be
1555 : : determined already during parsing, we don't know until end of TU
1556 : : whether additional constructs aren't added through declare variant
1557 : : unless "omp declare variant variant" attribute exists already
1558 : : (so in most of the cases), and we'd need to maintain set of
1559 : : surrounding OpenMP constructs, which is better handled during
1560 : : gimplification. */
1561 : 1437 : if (symtab->state == PARSING)
1562 : : {
1563 : 544 : ret = -1;
1564 : 544 : continue;
1565 : : }
1566 : :
1567 : 893 : int nconstructs = list_length (selectors);
1568 : 893 : enum tree_code *constructs = NULL;
1569 : 893 : if (nconstructs)
1570 : : {
1571 : : /* Even though this alloca appears in a loop over selector
1572 : : sets, it does not repeatedly grow the stack, because
1573 : : there can be only one construct selector set specified.
1574 : : This is enforced by omp_check_context_selector. */
1575 : 893 : constructs
1576 : 893 : = (enum tree_code *) alloca (nconstructs
1577 : : * sizeof (enum tree_code));
1578 : 893 : omp_construct_traits_to_codes (selectors, nconstructs,
1579 : : constructs);
1580 : : }
1581 : :
1582 : 893 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
1583 : : {
1584 : 9 : if (!cfun->after_inlining)
1585 : : {
1586 : 0 : ret = -1;
1587 : 0 : continue;
1588 : : }
1589 : : int i;
1590 : 9 : for (i = 0; i < nconstructs; ++i)
1591 : 9 : if (constructs[i] == OMP_SIMD)
1592 : : break;
1593 : 9 : if (i < nconstructs)
1594 : : {
1595 : 9 : ret = -1;
1596 : 9 : continue;
1597 : : }
1598 : : /* If there is no simd, assume it is ok after IPA,
1599 : : constructs should have been checked before. */
1600 : 0 : continue;
1601 : 0 : }
1602 : :
1603 : 884 : int r = omp_construct_selector_matches (constructs, nconstructs,
1604 : : NULL);
1605 : 884 : if (r == 0)
1606 : : return 0;
1607 : 616 : if (r == -1)
1608 : 20 : ret = -1;
1609 : 616 : continue;
1610 : 616 : }
1611 : 3042 : for (tree ts = selectors; ts; ts = TREE_CHAIN (ts))
1612 : : {
1613 : 1885 : enum omp_ts_code sel = OMP_TS_CODE (ts);
1614 : 1885 : switch (sel)
1615 : : {
1616 : 404 : case OMP_TRAIT_IMPLEMENTATION_VENDOR:
1617 : 404 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1618 : 1088 : for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
1619 : : {
1620 : 419 : const char *prop = omp_context_name_list_prop (p);
1621 : 419 : if (prop == NULL)
1622 : : return 0;
1623 : 418 : if (!strcmp (prop, "gnu"))
1624 : 280 : continue;
1625 : : return 0;
1626 : : }
1627 : : break;
1628 : 30 : case OMP_TRAIT_IMPLEMENTATION_EXTENSION:
1629 : 30 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1630 : : /* We don't support any extensions right now. */
1631 : : return 0;
1632 : : break;
1633 : 120 : case OMP_TRAIT_IMPLEMENTATION_ADMO:
1634 : 120 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1635 : : {
1636 : 120 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
1637 : : break;
1638 : :
1639 : 120 : enum omp_memory_order omo
1640 : : = ((enum omp_memory_order)
1641 : 120 : (omp_requires_mask
1642 : : & OMP_REQUIRES_ATOMIC_DEFAULT_MEM_ORDER));
1643 : 120 : if (omo == OMP_MEMORY_ORDER_UNSPECIFIED)
1644 : : {
1645 : : /* We don't know yet, until end of TU. */
1646 : 65 : if (symtab->state == PARSING)
1647 : : {
1648 : : ret = -1;
1649 : : break;
1650 : : }
1651 : : else
1652 : : omo = OMP_MEMORY_ORDER_RELAXED;
1653 : : }
1654 : 55 : tree p = OMP_TS_PROPERTIES (ts);
1655 : 55 : const char *prop = IDENTIFIER_POINTER (OMP_TP_NAME (p));
1656 : 55 : if (!strcmp (prop, "relaxed")
1657 : 2 : && omo != OMP_MEMORY_ORDER_RELAXED)
1658 : : return 0;
1659 : 53 : else if (!strcmp (prop, "seq_cst")
1660 : 53 : && omo != OMP_MEMORY_ORDER_SEQ_CST)
1661 : : return 0;
1662 : 53 : else if (!strcmp (prop, "acq_rel")
1663 : 0 : && omo != OMP_MEMORY_ORDER_ACQ_REL)
1664 : : return 0;
1665 : 53 : else if (!strcmp (prop, "acquire")
1666 : 0 : && omo != OMP_MEMORY_ORDER_ACQUIRE)
1667 : : return 0;
1668 : 53 : else if (!strcmp (prop, "release")
1669 : 0 : && omo != OMP_MEMORY_ORDER_RELEASE)
1670 : : return 0;
1671 : : }
1672 : : break;
1673 : 170 : case OMP_TRAIT_DEVICE_ARCH:
1674 : 170 : if (set == OMP_TRAIT_SET_DEVICE)
1675 : 420 : for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
1676 : : {
1677 : 175 : const char *arch = omp_context_name_list_prop (p);
1678 : 175 : if (arch == NULL)
1679 : : return 0;
1680 : 175 : int r = 0;
1681 : 175 : if (targetm.omp.device_kind_arch_isa != NULL)
1682 : 175 : r = targetm.omp.device_kind_arch_isa (omp_device_arch,
1683 : : arch);
1684 : 175 : if (r == 0 || (r == -1 && symtab->state != PARSING))
1685 : : {
1686 : : /* If we are or might be in a target region or
1687 : : declare target function, need to take into account
1688 : : also offloading values. */
1689 : 95 : if (!omp_maybe_offloaded ())
1690 : 95 : return 0;
1691 : : if (ENABLE_OFFLOADING)
1692 : : {
1693 : : const char *arches = omp_offload_device_arch;
1694 : : if (omp_offload_device_kind_arch_isa (arches,
1695 : : arch))
1696 : : {
1697 : : ret = -1;
1698 : : continue;
1699 : : }
1700 : : }
1701 : : return 0;
1702 : : }
1703 : : else if (r == -1)
1704 : : ret = -1;
1705 : : /* If arch matches on the host, it still might not match
1706 : : in the offloading region. */
1707 : : else if (omp_maybe_offloaded ())
1708 : : ret = -1;
1709 : : }
1710 : : break;
1711 : 20 : case OMP_TRAIT_IMPLEMENTATION_UNIFIED_ADDRESS:
1712 : 20 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1713 : : {
1714 : 20 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
1715 : : break;
1716 : :
1717 : 20 : if ((omp_requires_mask & OMP_REQUIRES_UNIFIED_ADDRESS) == 0)
1718 : : {
1719 : 20 : if (symtab->state == PARSING)
1720 : : ret = -1;
1721 : : else
1722 : : return 0;
1723 : : }
1724 : : }
1725 : : break;
1726 : 15 : case OMP_TRAIT_IMPLEMENTATION_UNIFIED_SHARED_MEMORY:
1727 : 15 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1728 : : {
1729 : 15 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
1730 : : break;
1731 : :
1732 : 15 : if ((omp_requires_mask
1733 : : & OMP_REQUIRES_UNIFIED_SHARED_MEMORY) == 0)
1734 : : {
1735 : 15 : if (symtab->state == PARSING)
1736 : : ret = -1;
1737 : : else
1738 : : return 0;
1739 : : }
1740 : : }
1741 : : break;
1742 : 5 : case OMP_TRAIT_IMPLEMENTATION_SELF_MAPS:
1743 : 5 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1744 : : {
1745 : 5 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
1746 : : break;
1747 : :
1748 : 5 : if ((omp_requires_mask
1749 : : & OMP_REQUIRES_SELF_MAPS) == 0)
1750 : : {
1751 : 5 : if (symtab->state == PARSING)
1752 : : ret = -1;
1753 : : else
1754 : : return 0;
1755 : : }
1756 : : }
1757 : : break;
1758 : 10 : case OMP_TRAIT_IMPLEMENTATION_DYNAMIC_ALLOCATORS:
1759 : 10 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1760 : : {
1761 : 10 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
1762 : : break;
1763 : :
1764 : 10 : if ((omp_requires_mask
1765 : : & OMP_REQUIRES_DYNAMIC_ALLOCATORS) == 0)
1766 : : {
1767 : 10 : if (symtab->state == PARSING)
1768 : : ret = -1;
1769 : : else
1770 : : return 0;
1771 : : }
1772 : : }
1773 : : break;
1774 : 10 : case OMP_TRAIT_IMPLEMENTATION_REVERSE_OFFLOAD:
1775 : 10 : if (set == OMP_TRAIT_SET_IMPLEMENTATION)
1776 : : {
1777 : 10 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
1778 : : break;
1779 : :
1780 : 10 : if ((omp_requires_mask & OMP_REQUIRES_REVERSE_OFFLOAD) == 0)
1781 : : {
1782 : 10 : if (symtab->state == PARSING)
1783 : : ret = -1;
1784 : : else
1785 : : return 0;
1786 : : }
1787 : : }
1788 : : break;
1789 : 231 : case OMP_TRAIT_DEVICE_KIND:
1790 : 231 : if (set == OMP_TRAIT_SET_DEVICE)
1791 : 654 : for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
1792 : : {
1793 : 255 : const char *prop = omp_context_name_list_prop (p);
1794 : 255 : if (prop == NULL)
1795 : : return 0;
1796 : 254 : if (!strcmp (prop, "any"))
1797 : 24 : continue;
1798 : 230 : if (!strcmp (prop, "host"))
1799 : : {
1800 : : #ifdef ACCEL_COMPILER
1801 : : return 0;
1802 : : #else
1803 : 61 : if (omp_maybe_offloaded ())
1804 : : ret = -1;
1805 : 61 : continue;
1806 : : #endif
1807 : : }
1808 : 169 : if (!strcmp (prop, "nohost"))
1809 : : {
1810 : : #ifndef ACCEL_COMPILER
1811 : : if (omp_maybe_offloaded ())
1812 : : ret = -1;
1813 : : else
1814 : : return 0;
1815 : : #endif
1816 : : continue;
1817 : : }
1818 : 153 : int r = 0;
1819 : 153 : if (targetm.omp.device_kind_arch_isa != NULL)
1820 : 153 : r = targetm.omp.device_kind_arch_isa (omp_device_kind,
1821 : : prop);
1822 : : else
1823 : 0 : r = strcmp (prop, "cpu") == 0;
1824 : 153 : if (r == 0 || (r == -1 && symtab->state != PARSING))
1825 : : {
1826 : : /* If we are or might be in a target region or
1827 : : declare target function, need to take into account
1828 : : also offloading values. */
1829 : : if (!omp_maybe_offloaded ())
1830 : : return 0;
1831 : : if (ENABLE_OFFLOADING)
1832 : : {
1833 : : const char *kinds = omp_offload_device_kind;
1834 : : if (omp_offload_device_kind_arch_isa (kinds, prop))
1835 : : {
1836 : : ret = -1;
1837 : : continue;
1838 : : }
1839 : : }
1840 : : return 0;
1841 : : }
1842 : : else if (r == -1)
1843 : : ret = -1;
1844 : : /* If kind matches on the host, it still might not match
1845 : : in the offloading region. */
1846 : : else if (omp_maybe_offloaded ())
1847 : : ret = -1;
1848 : : }
1849 : : break;
1850 : 371 : case OMP_TRAIT_DEVICE_ISA:
1851 : 371 : if (set == OMP_TRAIT_SET_DEVICE)
1852 : 1182 : for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
1853 : : {
1854 : 561 : const char *isa = omp_context_name_list_prop (p);
1855 : 561 : if (isa == NULL)
1856 : : return 0;
1857 : 561 : int r = 0;
1858 : 561 : if (targetm.omp.device_kind_arch_isa != NULL)
1859 : 561 : r = targetm.omp.device_kind_arch_isa (omp_device_isa,
1860 : : isa);
1861 : 561 : if (r == 0 || (r == -1 && symtab->state != PARSING))
1862 : : {
1863 : : /* If isa is valid on the target, but not in the
1864 : : current function and current function has
1865 : : #pragma omp declare simd on it, some simd clones
1866 : : might have the isa added later on. */
1867 : 110 : if (r == -1
1868 : 110 : && targetm.simd_clone.compute_vecsize_and_simdlen
1869 : 110 : && (cfun == NULL || !cfun->after_inlining))
1870 : : {
1871 : 26 : tree attrs
1872 : 26 : = DECL_ATTRIBUTES (current_function_decl);
1873 : 26 : if (lookup_attribute ("omp declare simd", attrs))
1874 : : {
1875 : 17 : ret = -1;
1876 : 17 : continue;
1877 : : }
1878 : : }
1879 : : /* If we are or might be in a target region or
1880 : : declare target function, need to take into account
1881 : : also offloading values. */
1882 : 121 : if (!omp_maybe_offloaded ())
1883 : 121 : return 0;
1884 : : if (ENABLE_OFFLOADING)
1885 : : {
1886 : : const char *isas = omp_offload_device_isa;
1887 : : if (omp_offload_device_kind_arch_isa (isas, isa))
1888 : : {
1889 : : ret = -1;
1890 : : continue;
1891 : : }
1892 : : }
1893 : : return 0;
1894 : : }
1895 : : else if (r == -1)
1896 : : ret = -1;
1897 : : /* If isa matches on the host, it still might not match
1898 : : in the offloading region. */
1899 : : else if (omp_maybe_offloaded ())
1900 : : ret = -1;
1901 : : }
1902 : : break;
1903 : 499 : case OMP_TRAIT_USER_CONDITION:
1904 : 499 : if (set == OMP_TRAIT_SET_USER)
1905 : 1007 : for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
1906 : 496 : if (OMP_TP_NAME (p) == NULL_TREE)
1907 : : {
1908 : 496 : if (integer_zerop (OMP_TP_VALUE (p)))
1909 : : return 0;
1910 : 369 : if (integer_nonzerop (OMP_TP_VALUE (p)))
1911 : : break;
1912 : : ret = -1;
1913 : : }
1914 : : break;
1915 : : default:
1916 : : break;
1917 : : }
1918 : : }
1919 : : }
1920 : : return ret;
1921 : : }
1922 : :
1923 : : /* Compare construct={simd} CLAUSES1 with CLAUSES2, return 0/-1/1/2 as
1924 : : in omp_context_selector_set_compare. */
1925 : :
1926 : : static int
1927 : 18 : omp_construct_simd_compare (tree clauses1, tree clauses2)
1928 : : {
1929 : 18 : if (clauses1 == NULL_TREE)
1930 : 0 : return clauses2 == NULL_TREE ? 0 : -1;
1931 : 18 : if (clauses2 == NULL_TREE)
1932 : : return 1;
1933 : :
1934 : 54 : int r = 0;
1935 : 36 : struct declare_variant_simd_data {
1936 : : bool inbranch, notinbranch;
1937 : : tree simdlen;
1938 : : auto_vec<tree,16> data_sharing;
1939 : : auto_vec<tree,16> aligned;
1940 : 36 : declare_variant_simd_data ()
1941 : 36 : : inbranch(false), notinbranch(false), simdlen(NULL_TREE) {}
1942 : 126 : } data[2];
1943 : : unsigned int i;
1944 : 54 : for (i = 0; i < 2; i++)
1945 : 196 : for (tree c = i ? clauses2 : clauses1; c; c = OMP_CLAUSE_CHAIN (c))
1946 : : {
1947 : 124 : vec<tree> *v;
1948 : 124 : switch (OMP_CLAUSE_CODE (c))
1949 : : {
1950 : 12 : case OMP_CLAUSE_INBRANCH:
1951 : 12 : data[i].inbranch = true;
1952 : 12 : continue;
1953 : 24 : case OMP_CLAUSE_NOTINBRANCH:
1954 : 24 : data[i].notinbranch = true;
1955 : 24 : continue;
1956 : 36 : case OMP_CLAUSE_SIMDLEN:
1957 : 36 : data[i].simdlen = OMP_CLAUSE_SIMDLEN_EXPR (c);
1958 : 36 : continue;
1959 : 36 : case OMP_CLAUSE_UNIFORM:
1960 : 36 : case OMP_CLAUSE_LINEAR:
1961 : 36 : v = &data[i].data_sharing;
1962 : 36 : break;
1963 : 16 : case OMP_CLAUSE_ALIGNED:
1964 : 16 : v = &data[i].aligned;
1965 : 16 : break;
1966 : 0 : default:
1967 : 0 : gcc_unreachable ();
1968 : : }
1969 : 52 : unsigned HOST_WIDE_INT argno = tree_to_uhwi (OMP_CLAUSE_DECL (c));
1970 : 52 : if (argno >= v->length ())
1971 : 52 : v->safe_grow_cleared (argno + 1, true);
1972 : 52 : (*v)[argno] = c;
1973 : : }
1974 : : /* Here, r is used as a bitmask, 2 is set if CLAUSES1 has something
1975 : : CLAUSES2 doesn't, 1 is set if CLAUSES2 has something CLAUSES1
1976 : : doesn't. Thus, r == 3 implies return value 2, r == 1 implies
1977 : : -1, r == 2 implies 1 and r == 0 implies 0. */
1978 : 18 : if (data[0].inbranch != data[1].inbranch)
1979 : 0 : r |= data[0].inbranch ? 2 : 1;
1980 : 18 : if (data[0].notinbranch != data[1].notinbranch)
1981 : 0 : r |= data[0].notinbranch ? 2 : 1;
1982 : 18 : if (!simple_cst_equal (data[0].simdlen, data[1].simdlen))
1983 : : {
1984 : 0 : if (data[0].simdlen && data[1].simdlen)
1985 : : return 2;
1986 : 0 : r |= data[0].simdlen ? 2 : 1;
1987 : : }
1988 : 54 : if (data[0].data_sharing.length () < data[1].data_sharing.length ()
1989 : 36 : || data[0].aligned.length () < data[1].aligned.length ())
1990 : 0 : r |= 1;
1991 : : tree c1, c2;
1992 : 58 : FOR_EACH_VEC_ELT (data[0].data_sharing, i, c1)
1993 : : {
1994 : 48 : c2 = (i < data[1].data_sharing.length ()
1995 : 96 : ? data[1].data_sharing[i] : NULL_TREE);
1996 : 48 : if ((c1 == NULL_TREE) != (c2 == NULL_TREE))
1997 : : {
1998 : 0 : r |= c1 != NULL_TREE ? 2 : 1;
1999 : 0 : continue;
2000 : : }
2001 : 48 : if (c1 == NULL_TREE)
2002 : 30 : continue;
2003 : 18 : if (OMP_CLAUSE_CODE (c1) != OMP_CLAUSE_CODE (c2))
2004 : : return 2;
2005 : 13 : if (OMP_CLAUSE_CODE (c1) != OMP_CLAUSE_LINEAR)
2006 : 7 : continue;
2007 : 6 : if (OMP_CLAUSE_LINEAR_VARIABLE_STRIDE (c1)
2008 : 6 : != OMP_CLAUSE_LINEAR_VARIABLE_STRIDE (c2))
2009 : : return 2;
2010 : 6 : if (OMP_CLAUSE_LINEAR_KIND (c1) != OMP_CLAUSE_LINEAR_KIND (c2))
2011 : : return 2;
2012 : 6 : if (!simple_cst_equal (OMP_CLAUSE_LINEAR_STEP (c1),
2013 : 6 : OMP_CLAUSE_LINEAR_STEP (c2)))
2014 : : return 2;
2015 : : }
2016 : 23 : FOR_EACH_VEC_ELT (data[0].aligned, i, c1)
2017 : : {
2018 : 36 : c2 = i < data[1].aligned.length () ? data[1].aligned[i] : NULL_TREE;
2019 : 18 : if ((c1 == NULL_TREE) != (c2 == NULL_TREE))
2020 : : {
2021 : 0 : r |= c1 != NULL_TREE ? 2 : 1;
2022 : 0 : continue;
2023 : : }
2024 : 18 : if (c1 == NULL_TREE)
2025 : 12 : continue;
2026 : 6 : if (!simple_cst_equal (OMP_CLAUSE_ALIGNED_ALIGNMENT (c1),
2027 : 6 : OMP_CLAUSE_ALIGNED_ALIGNMENT (c2)))
2028 : : return 2;
2029 : : }
2030 : 5 : switch (r)
2031 : : {
2032 : : case 0: return 0;
2033 : 0 : case 1: return -1;
2034 : 0 : case 2: return 1;
2035 : : case 3: return 2;
2036 : 0 : default: gcc_unreachable ();
2037 : : }
2038 : 54 : }
2039 : :
2040 : : /* Compare properties of selectors SEL from SET other than construct.
2041 : : CTX1 and CTX2 are the lists of properties to compare.
2042 : : Return 0/-1/1/2 as in omp_context_selector_set_compare.
2043 : : Unlike set names or selector names, properties can have duplicates. */
2044 : :
2045 : : static int
2046 : 71 : omp_context_selector_props_compare (enum omp_tss_code set,
2047 : : enum omp_ts_code sel,
2048 : : tree ctx1, tree ctx2)
2049 : : {
2050 : 71 : int ret = 0;
2051 : 208 : for (int pass = 0; pass < 2; pass++)
2052 : 426 : for (tree p1 = pass ? ctx2 : ctx1; p1; p1 = TREE_CHAIN (p1))
2053 : : {
2054 : 152 : tree p2;
2055 : 374 : for (p2 = pass ? ctx1 : ctx2; p2; p2 = TREE_CHAIN (p2))
2056 : 212 : if (OMP_TP_NAME (p1) == OMP_TP_NAME (p2))
2057 : : {
2058 : 212 : if (OMP_TP_NAME (p1) == NULL_TREE)
2059 : : {
2060 : 72 : if (set == OMP_TRAIT_SET_USER
2061 : 72 : && sel == OMP_TRAIT_USER_CONDITION)
2062 : : {
2063 : 144 : if (integer_zerop (OMP_TP_VALUE (p1))
2064 : 72 : != integer_zerop (OMP_TP_VALUE (p2)))
2065 : : return 2;
2066 : : break;
2067 : : }
2068 : 0 : if (simple_cst_equal (OMP_TP_VALUE (p1), OMP_TP_VALUE (p2)))
2069 : : break;
2070 : : }
2071 : 140 : else if (OMP_TP_NAME (p1) == OMP_TP_NAMELIST_NODE)
2072 : : {
2073 : : /* Handle string constant vs identifier comparison for
2074 : : name-list properties. */
2075 : 140 : const char *n1 = omp_context_name_list_prop (p1);
2076 : 140 : const char *n2 = omp_context_name_list_prop (p2);
2077 : 140 : if (n1 && n2 && !strcmp (n1, n2))
2078 : : break;
2079 : : }
2080 : : else
2081 : : break;
2082 : : }
2083 : 10 : if (p2 == NULL_TREE)
2084 : : {
2085 : 10 : int r = pass ? -1 : 1;
2086 : 10 : if (ret && ret != r)
2087 : : return 2;
2088 : 10 : else if (pass)
2089 : : return r;
2090 : : else
2091 : : {
2092 : : ret = r;
2093 : : break;
2094 : : }
2095 : : }
2096 : : }
2097 : : return ret;
2098 : : }
2099 : :
2100 : : /* Compare single context selector sets CTX1 and CTX2 with SET name.
2101 : : CTX1 and CTX2 are lists of trait-selectors.
2102 : : Return 0 if CTX1 is equal to CTX2,
2103 : : -1 if CTX1 is a strict subset of CTX2,
2104 : : 1 if CTX2 is a strict subset of CTX1, or
2105 : : 2 if neither context is a subset of another one. */
2106 : :
2107 : : int
2108 : 398 : omp_context_selector_set_compare (enum omp_tss_code set, tree ctx1, tree ctx2)
2109 : : {
2110 : :
2111 : : /* If either list includes an ignored selector trait, neither can
2112 : : be a subset of the other. */
2113 : 907 : for (tree ts = ctx1; ts; ts = TREE_CHAIN (ts))
2114 : 509 : if (OMP_TS_CODE (ts) == OMP_TRAIT_INVALID)
2115 : : return 2;
2116 : 931 : for (tree ts = ctx2; ts; ts = TREE_CHAIN (ts))
2117 : 533 : if (OMP_TS_CODE (ts) == OMP_TRAIT_INVALID)
2118 : : return 2;
2119 : :
2120 : 398 : bool swapped = false;
2121 : 398 : int ret = 0;
2122 : 398 : int len1 = list_length (ctx1);
2123 : 398 : int len2 = list_length (ctx2);
2124 : 398 : int cnt = 0;
2125 : 398 : if (len1 < len2)
2126 : : {
2127 : 49 : swapped = true;
2128 : 49 : std::swap (ctx1, ctx2);
2129 : 49 : std::swap (len1, len2);
2130 : : }
2131 : :
2132 : 398 : if (set == OMP_TRAIT_SET_CONSTRUCT)
2133 : : {
2134 : : tree ts1;
2135 : : tree ts2 = ctx2;
2136 : : /* Handle construct set specially. In this case the order
2137 : : of the selector matters too. */
2138 : 283 : for (ts1 = ctx1; ts1; ts1 = TREE_CHAIN (ts1))
2139 : 275 : if (OMP_TS_CODE (ts1) == OMP_TS_CODE (ts2))
2140 : : {
2141 : 209 : int r = 0;
2142 : 209 : if (OMP_TS_CODE (ts1) == OMP_TRAIT_CONSTRUCT_SIMD)
2143 : 18 : r = omp_construct_simd_compare (OMP_TS_PROPERTIES (ts1),
2144 : 18 : OMP_TS_PROPERTIES (ts2));
2145 : 209 : if (r == 2 || (ret && r && (ret < 0) != (r < 0)))
2146 : 13 : return 2;
2147 : 196 : if (ret == 0)
2148 : 134 : ret = r;
2149 : 196 : ts2 = TREE_CHAIN (ts2);
2150 : 196 : if (ts2 == NULL_TREE)
2151 : : {
2152 : 130 : ts1 = TREE_CHAIN (ts1);
2153 : 130 : break;
2154 : : }
2155 : : }
2156 : 66 : else if (ret < 0)
2157 : : return 2;
2158 : : else
2159 : : ret = 1;
2160 : 138 : if (ts2 != NULL_TREE)
2161 : : return 2;
2162 : 130 : if (ts1 != NULL_TREE)
2163 : : {
2164 : 31 : if (ret < 0)
2165 : : return 2;
2166 : : ret = 1;
2167 : : }
2168 : 99 : if (ret == 0)
2169 : : return 0;
2170 : 70 : return swapped ? -ret : ret;
2171 : : }
2172 : 391 : for (tree ts1 = ctx1; ts1; ts1 = TREE_CHAIN (ts1))
2173 : : {
2174 : 267 : enum omp_ts_code sel = OMP_TS_CODE (ts1);
2175 : 267 : tree ts2;
2176 : 350 : for (ts2 = ctx2; ts2; ts2 = TREE_CHAIN (ts2))
2177 : 277 : if (sel == OMP_TS_CODE (ts2))
2178 : : {
2179 : 194 : tree score1 = OMP_TS_SCORE (ts1);
2180 : 194 : tree score2 = OMP_TS_SCORE (ts2);
2181 : 194 : if (score1 && score2 && !simple_cst_equal (score1, score2))
2182 : : return 2;
2183 : :
2184 : 151 : int r = omp_context_selector_props_compare (set, OMP_TS_CODE (ts1),
2185 : 71 : OMP_TS_PROPERTIES (ts1),
2186 : 71 : OMP_TS_PROPERTIES (ts2));
2187 : 71 : if (r == 2 || (ret && r && (ret < 0) != (r < 0)))
2188 : : return 2;
2189 : 71 : if (ret == 0)
2190 : 66 : ret = r;
2191 : 71 : cnt++;
2192 : 71 : break;
2193 : : }
2194 : 144 : if (ts2 == NULL_TREE)
2195 : : {
2196 : 73 : if (ret == -1)
2197 : : return 2;
2198 : : ret = 1;
2199 : : }
2200 : : }
2201 : 124 : if (cnt < len2)
2202 : : return 2;
2203 : 66 : if (ret == 0)
2204 : : return 0;
2205 : 20 : return swapped ? -ret : ret;
2206 : : }
2207 : :
2208 : : /* Compare whole context selector specification CTX1 and CTX2.
2209 : : Return 0 if CTX1 is equal to CTX2,
2210 : : -1 if CTX1 is a strict subset of CTX2,
2211 : : 1 if CTX2 is a strict subset of CTX1, or
2212 : : 2 if neither context is a subset of another one. */
2213 : :
2214 : : static int
2215 : 468 : omp_context_selector_compare (tree ctx1, tree ctx2)
2216 : : {
2217 : 468 : bool swapped = false;
2218 : 468 : int ret = 0;
2219 : 468 : int len1 = list_length (ctx1);
2220 : 468 : int len2 = list_length (ctx2);
2221 : 468 : int cnt = 0;
2222 : 468 : if (len1 < len2)
2223 : : {
2224 : 64 : swapped = true;
2225 : 64 : std::swap (ctx1, ctx2);
2226 : 64 : std::swap (len1, len2);
2227 : : }
2228 : 874 : for (tree tss1 = ctx1; tss1; tss1 = TREE_CHAIN (tss1))
2229 : : {
2230 : 602 : enum omp_tss_code set = OMP_TSS_CODE (tss1);
2231 : 602 : tree tss2;
2232 : 919 : for (tss2 = ctx2; tss2; tss2 = TREE_CHAIN (tss2))
2233 : 612 : if (set == OMP_TSS_CODE (tss2))
2234 : : {
2235 : 295 : int r
2236 : : = omp_context_selector_set_compare
2237 : 590 : (set, OMP_TSS_TRAIT_SELECTORS (tss1),
2238 : 295 : OMP_TSS_TRAIT_SELECTORS (tss2));
2239 : 295 : if (r == 2 || (ret && r && (ret < 0) != (r < 0)))
2240 : : return 2;
2241 : 114 : if (ret == 0)
2242 : 93 : ret = r;
2243 : 114 : cnt++;
2244 : 114 : break;
2245 : : }
2246 : 421 : if (tss2 == NULL_TREE)
2247 : : {
2248 : 307 : if (ret == -1)
2249 : : return 2;
2250 : : ret = 1;
2251 : : }
2252 : : }
2253 : 272 : if (cnt < len2)
2254 : : return 2;
2255 : 94 : if (ret == 0)
2256 : : return 0;
2257 : 79 : return swapped ? -ret : ret;
2258 : : }
2259 : :
2260 : : /* From context selector CTX, return trait-selector with name SEL in
2261 : : trait-selector-set with name SET if any, or NULL_TREE if not found. */
2262 : : tree
2263 : 3075 : omp_get_context_selector (tree ctx, enum omp_tss_code set,
2264 : : enum omp_ts_code sel)
2265 : : {
2266 : 6233 : for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
2267 : 3549 : if (OMP_TSS_CODE (tss) == set)
2268 : 2484 : for (tree ts = OMP_TSS_TRAIT_SELECTORS (tss); ts; ts = TREE_CHAIN (ts))
2269 : 1604 : if (OMP_TS_CODE (ts) == sel)
2270 : : return ts;
2271 : : return NULL_TREE;
2272 : : }
2273 : :
2274 : : /* Similar, but returns the whole trait-selector list for SET in CTX. */
2275 : : tree
2276 : 1906 : omp_get_context_selector_list (tree ctx, enum omp_tss_code set)
2277 : : {
2278 : 3256 : for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
2279 : 2071 : if (OMP_TSS_CODE (tss) == set)
2280 : 721 : return OMP_TSS_TRAIT_SELECTORS (tss);
2281 : : return NULL_TREE;
2282 : : }
2283 : :
2284 : : /* Map string S onto a trait selector set code. */
2285 : : enum omp_tss_code
2286 : 2272 : omp_lookup_tss_code (const char * s)
2287 : : {
2288 : 6326 : for (int i = 0; i < OMP_TRAIT_SET_LAST; i++)
2289 : 6303 : if (strcmp (s, omp_tss_map[i]) == 0)
2290 : : return (enum omp_tss_code) i;
2291 : : return OMP_TRAIT_SET_INVALID;
2292 : : }
2293 : :
2294 : : /* Map string S onto a trait selector code for set SET. */
2295 : : enum omp_ts_code
2296 : 2705 : omp_lookup_ts_code (enum omp_tss_code set, const char *s)
2297 : : {
2298 : 2705 : unsigned int mask = 1 << set;
2299 : 30932 : for (int i = 0; i < OMP_TRAIT_LAST; i++)
2300 : 30843 : if ((mask & omp_ts_map[i].tss_mask) != 0
2301 : 7192 : && strcmp (s, omp_ts_map[i].name) == 0)
2302 : : return (enum omp_ts_code) i;
2303 : : return OMP_TRAIT_INVALID;
2304 : : }
2305 : :
2306 : : /* Needs to be a GC-friendly widest_int variant, but precision is
2307 : : desirable to be the same on all targets. */
2308 : : typedef generic_wide_int <fixed_wide_int_storage <1024> > score_wide_int;
2309 : :
2310 : : /* Compute *SCORE for context selector CTX. Return true if the score
2311 : : would be different depending on whether it is a declare simd clone or
2312 : : not. DECLARE_SIMD should be true for the case when it would be
2313 : : a declare simd clone. */
2314 : :
2315 : : static bool
2316 : 369 : omp_context_compute_score (tree ctx, score_wide_int *score, bool declare_simd)
2317 : : {
2318 : 369 : tree selectors
2319 : 369 : = omp_get_context_selector_list (ctx, OMP_TRAIT_SET_CONSTRUCT);
2320 : 369 : bool has_kind = omp_get_context_selector (ctx, OMP_TRAIT_SET_DEVICE,
2321 : 369 : OMP_TRAIT_DEVICE_KIND);
2322 : 369 : bool has_arch = omp_get_context_selector (ctx, OMP_TRAIT_SET_DEVICE,
2323 : 369 : OMP_TRAIT_DEVICE_ARCH);
2324 : 369 : bool has_isa = omp_get_context_selector (ctx, OMP_TRAIT_SET_DEVICE,
2325 : 369 : OMP_TRAIT_DEVICE_ISA);
2326 : 369 : bool ret = false;
2327 : 369 : *score = 1;
2328 : 826 : for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
2329 : 457 : if (OMP_TSS_TRAIT_SELECTORS (tss) != selectors)
2330 : 692 : for (tree ts = OMP_TSS_TRAIT_SELECTORS (tss); ts; ts = TREE_CHAIN (ts))
2331 : : {
2332 : 499 : tree s = OMP_TS_SCORE (ts);
2333 : 153 : if (s && TREE_CODE (s) == INTEGER_CST)
2334 : 306 : *score += score_wide_int::from (wi::to_wide (s),
2335 : 306 : TYPE_SIGN (TREE_TYPE (s)));
2336 : : }
2337 : :
2338 : 369 : if (selectors || has_kind || has_arch || has_isa)
2339 : : {
2340 : 233 : int nconstructs = list_length (selectors);
2341 : 233 : enum tree_code *constructs = NULL;
2342 : 233 : if (nconstructs)
2343 : : {
2344 : 111 : constructs
2345 : 111 : = (enum tree_code *) alloca (nconstructs
2346 : : * sizeof (enum tree_code));
2347 : 111 : omp_construct_traits_to_codes (selectors, nconstructs, constructs);
2348 : : }
2349 : 233 : int *scores
2350 : 233 : = (int *) alloca ((2 * nconstructs + 2) * sizeof (int));
2351 : 233 : if (omp_construct_selector_matches (constructs, nconstructs, scores)
2352 : : == 2)
2353 : 102 : ret = true;
2354 : 233 : int b = declare_simd ? nconstructs + 1 : 0;
2355 : 233 : if (scores[b + nconstructs] + 4U < score->get_precision ())
2356 : : {
2357 : 398 : for (int n = 0; n < nconstructs; ++n)
2358 : : {
2359 : 165 : if (scores[b + n] < 0)
2360 : : {
2361 : 0 : *score = -1;
2362 : 0 : return ret;
2363 : : }
2364 : 165 : *score += wi::shifted_mask <score_wide_int> (scores[b + n], 1, false);
2365 : : }
2366 : 233 : if (has_kind)
2367 : 136 : *score += wi::shifted_mask <score_wide_int> (scores[b + nconstructs],
2368 : 68 : 1, false);
2369 : 233 : if (has_arch)
2370 : 10 : *score += wi::shifted_mask <score_wide_int> (scores[b + nconstructs] + 1,
2371 : 5 : 1, false);
2372 : 233 : if (has_isa)
2373 : 98 : *score += wi::shifted_mask <score_wide_int> (scores[b + nconstructs] + 2,
2374 : 49 : 1, false);
2375 : : }
2376 : : else /* FIXME: Implement this. */
2377 : 0 : gcc_unreachable ();
2378 : : }
2379 : : return ret;
2380 : : }
2381 : :
2382 : : /* Class describing a single variant. */
2383 : 82 : struct GTY(()) omp_declare_variant_entry {
2384 : : /* NODE of the variant. */
2385 : : cgraph_node *variant;
2386 : : /* Score if not in declare simd clone. */
2387 : : score_wide_int score;
2388 : : /* Score if in declare simd clone. */
2389 : : score_wide_int score_in_declare_simd_clone;
2390 : : /* Context selector for the variant. */
2391 : : tree ctx;
2392 : : /* True if the context selector is known to match already. */
2393 : : bool matches;
2394 : : };
2395 : :
2396 : : /* Class describing a function with variants. */
2397 : : struct GTY((for_user)) omp_declare_variant_base_entry {
2398 : : /* NODE of the base function. */
2399 : : cgraph_node *base;
2400 : : /* NODE of the artificial function created for the deferred variant
2401 : : resolution. */
2402 : : cgraph_node *node;
2403 : : /* Vector of the variants. */
2404 : : vec<omp_declare_variant_entry, va_gc> *variants;
2405 : : };
2406 : :
2407 : : struct omp_declare_variant_hasher
2408 : : : ggc_ptr_hash<omp_declare_variant_base_entry> {
2409 : : static hashval_t hash (omp_declare_variant_base_entry *);
2410 : : static bool equal (omp_declare_variant_base_entry *,
2411 : : omp_declare_variant_base_entry *);
2412 : : };
2413 : :
2414 : : hashval_t
2415 : 28 : omp_declare_variant_hasher::hash (omp_declare_variant_base_entry *x)
2416 : : {
2417 : 28 : inchash::hash hstate;
2418 : 28 : hstate.add_int (DECL_UID (x->base->decl));
2419 : 28 : hstate.add_int (x->variants->length ());
2420 : 28 : omp_declare_variant_entry *variant;
2421 : 28 : unsigned int i;
2422 : 94 : FOR_EACH_VEC_SAFE_ELT (x->variants, i, variant)
2423 : : {
2424 : 66 : hstate.add_int (DECL_UID (variant->variant->decl));
2425 : 66 : hstate.add_wide_int (variant->score);
2426 : 66 : hstate.add_wide_int (variant->score_in_declare_simd_clone);
2427 : 66 : hstate.add_ptr (variant->ctx);
2428 : 66 : hstate.add_int (variant->matches);
2429 : : }
2430 : 28 : return hstate.end ();
2431 : : }
2432 : :
2433 : : bool
2434 : 11 : omp_declare_variant_hasher::equal (omp_declare_variant_base_entry *x,
2435 : : omp_declare_variant_base_entry *y)
2436 : : {
2437 : 11 : if (x->base != y->base
2438 : 11 : || x->variants->length () != y->variants->length ())
2439 : : return false;
2440 : : omp_declare_variant_entry *variant;
2441 : : unsigned int i;
2442 : 36 : FOR_EACH_VEC_SAFE_ELT (x->variants, i, variant)
2443 : 26 : if (variant->variant != (*y->variants)[i].variant
2444 : 26 : || variant->score != (*y->variants)[i].score
2445 : 52 : || (variant->score_in_declare_simd_clone
2446 : 26 : != (*y->variants)[i].score_in_declare_simd_clone)
2447 : 26 : || variant->ctx != (*y->variants)[i].ctx
2448 : 52 : || variant->matches != (*y->variants)[i].matches)
2449 : : return false;
2450 : : return true;
2451 : : }
2452 : :
2453 : : static GTY(()) hash_table<omp_declare_variant_hasher> *omp_declare_variants;
2454 : :
2455 : : struct omp_declare_variant_alt_hasher
2456 : : : ggc_ptr_hash<omp_declare_variant_base_entry> {
2457 : : static hashval_t hash (omp_declare_variant_base_entry *);
2458 : : static bool equal (omp_declare_variant_base_entry *,
2459 : : omp_declare_variant_base_entry *);
2460 : : };
2461 : :
2462 : : hashval_t
2463 : 0 : omp_declare_variant_alt_hasher::hash (omp_declare_variant_base_entry *x)
2464 : : {
2465 : 0 : return DECL_UID (x->node->decl);
2466 : : }
2467 : :
2468 : : bool
2469 : 121 : omp_declare_variant_alt_hasher::equal (omp_declare_variant_base_entry *x,
2470 : : omp_declare_variant_base_entry *y)
2471 : : {
2472 : 121 : return x->node == y->node;
2473 : : }
2474 : :
2475 : : static GTY(()) hash_table<omp_declare_variant_alt_hasher>
2476 : : *omp_declare_variant_alt;
2477 : :
2478 : : /* Try to resolve declare variant after gimplification. */
2479 : :
2480 : : static tree
2481 : 145 : omp_resolve_late_declare_variant (tree alt)
2482 : : {
2483 : 145 : cgraph_node *node = cgraph_node::get (alt);
2484 : 145 : cgraph_node *cur_node = cgraph_node::get (cfun->decl);
2485 : 145 : if (node == NULL
2486 : 145 : || !node->declare_variant_alt
2487 : 117 : || !cfun->after_inlining)
2488 : : return alt;
2489 : :
2490 : 117 : omp_declare_variant_base_entry entry;
2491 : 117 : entry.base = NULL;
2492 : 117 : entry.node = node;
2493 : 117 : entry.variants = NULL;
2494 : 117 : omp_declare_variant_base_entry *entryp
2495 : 117 : = omp_declare_variant_alt->find_with_hash (&entry, DECL_UID (alt));
2496 : :
2497 : 117 : unsigned int i, j;
2498 : 117 : omp_declare_variant_entry *varentry1, *varentry2;
2499 : 117 : auto_vec <bool, 16> matches;
2500 : 117 : unsigned int nmatches = 0;
2501 : 441 : FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry1)
2502 : : {
2503 : 333 : if (varentry1->matches)
2504 : : {
2505 : : /* This has been checked to be ok already. */
2506 : 216 : matches.safe_push (true);
2507 : 216 : nmatches++;
2508 : 216 : continue;
2509 : : }
2510 : 117 : switch (omp_context_selector_matches (varentry1->ctx))
2511 : : {
2512 : 84 : case 0:
2513 : 84 : matches.safe_push (false);
2514 : 84 : break;
2515 : : case -1:
2516 : : return alt;
2517 : 24 : default:
2518 : 24 : matches.safe_push (true);
2519 : 24 : nmatches++;
2520 : 24 : break;
2521 : : }
2522 : : }
2523 : :
2524 : 108 : if (nmatches == 0)
2525 : 0 : return entryp->base->decl;
2526 : :
2527 : : /* A context selector that is a strict subset of another context selector
2528 : : has a score of zero. */
2529 : 432 : FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry1)
2530 : 324 : if (matches[i])
2531 : : {
2532 : 480 : for (j = i + 1;
2533 : 804 : vec_safe_iterate (entryp->variants, j, &varentry2); ++j)
2534 : 240 : if (matches[j])
2535 : : {
2536 : 156 : int r = omp_context_selector_compare (varentry1->ctx,
2537 : : varentry2->ctx);
2538 : 156 : if (r == -1)
2539 : : {
2540 : : /* ctx1 is a strict subset of ctx2, ignore ctx1. */
2541 : 0 : matches[i] = false;
2542 : 0 : break;
2543 : : }
2544 : 156 : else if (r == 1)
2545 : : /* ctx2 is a strict subset of ctx1, remove ctx2. */
2546 : 0 : matches[j] = false;
2547 : : }
2548 : : }
2549 : :
2550 : 108 : score_wide_int max_score = -1;
2551 : 108 : varentry2 = NULL;
2552 : 432 : FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry1)
2553 : 324 : if (matches[i])
2554 : : {
2555 : 480 : score_wide_int score
2556 : 240 : = (cur_node->simdclone ? varentry1->score_in_declare_simd_clone
2557 : : : varentry1->score);
2558 : 240 : if (score > max_score)
2559 : : {
2560 : 162 : max_score = score;
2561 : 162 : varentry2 = varentry1;
2562 : : }
2563 : : }
2564 : 108 : return varentry2->variant->decl;
2565 : 117 : }
2566 : :
2567 : : /* Hook to adjust hash tables on cgraph_node removal. */
2568 : :
2569 : : static void
2570 : 15 : omp_declare_variant_remove_hook (struct cgraph_node *node, void *)
2571 : : {
2572 : 15 : if (!node->declare_variant_alt)
2573 : : return;
2574 : :
2575 : : /* Drop this hash table completely. */
2576 : 2 : omp_declare_variants = NULL;
2577 : : /* And remove node from the other hash table. */
2578 : 2 : if (omp_declare_variant_alt)
2579 : : {
2580 : 2 : omp_declare_variant_base_entry entry;
2581 : 2 : entry.base = NULL;
2582 : 2 : entry.node = node;
2583 : 2 : entry.variants = NULL;
2584 : 2 : omp_declare_variant_alt->remove_elt_with_hash (&entry,
2585 : 2 : DECL_UID (node->decl));
2586 : : }
2587 : : }
2588 : :
2589 : : /* Try to resolve declare variant, return the variant decl if it should
2590 : : be used instead of base, or base otherwise. */
2591 : :
2592 : : tree
2593 : 196518 : omp_resolve_declare_variant (tree base)
2594 : : {
2595 : 196518 : tree variant1 = NULL_TREE, variant2 = NULL_TREE;
2596 : 196518 : if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
2597 : 145 : return omp_resolve_late_declare_variant (base);
2598 : :
2599 : 196373 : if (omp_has_novariants () == 1)
2600 : : return base;
2601 : :
2602 : 196299 : auto_vec <tree, 16> variants;
2603 : 196299 : auto_vec <bool, 16> defer;
2604 : 196299 : bool any_deferred = false;
2605 : 197596 : for (tree attr = DECL_ATTRIBUTES (base); attr; attr = TREE_CHAIN (attr))
2606 : : {
2607 : 102806 : attr = lookup_attribute ("omp declare variant base", attr);
2608 : 102806 : if (attr == NULL_TREE)
2609 : : break;
2610 : 1333 : if (TREE_CODE (TREE_PURPOSE (TREE_VALUE (attr))) != FUNCTION_DECL)
2611 : 0 : continue;
2612 : 1333 : cgraph_node *node = cgraph_node::get (base);
2613 : : /* If this is already a magic decl created by this function,
2614 : : don't process it again. */
2615 : 1333 : if (node && node->declare_variant_alt)
2616 : 36 : return base;
2617 : 1297 : switch (omp_context_selector_matches (TREE_VALUE (TREE_VALUE (attr))))
2618 : : {
2619 : : case 0:
2620 : : /* No match, ignore. */
2621 : : break;
2622 : 37 : case -1:
2623 : : /* Needs to be deferred. */
2624 : 37 : any_deferred = true;
2625 : 37 : variants.safe_push (attr);
2626 : 37 : defer.safe_push (true);
2627 : 37 : break;
2628 : 982 : default:
2629 : 982 : variants.safe_push (attr);
2630 : 982 : defer.safe_push (false);
2631 : 982 : break;
2632 : : }
2633 : : }
2634 : 196299 : if (variants.length () == 0)
2635 : : return base;
2636 : :
2637 : 747 : if (any_deferred)
2638 : : {
2639 : 29 : score_wide_int max_score1 = 0;
2640 : 29 : score_wide_int max_score2 = 0;
2641 : 29 : bool first = true;
2642 : 29 : unsigned int i;
2643 : 29 : tree attr1, attr2;
2644 : 29 : omp_declare_variant_base_entry entry;
2645 : 29 : entry.base = cgraph_node::get_create (base);
2646 : 29 : entry.node = NULL;
2647 : 58 : vec_alloc (entry.variants, variants.length ());
2648 : 105 : FOR_EACH_VEC_ELT (variants, i, attr1)
2649 : : {
2650 : 76 : score_wide_int score1;
2651 : 76 : score_wide_int score2;
2652 : 76 : bool need_two;
2653 : 76 : tree ctx = TREE_VALUE (TREE_VALUE (attr1));
2654 : 76 : need_two = omp_context_compute_score (ctx, &score1, false);
2655 : 76 : if (need_two)
2656 : 51 : omp_context_compute_score (ctx, &score2, true);
2657 : : else
2658 : 25 : score2 = score1;
2659 : 76 : if (first)
2660 : : {
2661 : 29 : first = false;
2662 : 29 : max_score1 = score1;
2663 : 29 : max_score2 = score2;
2664 : 29 : if (!defer[i])
2665 : : {
2666 : 8 : variant1 = attr1;
2667 : 8 : variant2 = attr1;
2668 : : }
2669 : : }
2670 : : else
2671 : : {
2672 : 47 : if (max_score1 == score1)
2673 : : variant1 = NULL_TREE;
2674 : 38 : else if (score1 > max_score1)
2675 : : {
2676 : 19 : max_score1 = score1;
2677 : 19 : variant1 = defer[i] ? NULL_TREE : attr1;
2678 : : }
2679 : 47 : if (max_score2 == score2)
2680 : : variant2 = NULL_TREE;
2681 : 47 : else if (score2 > max_score2)
2682 : : {
2683 : 19 : max_score2 = score2;
2684 : 19 : variant2 = defer[i] ? NULL_TREE : attr1;
2685 : : }
2686 : : }
2687 : 76 : omp_declare_variant_entry varentry;
2688 : 76 : varentry.variant
2689 : 76 : = cgraph_node::get_create (TREE_PURPOSE (TREE_VALUE (attr1)));
2690 : 76 : varentry.score = score1;
2691 : 76 : varentry.score_in_declare_simd_clone = score2;
2692 : 76 : varentry.ctx = ctx;
2693 : 76 : varentry.matches = !defer[i];
2694 : 76 : entry.variants->quick_push (varentry);
2695 : : }
2696 : :
2697 : : /* If there is a clear winner variant with the score which is not
2698 : : deferred, verify it is not a strict subset of any other context
2699 : : selector and if it is not, it is the best alternative no matter
2700 : : whether the others do or don't match. */
2701 : 29 : if (variant1 && variant1 == variant2)
2702 : : {
2703 : 5 : tree ctx1 = TREE_VALUE (TREE_VALUE (variant1));
2704 : 25 : FOR_EACH_VEC_ELT (variants, i, attr2)
2705 : : {
2706 : 20 : if (attr2 == variant1)
2707 : 5 : continue;
2708 : 15 : tree ctx2 = TREE_VALUE (TREE_VALUE (attr2));
2709 : 15 : int r = omp_context_selector_compare (ctx1, ctx2);
2710 : 15 : if (r == -1)
2711 : : {
2712 : : /* The winner is a strict subset of ctx2, can't
2713 : : decide now. */
2714 : : variant1 = NULL_TREE;
2715 : : break;
2716 : : }
2717 : : }
2718 : 5 : if (variant1)
2719 : : {
2720 : 5 : vec_free (entry.variants);
2721 : 5 : return TREE_PURPOSE (TREE_VALUE (variant1));
2722 : : }
2723 : : }
2724 : :
2725 : 24 : static struct cgraph_node_hook_list *node_removal_hook_holder;
2726 : 24 : if (!node_removal_hook_holder)
2727 : 10 : node_removal_hook_holder
2728 : 10 : = symtab->add_cgraph_removal_hook (omp_declare_variant_remove_hook,
2729 : : NULL);
2730 : :
2731 : 24 : if (omp_declare_variants == NULL)
2732 : 10 : omp_declare_variants
2733 : 10 : = hash_table<omp_declare_variant_hasher>::create_ggc (64);
2734 : 24 : omp_declare_variant_base_entry **slot
2735 : 24 : = omp_declare_variants->find_slot (&entry, INSERT);
2736 : 24 : if (*slot != NULL)
2737 : : {
2738 : 10 : vec_free (entry.variants);
2739 : 10 : return (*slot)->node->decl;
2740 : : }
2741 : :
2742 : 14 : *slot = ggc_cleared_alloc<omp_declare_variant_base_entry> ();
2743 : 14 : (*slot)->base = entry.base;
2744 : 14 : (*slot)->node = entry.base;
2745 : 14 : (*slot)->variants = entry.variants;
2746 : 14 : tree alt = build_decl (DECL_SOURCE_LOCATION (base), FUNCTION_DECL,
2747 : 14 : DECL_NAME (base), TREE_TYPE (base));
2748 : 14 : if (DECL_ASSEMBLER_NAME_SET_P (base))
2749 : 4 : SET_DECL_ASSEMBLER_NAME (alt, DECL_ASSEMBLER_NAME (base));
2750 : 14 : DECL_ARTIFICIAL (alt) = 1;
2751 : 14 : DECL_IGNORED_P (alt) = 1;
2752 : 14 : TREE_STATIC (alt) = 1;
2753 : 14 : tree attributes = DECL_ATTRIBUTES (base);
2754 : 14 : if (lookup_attribute ("noipa", attributes) == NULL)
2755 : : {
2756 : 14 : attributes = tree_cons (get_identifier ("noipa"), NULL, attributes);
2757 : 14 : if (lookup_attribute ("noinline", attributes) == NULL)
2758 : 14 : attributes = tree_cons (get_identifier ("noinline"), NULL,
2759 : : attributes);
2760 : 14 : if (lookup_attribute ("noclone", attributes) == NULL)
2761 : 14 : attributes = tree_cons (get_identifier ("noclone"), NULL,
2762 : : attributes);
2763 : 14 : if (lookup_attribute ("no_icf", attributes) == NULL)
2764 : 14 : attributes = tree_cons (get_identifier ("no_icf"), NULL,
2765 : : attributes);
2766 : : }
2767 : 14 : DECL_ATTRIBUTES (alt) = attributes;
2768 : 14 : DECL_INITIAL (alt) = error_mark_node;
2769 : 14 : (*slot)->node = cgraph_node::create (alt);
2770 : 14 : (*slot)->node->declare_variant_alt = 1;
2771 : 14 : (*slot)->node->create_reference (entry.base, IPA_REF_ADDR);
2772 : 14 : omp_declare_variant_entry *varentry;
2773 : 58 : FOR_EACH_VEC_SAFE_ELT (entry.variants, i, varentry)
2774 : 30 : (*slot)->node->create_reference (varentry->variant, IPA_REF_ADDR);
2775 : 14 : if (omp_declare_variant_alt == NULL)
2776 : 10 : omp_declare_variant_alt
2777 : 10 : = hash_table<omp_declare_variant_alt_hasher>::create_ggc (64);
2778 : 14 : *omp_declare_variant_alt->find_slot_with_hash (*slot, DECL_UID (alt),
2779 : 14 : INSERT) = *slot;
2780 : 14 : return alt;
2781 : : }
2782 : :
2783 : 718 : if (variants.length () == 1)
2784 : 568 : return TREE_PURPOSE (TREE_VALUE (variants[0]));
2785 : :
2786 : : /* A context selector that is a strict subset of another context selector
2787 : : has a score of zero. */
2788 : : tree attr1, attr2;
2789 : : unsigned int i, j;
2790 : 525 : FOR_EACH_VEC_ELT (variants, i, attr1)
2791 : 375 : if (attr1)
2792 : : {
2793 : 331 : tree ctx1 = TREE_VALUE (TREE_VALUE (attr1));
2794 : 968 : FOR_EACH_VEC_ELT_FROM (variants, j, attr2, i + 1)
2795 : 297 : if (attr2)
2796 : : {
2797 : 297 : tree ctx2 = TREE_VALUE (TREE_VALUE (attr2));
2798 : 297 : int r = omp_context_selector_compare (ctx1, ctx2);
2799 : 297 : if (r == -1)
2800 : : {
2801 : : /* ctx1 is a strict subset of ctx2, remove
2802 : : attr1 from the vector. */
2803 : 35 : variants[i] = NULL_TREE;
2804 : 35 : break;
2805 : : }
2806 : 262 : else if (r == 1)
2807 : : /* ctx2 is a strict subset of ctx1, remove attr2
2808 : : from the vector. */
2809 : 44 : variants[j] = NULL_TREE;
2810 : : }
2811 : : }
2812 : 150 : score_wide_int max_score1 = 0;
2813 : 150 : score_wide_int max_score2 = 0;
2814 : 150 : bool first = true;
2815 : 525 : FOR_EACH_VEC_ELT (variants, i, attr1)
2816 : 375 : if (attr1)
2817 : : {
2818 : 296 : if (variant1)
2819 : : {
2820 : 146 : score_wide_int score1;
2821 : 146 : score_wide_int score2;
2822 : 146 : bool need_two;
2823 : 146 : tree ctx;
2824 : 146 : if (first)
2825 : : {
2826 : 96 : first = false;
2827 : 96 : ctx = TREE_VALUE (TREE_VALUE (variant1));
2828 : 96 : need_two = omp_context_compute_score (ctx, &max_score1, false);
2829 : 96 : if (need_two)
2830 : 0 : omp_context_compute_score (ctx, &max_score2, true);
2831 : : else
2832 : 96 : max_score2 = max_score1;
2833 : : }
2834 : 146 : ctx = TREE_VALUE (TREE_VALUE (attr1));
2835 : 146 : need_two = omp_context_compute_score (ctx, &score1, false);
2836 : 146 : if (need_two)
2837 : 0 : omp_context_compute_score (ctx, &score2, true);
2838 : : else
2839 : 146 : score2 = score1;
2840 : 146 : if (score1 > max_score1)
2841 : : {
2842 : 51 : max_score1 = score1;
2843 : 51 : variant1 = attr1;
2844 : : }
2845 : 146 : if (score2 > max_score2)
2846 : : {
2847 : 51 : max_score2 = score2;
2848 : 51 : variant2 = attr1;
2849 : : }
2850 : : }
2851 : : else
2852 : : {
2853 : : variant1 = attr1;
2854 : : variant2 = attr1;
2855 : : }
2856 : : }
2857 : : /* If there is a disagreement on which variant has the highest score
2858 : : depending on whether it will be in a declare simd clone or not,
2859 : : punt for now and defer until after IPA where we will know that. */
2860 : 150 : return ((variant1 && variant1 == variant2)
2861 : 300 : ? TREE_PURPOSE (TREE_VALUE (variant1)) : base);
2862 : 196299 : }
2863 : :
2864 : : void
2865 : 2 : omp_lto_output_declare_variant_alt (lto_simple_output_block *ob,
2866 : : cgraph_node *node,
2867 : : lto_symtab_encoder_t encoder)
2868 : : {
2869 : 2 : gcc_assert (node->declare_variant_alt);
2870 : :
2871 : 2 : omp_declare_variant_base_entry entry;
2872 : 2 : entry.base = NULL;
2873 : 2 : entry.node = node;
2874 : 2 : entry.variants = NULL;
2875 : 2 : omp_declare_variant_base_entry *entryp
2876 : 2 : = omp_declare_variant_alt->find_with_hash (&entry, DECL_UID (node->decl));
2877 : 2 : gcc_assert (entryp);
2878 : :
2879 : 2 : int nbase = lto_symtab_encoder_lookup (encoder, entryp->base);
2880 : 2 : gcc_assert (nbase != LCC_NOT_FOUND);
2881 : 2 : streamer_write_hwi_stream (ob->main_stream, nbase);
2882 : :
2883 : 2 : streamer_write_hwi_stream (ob->main_stream, entryp->variants->length ());
2884 : :
2885 : 2 : unsigned int i;
2886 : 2 : omp_declare_variant_entry *varentry;
2887 : 8 : FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry)
2888 : : {
2889 : 6 : int nvar = lto_symtab_encoder_lookup (encoder, varentry->variant);
2890 : 6 : gcc_assert (nvar != LCC_NOT_FOUND);
2891 : 6 : streamer_write_hwi_stream (ob->main_stream, nvar);
2892 : :
2893 : 6 : for (score_wide_int *w = &varentry->score; ;
2894 : : w = &varentry->score_in_declare_simd_clone)
2895 : : {
2896 : 12 : unsigned len = w->get_len ();
2897 : 12 : streamer_write_hwi_stream (ob->main_stream, len);
2898 : 12 : const HOST_WIDE_INT *val = w->get_val ();
2899 : 24 : for (unsigned j = 0; j < len; j++)
2900 : 12 : streamer_write_hwi_stream (ob->main_stream, val[j]);
2901 : 12 : if (w == &varentry->score_in_declare_simd_clone)
2902 : : break;
2903 : : }
2904 : :
2905 : 6 : HOST_WIDE_INT cnt = -1;
2906 : 6 : HOST_WIDE_INT i = varentry->matches ? 1 : 0;
2907 : 6 : for (tree attr = DECL_ATTRIBUTES (entryp->base->decl);
2908 : 12 : attr; attr = TREE_CHAIN (attr), i += 2)
2909 : : {
2910 : 12 : attr = lookup_attribute ("omp declare variant base", attr);
2911 : 12 : if (attr == NULL_TREE)
2912 : : break;
2913 : :
2914 : 12 : if (varentry->ctx == TREE_VALUE (TREE_VALUE (attr)))
2915 : : {
2916 : : cnt = i;
2917 : : break;
2918 : : }
2919 : : }
2920 : :
2921 : 6 : gcc_assert (cnt != -1);
2922 : 6 : streamer_write_hwi_stream (ob->main_stream, cnt);
2923 : : }
2924 : 2 : }
2925 : :
2926 : : void
2927 : 2 : omp_lto_input_declare_variant_alt (lto_input_block *ib, cgraph_node *node,
2928 : : vec<symtab_node *> nodes)
2929 : : {
2930 : 2 : gcc_assert (node->declare_variant_alt);
2931 : 2 : omp_declare_variant_base_entry *entryp
2932 : 2 : = ggc_cleared_alloc<omp_declare_variant_base_entry> ();
2933 : 2 : entryp->base = dyn_cast<cgraph_node *> (nodes[streamer_read_hwi (ib)]);
2934 : 2 : entryp->node = node;
2935 : 2 : unsigned int len = streamer_read_hwi (ib);
2936 : 2 : vec_alloc (entryp->variants, len);
2937 : :
2938 : 8 : for (unsigned int i = 0; i < len; i++)
2939 : : {
2940 : 6 : omp_declare_variant_entry varentry;
2941 : 6 : varentry.variant
2942 : 6 : = dyn_cast<cgraph_node *> (nodes[streamer_read_hwi (ib)]);
2943 : 6 : for (score_wide_int *w = &varentry.score; ;
2944 : 6 : w = &varentry.score_in_declare_simd_clone)
2945 : : {
2946 : 12 : unsigned len2 = streamer_read_hwi (ib);
2947 : 12 : HOST_WIDE_INT arr[WIDE_INT_MAX_HWIS (1024)];
2948 : 12 : gcc_assert (len2 <= WIDE_INT_MAX_HWIS (1024));
2949 : 24 : for (unsigned int j = 0; j < len2; j++)
2950 : 12 : arr[j] = streamer_read_hwi (ib);
2951 : 12 : *w = score_wide_int::from_array (arr, len2, true);
2952 : 12 : if (w == &varentry.score_in_declare_simd_clone)
2953 : : break;
2954 : 6 : }
2955 : :
2956 : 6 : HOST_WIDE_INT cnt = streamer_read_hwi (ib);
2957 : 6 : HOST_WIDE_INT j = 0;
2958 : 6 : varentry.ctx = NULL_TREE;
2959 : 6 : varentry.matches = (cnt & 1) ? true : false;
2960 : 6 : cnt &= ~HOST_WIDE_INT_1;
2961 : 6 : for (tree attr = DECL_ATTRIBUTES (entryp->base->decl);
2962 : 12 : attr; attr = TREE_CHAIN (attr), j += 2)
2963 : : {
2964 : 12 : attr = lookup_attribute ("omp declare variant base", attr);
2965 : 12 : if (attr == NULL_TREE)
2966 : : break;
2967 : :
2968 : 12 : if (cnt == j)
2969 : : {
2970 : 6 : varentry.ctx = TREE_VALUE (TREE_VALUE (attr));
2971 : 6 : break;
2972 : : }
2973 : : }
2974 : 6 : gcc_assert (varentry.ctx != NULL_TREE);
2975 : 6 : entryp->variants->quick_push (varentry);
2976 : : }
2977 : 2 : if (omp_declare_variant_alt == NULL)
2978 : 2 : omp_declare_variant_alt
2979 : 2 : = hash_table<omp_declare_variant_alt_hasher>::create_ggc (64);
2980 : 2 : *omp_declare_variant_alt->find_slot_with_hash (entryp, DECL_UID (node->decl),
2981 : 2 : INSERT) = entryp;
2982 : 2 : }
2983 : :
2984 : : /* Encode an oacc launch argument. This matches the GOMP_LAUNCH_PACK
2985 : : macro on gomp-constants.h. We do not check for overflow. */
2986 : :
2987 : : tree
2988 : 10245 : oacc_launch_pack (unsigned code, tree device, unsigned op)
2989 : : {
2990 : 10245 : tree res;
2991 : :
2992 : 10245 : res = build_int_cst (unsigned_type_node, GOMP_LAUNCH_PACK (code, 0, op));
2993 : 10245 : if (device)
2994 : : {
2995 : 0 : device = fold_build2 (LSHIFT_EXPR, unsigned_type_node,
2996 : : device, build_int_cst (unsigned_type_node,
2997 : : GOMP_LAUNCH_DEVICE_SHIFT));
2998 : 0 : res = fold_build2 (BIT_IOR_EXPR, unsigned_type_node, res, device);
2999 : : }
3000 : 10245 : return res;
3001 : : }
3002 : :
3003 : : /* Openacc compute grid dimension clauses are converted to an attribute
3004 : : attached to the function. This permits the target-side code to (a) massage
3005 : : the dimensions, (b) emit that data and (c) optimize. Non-constant
3006 : : dimensions are pushed onto ARGS.
3007 : :
3008 : : The attribute value is a TREE_LIST. A set of dimensions is
3009 : : represented as a list of INTEGER_CST. Those that are runtime
3010 : : exprs are represented as an INTEGER_CST of zero.
3011 : :
3012 : : TODO: Normally the attribute will just contain a single such list. If
3013 : : however it contains a list of lists, this will represent the use of
3014 : : device_type. Each member of the outer list is an assoc list of
3015 : : dimensions, keyed by the device type. The first entry will be the
3016 : : default. Well, that's the plan. */
3017 : :
3018 : : /* Replace any existing oacc fn attribute in ATTRIBS with updated
3019 : : dimensions. */
3020 : :
3021 : : tree
3022 : 19566 : oacc_replace_fn_attrib_attr (tree attribs, tree dims)
3023 : : {
3024 : 19566 : tree ident = get_identifier (OACC_FN_ATTRIB);
3025 : :
3026 : : /* If we happen to be present as the first attrib, drop it. */
3027 : 38076 : if (attribs && TREE_PURPOSE (attribs) == ident)
3028 : 9384 : attribs = TREE_CHAIN (attribs);
3029 : 19566 : return tree_cons (ident, dims, attribs);
3030 : : }
3031 : :
3032 : : /* Replace any existing oacc fn attribute on FN with updated
3033 : : dimensions. */
3034 : :
3035 : : void
3036 : 19228 : oacc_replace_fn_attrib (tree fn, tree dims)
3037 : : {
3038 : 19228 : DECL_ATTRIBUTES (fn)
3039 : 19228 : = oacc_replace_fn_attrib_attr (DECL_ATTRIBUTES (fn), dims);
3040 : 19228 : }
3041 : :
3042 : : /* Scan CLAUSES for launch dimensions and attach them to the oacc
3043 : : function attribute. Push any that are non-constant onto the ARGS
3044 : : list, along with an appropriate GOMP_LAUNCH_DIM tag. */
3045 : :
3046 : : void
3047 : 9707 : oacc_set_fn_attrib (tree fn, tree clauses, vec<tree> *args)
3048 : : {
3049 : : /* Must match GOMP_DIM ordering. */
3050 : 9707 : static const omp_clause_code ids[]
3051 : : = { OMP_CLAUSE_NUM_GANGS, OMP_CLAUSE_NUM_WORKERS,
3052 : : OMP_CLAUSE_VECTOR_LENGTH };
3053 : 9707 : unsigned ix;
3054 : 9707 : tree dims[GOMP_DIM_MAX];
3055 : :
3056 : 9707 : tree attr = NULL_TREE;
3057 : 9707 : unsigned non_const = 0;
3058 : :
3059 : 38828 : for (ix = GOMP_DIM_MAX; ix--;)
3060 : : {
3061 : 29121 : tree clause = omp_find_clause (clauses, ids[ix]);
3062 : 29121 : tree dim = NULL_TREE;
3063 : :
3064 : 29121 : if (clause)
3065 : 3287 : dim = OMP_CLAUSE_EXPR (clause, ids[ix]);
3066 : 29121 : dims[ix] = dim;
3067 : 29121 : if (dim && TREE_CODE (dim) != INTEGER_CST)
3068 : : {
3069 : 157 : dim = integer_zero_node;
3070 : 157 : non_const |= GOMP_DIM_MASK (ix);
3071 : : }
3072 : 29121 : attr = tree_cons (NULL_TREE, dim, attr);
3073 : : }
3074 : :
3075 : 9707 : oacc_replace_fn_attrib (fn, attr);
3076 : :
3077 : 9707 : if (non_const)
3078 : : {
3079 : : /* Push a dynamic argument set. */
3080 : 101 : args->safe_push (oacc_launch_pack (GOMP_LAUNCH_DIM,
3081 : : NULL_TREE, non_const));
3082 : 404 : for (unsigned ix = 0; ix != GOMP_DIM_MAX; ix++)
3083 : 303 : if (non_const & GOMP_DIM_MASK (ix))
3084 : 157 : args->safe_push (dims[ix]);
3085 : : }
3086 : 9707 : }
3087 : :
3088 : : /* Verify OpenACC routine clauses.
3089 : :
3090 : : Returns 0 if FNDECL should be marked with an OpenACC 'routine' directive, 1
3091 : : if it has already been marked in compatible way, and -1 if incompatible.
3092 : : Upon returning, the chain of clauses will contain exactly one clause
3093 : : specifying the level of parallelism. */
3094 : :
3095 : : int
3096 : 1199 : oacc_verify_routine_clauses (tree fndecl, tree *clauses, location_t loc,
3097 : : const char *routine_str)
3098 : : {
3099 : 1199 : tree c_level = NULL_TREE;
3100 : 1199 : tree c_nohost = NULL_TREE;
3101 : 1199 : tree c_p = NULL_TREE;
3102 : 3612 : for (tree c = *clauses; c; c_p = c, c = OMP_CLAUSE_CHAIN (c))
3103 : 2413 : switch (OMP_CLAUSE_CODE (c))
3104 : : {
3105 : 2314 : case OMP_CLAUSE_GANG:
3106 : 2314 : case OMP_CLAUSE_WORKER:
3107 : 2314 : case OMP_CLAUSE_VECTOR:
3108 : 2314 : case OMP_CLAUSE_SEQ:
3109 : 2314 : if (c_level == NULL_TREE)
3110 : : c_level = c;
3111 : 1496 : else if (OMP_CLAUSE_CODE (c) == OMP_CLAUSE_CODE (c_level))
3112 : : {
3113 : : /* This has already been diagnosed in the front ends. */
3114 : : /* Drop the duplicate clause. */
3115 : 352 : gcc_checking_assert (c_p != NULL_TREE);
3116 : 352 : OMP_CLAUSE_CHAIN (c_p) = OMP_CLAUSE_CHAIN (c);
3117 : 352 : c = c_p;
3118 : : }
3119 : : else
3120 : : {
3121 : 1144 : error_at (OMP_CLAUSE_LOCATION (c),
3122 : : "%qs specifies a conflicting level of parallelism",
3123 : 1144 : omp_clause_code_name[OMP_CLAUSE_CODE (c)]);
3124 : 1144 : inform (OMP_CLAUSE_LOCATION (c_level),
3125 : : "... to the previous %qs clause here",
3126 : 1144 : omp_clause_code_name[OMP_CLAUSE_CODE (c_level)]);
3127 : : /* Drop the conflicting clause. */
3128 : 1144 : gcc_checking_assert (c_p != NULL_TREE);
3129 : 1144 : OMP_CLAUSE_CHAIN (c_p) = OMP_CLAUSE_CHAIN (c);
3130 : 1144 : c = c_p;
3131 : : }
3132 : : break;
3133 : : case OMP_CLAUSE_NOHOST:
3134 : : /* Don't worry about duplicate clauses here. */
3135 : : c_nohost = c;
3136 : : break;
3137 : 0 : default:
3138 : 0 : gcc_unreachable ();
3139 : : }
3140 : 1199 : if (c_level == NULL_TREE)
3141 : : {
3142 : : /* Default to an implicit 'seq' clause. */
3143 : 381 : c_level = build_omp_clause (loc, OMP_CLAUSE_SEQ);
3144 : 381 : OMP_CLAUSE_CHAIN (c_level) = *clauses;
3145 : 381 : *clauses = c_level;
3146 : : }
3147 : : /* In *clauses, we now have exactly one clause specifying the level of
3148 : : parallelism. */
3149 : :
3150 : 1199 : tree attr
3151 : 1199 : = lookup_attribute ("omp declare target", DECL_ATTRIBUTES (fndecl));
3152 : 1199 : if (attr != NULL_TREE)
3153 : : {
3154 : : /* Diagnose if "#pragma omp declare target" has also been applied. */
3155 : 429 : if (TREE_VALUE (attr) == NULL_TREE)
3156 : : {
3157 : : /* See <https://gcc.gnu.org/PR93465>; the semantics of combining
3158 : : OpenACC and OpenMP 'target' are not clear. */
3159 : 32 : error_at (loc,
3160 : : "cannot apply %qs to %qD, which has also been"
3161 : : " marked with an OpenMP 'declare target' directive",
3162 : : routine_str, fndecl);
3163 : : /* Incompatible. */
3164 : 32 : return -1;
3165 : : }
3166 : :
3167 : : /* If a "#pragma acc routine" has already been applied, just verify
3168 : : this one for compatibility. */
3169 : : /* Collect previous directive's clauses. */
3170 : : tree c_level_p = NULL_TREE;
3171 : : tree c_nohost_p = NULL_TREE;
3172 : 850 : for (tree c = TREE_VALUE (attr); c; c = OMP_CLAUSE_CHAIN (c))
3173 : 453 : switch (OMP_CLAUSE_CODE (c))
3174 : : {
3175 : 397 : case OMP_CLAUSE_GANG:
3176 : 397 : case OMP_CLAUSE_WORKER:
3177 : 397 : case OMP_CLAUSE_VECTOR:
3178 : 397 : case OMP_CLAUSE_SEQ:
3179 : 397 : gcc_checking_assert (c_level_p == NULL_TREE);
3180 : : c_level_p = c;
3181 : : break;
3182 : 56 : case OMP_CLAUSE_NOHOST:
3183 : 56 : gcc_checking_assert (c_nohost_p == NULL_TREE);
3184 : : c_nohost_p = c;
3185 : : break;
3186 : 0 : default:
3187 : 0 : gcc_unreachable ();
3188 : : }
3189 : 397 : gcc_checking_assert (c_level_p != NULL_TREE);
3190 : : /* ..., and compare to current directive's, which we've already collected
3191 : : above. */
3192 : 397 : tree c_diag;
3193 : 397 : tree c_diag_p;
3194 : : /* Matching level of parallelism? */
3195 : 397 : if (OMP_CLAUSE_CODE (c_level) != OMP_CLAUSE_CODE (c_level_p))
3196 : : {
3197 : 97 : c_diag = c_level;
3198 : 97 : c_diag_p = c_level_p;
3199 : 97 : goto incompatible;
3200 : : }
3201 : : /* Matching 'nohost' clauses? */
3202 : 300 : if ((c_nohost == NULL_TREE) != (c_nohost_p == NULL_TREE))
3203 : : {
3204 : 56 : c_diag = c_nohost;
3205 : 56 : c_diag_p = c_nohost_p;
3206 : 56 : goto incompatible;
3207 : : }
3208 : : /* Compatible. */
3209 : : return 1;
3210 : :
3211 : 153 : incompatible:
3212 : 153 : if (c_diag != NULL_TREE)
3213 : 125 : error_at (OMP_CLAUSE_LOCATION (c_diag),
3214 : : "incompatible %qs clause when applying"
3215 : : " %qs to %qD, which has already been"
3216 : : " marked with an OpenACC 'routine' directive",
3217 : 125 : omp_clause_code_name[OMP_CLAUSE_CODE (c_diag)],
3218 : : routine_str, fndecl);
3219 : 28 : else if (c_diag_p != NULL_TREE)
3220 : 28 : error_at (loc,
3221 : : "missing %qs clause when applying"
3222 : : " %qs to %qD, which has already been"
3223 : : " marked with an OpenACC 'routine' directive",
3224 : 28 : omp_clause_code_name[OMP_CLAUSE_CODE (c_diag_p)],
3225 : : routine_str, fndecl);
3226 : : else
3227 : 0 : gcc_unreachable ();
3228 : 153 : if (c_diag_p != NULL_TREE)
3229 : 125 : inform (OMP_CLAUSE_LOCATION (c_diag_p),
3230 : : "... with %qs clause here",
3231 : 125 : omp_clause_code_name[OMP_CLAUSE_CODE (c_diag_p)]);
3232 : : else
3233 : : {
3234 : : /* In the front ends, we don't preserve location information for the
3235 : : OpenACC routine directive itself. However, that of c_level_p
3236 : : should be close. */
3237 : 28 : location_t loc_routine = OMP_CLAUSE_LOCATION (c_level_p);
3238 : 28 : inform (loc_routine, "... without %qs clause near to here",
3239 : 28 : omp_clause_code_name[OMP_CLAUSE_CODE (c_diag)]);
3240 : : }
3241 : : /* Incompatible. */
3242 : 153 : return -1;
3243 : : }
3244 : :
3245 : : return 0;
3246 : : }
3247 : :
3248 : : /* Process the OpenACC 'routine' directive clauses to generate an attribute
3249 : : for the level of parallelism. All dimensions have a size of zero
3250 : : (dynamic). TREE_PURPOSE is set to indicate whether that dimension
3251 : : can have a loop partitioned on it. non-zero indicates
3252 : : yes, zero indicates no. By construction once a non-zero has been
3253 : : reached, further inner dimensions must also be non-zero. We set
3254 : : TREE_VALUE to zero for the dimensions that may be partitioned and
3255 : : 1 for the other ones -- if a loop is (erroneously) spawned at
3256 : : an outer level, we don't want to try and partition it. */
3257 : :
3258 : : tree
3259 : 1100 : oacc_build_routine_dims (tree clauses)
3260 : : {
3261 : : /* Must match GOMP_DIM ordering. */
3262 : 1100 : static const omp_clause_code ids[]
3263 : : = {OMP_CLAUSE_GANG, OMP_CLAUSE_WORKER, OMP_CLAUSE_VECTOR, OMP_CLAUSE_SEQ};
3264 : 1100 : int ix;
3265 : 1100 : int level = -1;
3266 : :
3267 : 2243 : for (; clauses; clauses = OMP_CLAUSE_CHAIN (clauses))
3268 : 2247 : for (ix = GOMP_DIM_MAX + 1; ix--;)
3269 : 2204 : if (OMP_CLAUSE_CODE (clauses) == ids[ix])
3270 : : {
3271 : : level = ix;
3272 : : break;
3273 : : }
3274 : 1100 : gcc_checking_assert (level >= 0);
3275 : :
3276 : : tree dims = NULL_TREE;
3277 : :
3278 : 4400 : for (ix = GOMP_DIM_MAX; ix--;)
3279 : 3300 : dims = tree_cons (build_int_cst (boolean_type_node, ix >= level),
3280 : 3300 : build_int_cst (integer_type_node, ix < level), dims);
3281 : :
3282 : 1100 : return dims;
3283 : : }
3284 : :
3285 : : /* Retrieve the oacc function attrib and return it. Non-oacc
3286 : : functions will return NULL. */
3287 : :
3288 : : tree
3289 : 303270 : oacc_get_fn_attrib (tree fn)
3290 : : {
3291 : 303270 : return lookup_attribute (OACC_FN_ATTRIB, DECL_ATTRIBUTES (fn));
3292 : : }
3293 : :
3294 : : /* Return true if FN is an OpenMP or OpenACC offloading function. */
3295 : :
3296 : : bool
3297 : 538 : offloading_function_p (tree fn)
3298 : : {
3299 : 538 : tree attrs = DECL_ATTRIBUTES (fn);
3300 : 538 : return (lookup_attribute ("omp declare target", attrs)
3301 : 538 : || lookup_attribute ("omp target entrypoint", attrs));
3302 : : }
3303 : :
3304 : : /* Extract an oacc execution dimension from FN. FN must be an
3305 : : offloaded function or routine that has already had its execution
3306 : : dimensions lowered to the target-specific values. */
3307 : :
3308 : : int
3309 : 57220 : oacc_get_fn_dim_size (tree fn, int axis)
3310 : : {
3311 : 57220 : tree attrs = oacc_get_fn_attrib (fn);
3312 : :
3313 : 57220 : gcc_assert (axis < GOMP_DIM_MAX);
3314 : :
3315 : 57220 : tree dims = TREE_VALUE (attrs);
3316 : 116003 : while (axis--)
3317 : 58783 : dims = TREE_CHAIN (dims);
3318 : :
3319 : 57220 : int size = TREE_INT_CST_LOW (TREE_VALUE (dims));
3320 : :
3321 : 57220 : return size;
3322 : : }
3323 : :
3324 : : /* Extract the dimension axis from an IFN_GOACC_DIM_POS or
3325 : : IFN_GOACC_DIM_SIZE call. */
3326 : :
3327 : : int
3328 : 84845 : oacc_get_ifn_dim_arg (const gimple *stmt)
3329 : : {
3330 : 84845 : gcc_checking_assert (gimple_call_internal_fn (stmt) == IFN_GOACC_DIM_SIZE
3331 : : || gimple_call_internal_fn (stmt) == IFN_GOACC_DIM_POS);
3332 : 84845 : tree arg = gimple_call_arg (stmt, 0);
3333 : 84845 : HOST_WIDE_INT axis = TREE_INT_CST_LOW (arg);
3334 : :
3335 : 84845 : gcc_checking_assert (axis >= 0 && axis < GOMP_DIM_MAX);
3336 : 84845 : return (int) axis;
3337 : : }
3338 : :
3339 : : /* Build COMPONENT_REF and set TREE_THIS_VOLATILE and TREE_READONLY on it
3340 : : as appropriate. */
3341 : :
3342 : : tree
3343 : 283495 : omp_build_component_ref (tree obj, tree field)
3344 : : {
3345 : 283495 : tree ret = build3 (COMPONENT_REF, TREE_TYPE (field), obj, field, NULL);
3346 : 283495 : if (TREE_THIS_VOLATILE (field))
3347 : 62 : TREE_THIS_VOLATILE (ret) |= 1;
3348 : 283495 : if (TREE_READONLY (field))
3349 : 0 : TREE_READONLY (ret) |= 1;
3350 : 283495 : return ret;
3351 : : }
3352 : :
3353 : : /* Return true if NAME is the name of an omp_* runtime API call. */
3354 : : bool
3355 : 7223 : omp_runtime_api_procname (const char *name)
3356 : : {
3357 : 7223 : if (!startswith (name, "omp_"))
3358 : : return false;
3359 : :
3360 : : static const char *omp_runtime_apis[] =
3361 : : {
3362 : : /* This array has 3 sections. First omp_* calls that don't
3363 : : have any suffixes. */
3364 : : "aligned_alloc",
3365 : : "aligned_calloc",
3366 : : "alloc",
3367 : : "calloc",
3368 : : "free",
3369 : : "get_interop_int",
3370 : : "get_interop_ptr",
3371 : : "get_mapped_ptr",
3372 : : "get_num_interop_properties",
3373 : : "realloc",
3374 : : "target_alloc",
3375 : : "target_associate_ptr",
3376 : : "target_disassociate_ptr",
3377 : : "target_free",
3378 : : "target_is_accessible",
3379 : : "target_is_present",
3380 : : "target_memcpy",
3381 : : "target_memcpy_async",
3382 : : "target_memcpy_rect",
3383 : : "target_memcpy_rect_async",
3384 : : NULL,
3385 : : /* Now omp_* calls that are available as omp_* and omp_*_; however, the
3386 : : DECL_NAME is always omp_* without tailing underscore. */
3387 : : "capture_affinity",
3388 : : "destroy_allocator",
3389 : : "destroy_lock",
3390 : : "destroy_nest_lock",
3391 : : "display_affinity",
3392 : : "fulfill_event",
3393 : : "get_active_level",
3394 : : "get_affinity_format",
3395 : : "get_cancellation",
3396 : : "get_default_allocator",
3397 : : "get_default_device",
3398 : : "get_device_from_uid",
3399 : : "get_device_num",
3400 : : "get_dynamic",
3401 : : "get_initial_device",
3402 : : "get_interop_name",
3403 : : "get_interop_rc_desc",
3404 : : "get_interop_str",
3405 : : "get_interop_type_desc",
3406 : : "get_level",
3407 : : "get_max_active_levels",
3408 : : "get_max_task_priority",
3409 : : "get_max_teams",
3410 : : "get_max_threads",
3411 : : "get_nested",
3412 : : "get_num_devices",
3413 : : "get_num_places",
3414 : : "get_num_procs",
3415 : : "get_num_teams",
3416 : : "get_num_threads",
3417 : : "get_partition_num_places",
3418 : : "get_place_num",
3419 : : "get_proc_bind",
3420 : : "get_supported_active_levels",
3421 : : "get_team_num",
3422 : : "get_teams_thread_limit",
3423 : : "get_thread_limit",
3424 : : "get_thread_num",
3425 : : "get_wtick",
3426 : : "get_wtime",
3427 : : "in_explicit_task",
3428 : : "in_final",
3429 : : "in_parallel",
3430 : : "init_lock",
3431 : : "init_nest_lock",
3432 : : "is_initial_device",
3433 : : "pause_resource",
3434 : : "pause_resource_all",
3435 : : "set_affinity_format",
3436 : : "set_default_allocator",
3437 : : "set_lock",
3438 : : "set_nest_lock",
3439 : : "test_lock",
3440 : : "test_nest_lock",
3441 : : "unset_lock",
3442 : : "unset_nest_lock",
3443 : : NULL,
3444 : : /* And finally calls available as omp_*, omp_*_ and omp_*_8_; however,
3445 : : as DECL_NAME only omp_* and omp_*_8 appear. */
3446 : : "display_env",
3447 : : "get_ancestor_thread_num",
3448 : : "get_uid_from_device",
3449 : : "get_partition_place_nums",
3450 : : "get_place_num_procs",
3451 : : "get_place_proc_ids",
3452 : : "get_schedule",
3453 : : "get_team_size",
3454 : : "init_allocator",
3455 : : "set_default_device",
3456 : : "set_dynamic",
3457 : : "set_max_active_levels",
3458 : : "set_nested",
3459 : : "set_num_teams",
3460 : : "set_num_threads",
3461 : : "set_schedule",
3462 : : "set_teams_thread_limit"
3463 : : };
3464 : :
3465 : : int mode = 0;
3466 : 41167 : for (unsigned i = 0; i < ARRAY_SIZE (omp_runtime_apis); i++)
3467 : : {
3468 : 41166 : if (omp_runtime_apis[i] == NULL)
3469 : : {
3470 : 805 : mode++;
3471 : 805 : continue;
3472 : : }
3473 : 40361 : size_t len = strlen (omp_runtime_apis[i]);
3474 : 40361 : if (strncmp (name + 4, omp_runtime_apis[i], len) == 0
3475 : 917 : && (name[4 + len] == '\0'
3476 : 3 : || (mode > 1 && strcmp (name + 4 + len, "_8") == 0)))
3477 : : return true;
3478 : : }
3479 : : return false;
3480 : : }
3481 : :
3482 : : /* Return true if FNDECL is an omp_* runtime API call. */
3483 : :
3484 : : bool
3485 : 7380 : omp_runtime_api_call (const_tree fndecl)
3486 : : {
3487 : 7380 : tree declname = DECL_NAME (fndecl);
3488 : 7380 : if (!declname
3489 : 7380 : || (DECL_CONTEXT (fndecl) != NULL_TREE
3490 : 5368 : && TREE_CODE (DECL_CONTEXT (fndecl)) != TRANSLATION_UNIT_DECL)
3491 : 14184 : || !TREE_PUBLIC (fndecl))
3492 : : return false;
3493 : 6745 : return omp_runtime_api_procname (IDENTIFIER_POINTER (declname));
3494 : : }
3495 : :
3496 : : /* See "Additional Definitions for the OpenMP API Specification" document;
3497 : : associated IDs are 1, 2, ... */
3498 : : static const char* omp_interop_fr_str[] = {"cuda", "cuda_driver", "opencl",
3499 : : "sycl", "hip", "level_zero", "hsa"};
3500 : :
3501 : : /* Returns the foreign-runtime ID if found or 0 otherwise. */
3502 : :
3503 : : char
3504 : 228 : omp_get_fr_id_from_name (const char *str)
3505 : : {
3506 : 228 : static_assert (GOMP_INTEROP_IFR_LAST == ARRAY_SIZE (omp_interop_fr_str), "");
3507 : :
3508 : 993 : for (unsigned i = 0; i < ARRAY_SIZE (omp_interop_fr_str); ++i)
3509 : 957 : if (!strcmp (str, omp_interop_fr_str[i]))
3510 : 192 : return i + 1;
3511 : : return GOMP_INTEROP_IFR_UNKNOWN;
3512 : : }
3513 : :
3514 : : /* Returns the string value to a foreign-runtime integer value or NULL if value
3515 : : is not known. */
3516 : :
3517 : : const char *
3518 : 386 : omp_get_name_from_fr_id (int fr_id)
3519 : : {
3520 : 386 : if (fr_id < 1 || fr_id > (int) ARRAY_SIZE (omp_interop_fr_str))
3521 : : return "<unknown>";
3522 : 272 : return omp_interop_fr_str[fr_id-1];
3523 : : }
3524 : :
3525 : : namespace omp_addr_tokenizer {
3526 : :
3527 : : /* We scan an expression by recursive descent, and build a vector of
3528 : : "omp_addr_token *" pointers representing a "parsed" version of the
3529 : : expression. The grammar we use is something like this:
3530 : :
3531 : : expr0::
3532 : : expr [section-access]
3533 : :
3534 : : expr::
3535 : : structured-expr access-method
3536 : : | array-base access-method
3537 : :
3538 : : structured-expr::
3539 : : structure-base component-selector
3540 : :
3541 : : arbitrary-expr::
3542 : : (anything else)
3543 : :
3544 : : structure-base::
3545 : : DECL access-method
3546 : : | structured-expr access-method
3547 : : | arbitrary-expr access-method
3548 : :
3549 : : array-base::
3550 : : DECL
3551 : : | arbitrary-expr
3552 : :
3553 : : access-method::
3554 : : DIRECT
3555 : : | REF
3556 : : | POINTER
3557 : : | REF_TO_POINTER
3558 : : | POINTER_OFFSET
3559 : : | REF_TO_POINTER_OFFSET
3560 : : | INDEXED_ARRAY
3561 : : | INDEXED_REF_TO_ARRAY
3562 : : | index-expr
3563 : :
3564 : : index-expr::
3565 : : INDEX_EXPR access-method
3566 : :
3567 : : component-selector::
3568 : : component-selector COMPONENT_REF
3569 : : | component-selector ARRAY_REF
3570 : : | COMPONENT_REF
3571 : :
3572 : : This tokenized form is then used both in parsing, for OpenMP clause
3573 : : expansion (for C and C++) and in gimplify.cc for sibling-list handling
3574 : : (for C, C++ and Fortran). */
3575 : :
3576 : 27970 : omp_addr_token::omp_addr_token (token_type t, tree e)
3577 : 27970 : : type(t), expr(e)
3578 : : {
3579 : 27970 : }
3580 : :
3581 : 112005 : omp_addr_token::omp_addr_token (access_method_kinds k, tree e)
3582 : 112005 : : type(ACCESS_METHOD), expr(e)
3583 : : {
3584 : 112005 : u.access_kind = k;
3585 : 112005 : }
3586 : :
3587 : 86339 : omp_addr_token::omp_addr_token (token_type t, structure_base_kinds k, tree e)
3588 : 86339 : : type(t), expr(e)
3589 : : {
3590 : 86339 : u.structure_base_kind = k;
3591 : 86339 : }
3592 : :
3593 : : static bool
3594 : 86342 : omp_parse_component_selector (tree *expr0)
3595 : : {
3596 : 86342 : tree expr = *expr0;
3597 : 86342 : tree last_component = NULL_TREE;
3598 : :
3599 : 86342 : while (TREE_CODE (expr) == COMPONENT_REF
3600 : 121254 : || TREE_CODE (expr) == ARRAY_REF)
3601 : : {
3602 : 34912 : if (TREE_CODE (expr) == COMPONENT_REF)
3603 : 33147 : last_component = expr;
3604 : :
3605 : 34912 : expr = TREE_OPERAND (expr, 0);
3606 : :
3607 : 34912 : if (TREE_CODE (TREE_TYPE (expr)) == REFERENCE_TYPE)
3608 : : break;
3609 : : }
3610 : :
3611 : 86342 : if (!last_component)
3612 : : return false;
3613 : :
3614 : 27970 : *expr0 = last_component;
3615 : 27970 : return true;
3616 : : }
3617 : :
3618 : : /* This handles references that have had convert_from_reference called on
3619 : : them, and also those that haven't. */
3620 : :
3621 : : static bool
3622 : 145503 : omp_parse_ref (tree *expr0)
3623 : : {
3624 : 145503 : tree expr = *expr0;
3625 : :
3626 : 145503 : if (TREE_CODE (TREE_TYPE (expr)) == REFERENCE_TYPE)
3627 : : return true;
3628 : 141528 : else if ((TREE_CODE (expr) == INDIRECT_REF
3629 : 103823 : || (TREE_CODE (expr) == MEM_REF
3630 : 0 : && integer_zerop (TREE_OPERAND (expr, 1))))
3631 : 141528 : && TREE_CODE (TREE_TYPE (TREE_OPERAND (expr, 0))) == REFERENCE_TYPE)
3632 : : {
3633 : 9313 : *expr0 = TREE_OPERAND (expr, 0);
3634 : 9313 : return true;
3635 : : }
3636 : :
3637 : : return false;
3638 : : }
3639 : :
3640 : : static bool
3641 : 102528 : omp_parse_pointer (tree *expr0, bool *has_offset)
3642 : : {
3643 : 102528 : tree expr = *expr0;
3644 : :
3645 : 102528 : *has_offset = false;
3646 : :
3647 : 102528 : if ((TREE_CODE (expr) == INDIRECT_REF
3648 : 75233 : || (TREE_CODE (expr) == MEM_REF
3649 : 0 : && integer_zerop (TREE_OPERAND (expr, 1))))
3650 : 102528 : && TREE_CODE (TREE_TYPE (TREE_OPERAND (expr, 0))) == POINTER_TYPE)
3651 : : {
3652 : 27295 : expr = TREE_OPERAND (expr, 0);
3653 : :
3654 : : /* The Fortran FE sometimes emits a no-op cast here. */
3655 : 27295 : STRIP_NOPS (expr);
3656 : :
3657 : 30213 : while (1)
3658 : : {
3659 : 30213 : if (TREE_CODE (expr) == COMPOUND_EXPR)
3660 : : {
3661 : 101 : expr = TREE_OPERAND (expr, 1);
3662 : 101 : STRIP_NOPS (expr);
3663 : : }
3664 : 30112 : else if (TREE_CODE (expr) == SAVE_EXPR)
3665 : 89 : expr = TREE_OPERAND (expr, 0);
3666 : 30023 : else if (TREE_CODE (expr) == POINTER_PLUS_EXPR)
3667 : : {
3668 : 2728 : *has_offset = true;
3669 : 2728 : expr = TREE_OPERAND (expr, 0);
3670 : : }
3671 : : else
3672 : : break;
3673 : : }
3674 : :
3675 : 27295 : STRIP_NOPS (expr);
3676 : :
3677 : 27295 : *expr0 = expr;
3678 : 27295 : return true;
3679 : : }
3680 : :
3681 : : return false;
3682 : : }
3683 : :
3684 : : static bool
3685 : 112005 : omp_parse_access_method (tree *expr0, enum access_method_kinds *kind)
3686 : : {
3687 : 112005 : tree expr = *expr0;
3688 : 112005 : bool has_offset;
3689 : :
3690 : 112005 : if (omp_parse_ref (&expr))
3691 : 9477 : *kind = ACCESS_REF;
3692 : 102528 : else if (omp_parse_pointer (&expr, &has_offset))
3693 : : {
3694 : 27295 : if (omp_parse_ref (&expr))
3695 : 5574 : *kind = has_offset ? ACCESS_REF_TO_POINTER_OFFSET
3696 : : : ACCESS_REF_TO_POINTER;
3697 : : else
3698 : 46288 : *kind = has_offset ? ACCESS_POINTER_OFFSET : ACCESS_POINTER;
3699 : : }
3700 : 75233 : else if (TREE_CODE (expr) == ARRAY_REF)
3701 : : {
3702 : 12690 : while (TREE_CODE (expr) == ARRAY_REF)
3703 : 6487 : expr = TREE_OPERAND (expr, 0);
3704 : 6203 : if (omp_parse_ref (&expr))
3705 : 640 : *kind = ACCESS_INDEXED_REF_TO_ARRAY;
3706 : : else
3707 : 5563 : *kind = ACCESS_INDEXED_ARRAY;
3708 : : }
3709 : : else
3710 : 69030 : *kind = ACCESS_DIRECT;
3711 : :
3712 : 112005 : STRIP_NOPS (expr);
3713 : :
3714 : 112005 : *expr0 = expr;
3715 : 112005 : return true;
3716 : : }
3717 : :
3718 : : static bool
3719 : 112005 : omp_parse_access_methods (vec<omp_addr_token *> &addr_tokens, tree *expr0)
3720 : : {
3721 : 112005 : tree expr = *expr0;
3722 : 112005 : enum access_method_kinds kind;
3723 : 112005 : tree am_expr;
3724 : :
3725 : 112005 : if (omp_parse_access_method (&expr, &kind))
3726 : 112005 : am_expr = expr;
3727 : :
3728 : 112005 : if (TREE_CODE (expr) == INDIRECT_REF
3729 : 112005 : || TREE_CODE (expr) == MEM_REF
3730 : 110908 : || TREE_CODE (expr) == ARRAY_REF)
3731 : 1488 : omp_parse_access_methods (addr_tokens, &expr);
3732 : :
3733 : 112005 : addr_tokens.safe_push (new omp_addr_token (kind, am_expr));
3734 : :
3735 : 112005 : *expr0 = expr;
3736 : 112005 : return true;
3737 : : }
3738 : :
3739 : : static bool omp_parse_structured_expr (vec<omp_addr_token *> &, tree *);
3740 : :
3741 : : static bool
3742 : 86339 : omp_parse_structure_base (vec<omp_addr_token *> &addr_tokens,
3743 : : tree *expr0, structure_base_kinds *kind,
3744 : : vec<omp_addr_token *> &base_access_tokens,
3745 : : bool allow_structured = true)
3746 : : {
3747 : 86339 : tree expr = *expr0;
3748 : :
3749 : 86339 : if (allow_structured)
3750 : 27970 : omp_parse_access_methods (base_access_tokens, &expr);
3751 : :
3752 : 86339 : if (DECL_P (expr))
3753 : : {
3754 : 82024 : *kind = BASE_DECL;
3755 : 82024 : return true;
3756 : : }
3757 : :
3758 : 4315 : if (allow_structured && omp_parse_structured_expr (addr_tokens, &expr))
3759 : : {
3760 : 3792 : *kind = BASE_COMPONENT_EXPR;
3761 : 3792 : *expr0 = expr;
3762 : 3792 : return true;
3763 : : }
3764 : :
3765 : 523 : *kind = BASE_ARBITRARY_EXPR;
3766 : 523 : *expr0 = expr;
3767 : 523 : return true;
3768 : : }
3769 : :
3770 : : static bool
3771 : 86342 : omp_parse_structured_expr (vec<omp_addr_token *> &addr_tokens, tree *expr0)
3772 : : {
3773 : 86342 : tree expr = *expr0;
3774 : 86342 : tree base_component = NULL_TREE;
3775 : 86342 : structure_base_kinds struct_base_kind;
3776 : 86342 : auto_vec<omp_addr_token *> base_access_tokens;
3777 : :
3778 : 86342 : if (omp_parse_component_selector (&expr))
3779 : 27970 : base_component = expr;
3780 : : else
3781 : : return false;
3782 : :
3783 : 27970 : gcc_assert (TREE_CODE (expr) == COMPONENT_REF);
3784 : 27970 : expr = TREE_OPERAND (expr, 0);
3785 : :
3786 : 27970 : tree structure_base = expr;
3787 : :
3788 : 27970 : if (!omp_parse_structure_base (addr_tokens, &expr, &struct_base_kind,
3789 : : base_access_tokens))
3790 : : return false;
3791 : :
3792 : 27970 : addr_tokens.safe_push (new omp_addr_token (STRUCTURE_BASE, struct_base_kind,
3793 : 27970 : structure_base));
3794 : 27970 : addr_tokens.safe_splice (base_access_tokens);
3795 : 27970 : addr_tokens.safe_push (new omp_addr_token (COMPONENT_SELECTOR,
3796 : 27970 : base_component));
3797 : :
3798 : 27970 : *expr0 = expr;
3799 : :
3800 : 27970 : return true;
3801 : 86342 : }
3802 : :
3803 : : static bool
3804 : 58369 : omp_parse_array_expr (vec<omp_addr_token *> &addr_tokens, tree *expr0)
3805 : : {
3806 : 58369 : tree expr = *expr0;
3807 : 58369 : structure_base_kinds s_kind;
3808 : 58369 : auto_vec<omp_addr_token *> base_access_tokens;
3809 : :
3810 : 58369 : if (!omp_parse_structure_base (addr_tokens, &expr, &s_kind,
3811 : : base_access_tokens, false))
3812 : : return false;
3813 : :
3814 : 58369 : addr_tokens.safe_push (new omp_addr_token (ARRAY_BASE, s_kind, expr));
3815 : 58369 : addr_tokens.safe_splice (base_access_tokens);
3816 : :
3817 : 58369 : *expr0 = expr;
3818 : 58369 : return true;
3819 : 58369 : }
3820 : :
3821 : : /* Return TRUE if the ACCESS_METHOD token at index 'i' has a further
3822 : : ACCESS_METHOD chained after it (e.g., if we're processing an expression
3823 : : containing multiple pointer indirections). */
3824 : :
3825 : : bool
3826 : 44265 : omp_access_chain_p (vec<omp_addr_token *> &addr_tokens, unsigned i)
3827 : : {
3828 : 44265 : gcc_assert (addr_tokens[i]->type == ACCESS_METHOD);
3829 : 44265 : return (i + 1 < addr_tokens.length ()
3830 : 44265 : && addr_tokens[i + 1]->type == ACCESS_METHOD);
3831 : : }
3832 : :
3833 : : /* Return the address of the object accessed by the ACCESS_METHOD token
3834 : : at 'i': either of the next access method's expr, or of EXPR if we're at
3835 : : the end of the list of tokens. */
3836 : :
3837 : : tree
3838 : 4186 : omp_accessed_addr (vec<omp_addr_token *> &addr_tokens, unsigned i, tree expr)
3839 : : {
3840 : 4186 : if (i + 1 < addr_tokens.length ())
3841 : 26 : return build_fold_addr_expr (addr_tokens[i + 1]->expr);
3842 : : else
3843 : 4160 : return build_fold_addr_expr (expr);
3844 : : }
3845 : :
3846 : : } /* namespace omp_addr_tokenizer. */
3847 : :
3848 : : bool
3849 : 82547 : omp_parse_expr (vec<omp_addr_token *> &addr_tokens, tree expr)
3850 : : {
3851 : 82547 : using namespace omp_addr_tokenizer;
3852 : 82547 : auto_vec<omp_addr_token *> expr_access_tokens;
3853 : :
3854 : 82547 : if (!omp_parse_access_methods (expr_access_tokens, &expr))
3855 : : return false;
3856 : :
3857 : 82547 : if (omp_parse_structured_expr (addr_tokens, &expr))
3858 : : ;
3859 : 58369 : else if (omp_parse_array_expr (addr_tokens, &expr))
3860 : : ;
3861 : : else
3862 : : return false;
3863 : :
3864 : 82547 : addr_tokens.safe_splice (expr_access_tokens);
3865 : :
3866 : 82547 : return true;
3867 : 82547 : }
3868 : :
3869 : : DEBUG_FUNCTION void
3870 : 0 : debug_omp_tokenized_addr (vec<omp_addr_token *> &addr_tokens,
3871 : : bool with_exprs)
3872 : : {
3873 : 0 : using namespace omp_addr_tokenizer;
3874 : 0 : const char *sep = with_exprs ? " " : "";
3875 : :
3876 : 0 : for (auto e : addr_tokens)
3877 : : {
3878 : 0 : const char *pfx = "";
3879 : :
3880 : 0 : fputs (sep, stderr);
3881 : :
3882 : 0 : switch (e->type)
3883 : : {
3884 : 0 : case COMPONENT_SELECTOR:
3885 : 0 : fputs ("component_selector", stderr);
3886 : 0 : break;
3887 : 0 : case ACCESS_METHOD:
3888 : 0 : switch (e->u.access_kind)
3889 : : {
3890 : 0 : case ACCESS_DIRECT:
3891 : 0 : fputs ("access_direct", stderr);
3892 : 0 : break;
3893 : 0 : case ACCESS_REF:
3894 : 0 : fputs ("access_ref", stderr);
3895 : 0 : break;
3896 : 0 : case ACCESS_POINTER:
3897 : 0 : fputs ("access_pointer", stderr);
3898 : 0 : break;
3899 : 0 : case ACCESS_POINTER_OFFSET:
3900 : 0 : fputs ("access_pointer_offset", stderr);
3901 : 0 : break;
3902 : 0 : case ACCESS_REF_TO_POINTER:
3903 : 0 : fputs ("access_ref_to_pointer", stderr);
3904 : 0 : break;
3905 : 0 : case ACCESS_REF_TO_POINTER_OFFSET:
3906 : 0 : fputs ("access_ref_to_pointer_offset", stderr);
3907 : 0 : break;
3908 : 0 : case ACCESS_INDEXED_ARRAY:
3909 : 0 : fputs ("access_indexed_array", stderr);
3910 : 0 : break;
3911 : 0 : case ACCESS_INDEXED_REF_TO_ARRAY:
3912 : 0 : fputs ("access_indexed_ref_to_array", stderr);
3913 : 0 : break;
3914 : : }
3915 : : break;
3916 : 0 : case ARRAY_BASE:
3917 : 0 : case STRUCTURE_BASE:
3918 : 0 : pfx = e->type == ARRAY_BASE ? "array_" : "struct_";
3919 : 0 : switch (e->u.structure_base_kind)
3920 : : {
3921 : 0 : case BASE_DECL:
3922 : 0 : fprintf (stderr, "%sbase_decl", pfx);
3923 : 0 : break;
3924 : 0 : case BASE_COMPONENT_EXPR:
3925 : 0 : fputs ("base_component_expr", stderr);
3926 : 0 : break;
3927 : 0 : case BASE_ARBITRARY_EXPR:
3928 : 0 : fprintf (stderr, "%sbase_arbitrary_expr", pfx);
3929 : 0 : break;
3930 : : }
3931 : : break;
3932 : : }
3933 : 0 : if (with_exprs)
3934 : : {
3935 : 0 : fputs (" [", stderr);
3936 : 0 : print_generic_expr (stderr, e->expr);
3937 : 0 : fputc (']', stderr);
3938 : 0 : sep = ",\n ";
3939 : : }
3940 : : else
3941 : : sep = " ";
3942 : : }
3943 : :
3944 : 0 : fputs ("\n", stderr);
3945 : 0 : }
3946 : :
3947 : : /* Return number of iterations of loop I in FOR_STMT. If PSTEP is non-NULL,
3948 : : *PSTEP will be the loop step. */
3949 : :
3950 : : tree
3951 : 2647 : omp_loop_number_of_iterations (tree for_stmt, int i, tree *pstep)
3952 : : {
3953 : 2647 : tree t = TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), i);
3954 : 2647 : gcc_assert (TREE_CODE (t) == MODIFY_EXPR);
3955 : 2647 : tree decl = TREE_OPERAND (t, 0);
3956 : 2647 : tree n1 = TREE_OPERAND (t, 1);
3957 : 2647 : tree type = TREE_TYPE (decl);
3958 : 2647 : tree cond = TREE_VEC_ELT (OMP_FOR_COND (for_stmt), i);
3959 : 2647 : gcc_assert (COMPARISON_CLASS_P (cond));
3960 : 2647 : gcc_assert (TREE_OPERAND (cond, 0) == decl);
3961 : 2647 : tree_code cond_code = TREE_CODE (cond);
3962 : 2647 : tree n2 = TREE_OPERAND (cond, 1);
3963 : 2647 : t = TREE_VEC_ELT (OMP_FOR_INCR (for_stmt), i);
3964 : 2647 : tree step = NULL_TREE;
3965 : 2647 : switch (TREE_CODE (t))
3966 : : {
3967 : 1141 : case PREINCREMENT_EXPR:
3968 : 1141 : case POSTINCREMENT_EXPR:
3969 : 1141 : gcc_assert (!POINTER_TYPE_P (type));
3970 : 1141 : gcc_assert (TREE_OPERAND (t, 0) == decl);
3971 : 1141 : step = build_int_cst (type, 1);
3972 : 1141 : break;
3973 : 42 : case PREDECREMENT_EXPR:
3974 : 42 : case POSTDECREMENT_EXPR:
3975 : 42 : gcc_assert (!POINTER_TYPE_P (type));
3976 : 42 : gcc_assert (TREE_OPERAND (t, 0) == decl);
3977 : 42 : step = build_int_cst (type, -1);
3978 : 42 : break;
3979 : 1464 : case MODIFY_EXPR:
3980 : 1464 : gcc_assert (TREE_OPERAND (t, 0) == decl);
3981 : 1464 : t = TREE_OPERAND (t, 1);
3982 : 1464 : switch (TREE_CODE (t))
3983 : : {
3984 : 1389 : case PLUS_EXPR:
3985 : 1389 : if (TREE_OPERAND (t, 1) == decl)
3986 : : {
3987 : 3 : TREE_OPERAND (t, 1) = TREE_OPERAND (t, 0);
3988 : 3 : TREE_OPERAND (t, 0) = decl;
3989 : : }
3990 : : /* FALLTHRU */
3991 : : case POINTER_PLUS_EXPR:
3992 : : case MINUS_EXPR:
3993 : 1464 : step = omp_get_for_step_from_incr (EXPR_LOCATION (t), t);
3994 : 1464 : break;
3995 : 0 : default:
3996 : 0 : gcc_unreachable ();
3997 : : }
3998 : 1464 : break;
3999 : 0 : default:
4000 : 0 : gcc_unreachable ();
4001 : : }
4002 : 2647 : omp_adjust_for_condition (EXPR_LOCATION (for_stmt), &cond_code, &n2,
4003 : : decl, step);
4004 : 2647 : if (pstep)
4005 : 2099 : *pstep = step;
4006 : 2647 : if (INTEGRAL_TYPE_P (type)
4007 : 2647 : && TYPE_PRECISION (type) < TYPE_PRECISION (long_long_integer_type_node))
4008 : : {
4009 : 2538 : n1 = fold_convert (long_long_integer_type_node, n1);
4010 : 2538 : n2 = fold_convert (long_long_integer_type_node, n2);
4011 : 2538 : step = fold_convert (long_long_integer_type_node, step);
4012 : : }
4013 : 2647 : if (cond_code == LT_EXPR
4014 : 156 : || POINTER_TYPE_P (type)
4015 : 2783 : || !TYPE_UNSIGNED (TREE_TYPE (n1)))
4016 : : {
4017 : 2647 : if (POINTER_TYPE_P (type))
4018 : 60 : t = fold_build2 (POINTER_DIFF_EXPR, ssizetype, n2, n1);
4019 : : else
4020 : 2587 : t = fold_build2 (MINUS_EXPR, TREE_TYPE (n1), n2, n1);
4021 : 2647 : t = fold_build2 (CEIL_DIV_EXPR, TREE_TYPE (t), t, step);
4022 : : }
4023 : : else
4024 : : {
4025 : 0 : t = fold_build2 (MINUS_EXPR, type, n1, n2);
4026 : 0 : t = fold_build2 (CEIL_DIV_EXPR, type, t,
4027 : : fold_build1 (NEGATE_EXPR, type, step));
4028 : : }
4029 : 2647 : return t;
4030 : : }
4031 : :
4032 : : /* Tile transformation:
4033 : : Original loop:
4034 : :
4035 : : #pragma omp tile sizes(16, 32)
4036 : : for (i = 0; i < k; ++i)
4037 : : for (j = 0; j < 128; j += 2)
4038 : : {
4039 : : baz (i, j);
4040 : : }
4041 : :
4042 : : Transformed loop:
4043 : : #pragma omp tile sizes(16, 32)
4044 : : for (i.0 = 0; i.0 < k; i.0 += 16)
4045 : : for (j.0 = 0; j.0 < 128; j.0 += 64)
4046 : : {
4047 : : i = i.0;
4048 : : i.1 = MIN_EXPR <i.0 + 16, k>;
4049 : : goto <D.2783>;
4050 : : <D.2782>:;
4051 : : j = j.0;
4052 : : j.1 = j.0 + 32;
4053 : : goto <D.2786>;
4054 : : <D.2785>:;
4055 : : {
4056 : : baz (i, j);
4057 : : }
4058 : : j += 2;
4059 : : <D.2786>:;
4060 : : if (j < j.1) goto <D.2785>; else goto <D.2787>;
4061 : : <D.2787>:;
4062 : : ++i;
4063 : : <D.2783>:;
4064 : : if (i < i.1) goto <D.2782>; else goto <D.2784>;
4065 : : <D.2784>:;
4066 : : }
4067 : :
4068 : : where the grid loops have canonical form, but the inner
4069 : : loops don't and so are immediately lowered. */
4070 : :
4071 : : static void
4072 : 1958 : omp_apply_tile (tree for_stmt, tree sizes, int size)
4073 : : {
4074 : 1958 : tree pre_body = NULL_TREE, post_body = NULL_TREE;
4075 : 1958 : tree orig_sizes = sizes;
4076 : 1958 : if (OMP_FOR_NON_RECTANGULAR (for_stmt))
4077 : : {
4078 : 51 : error_at (EXPR_LOCATION (for_stmt), "non-rectangular %<tile%>");
4079 : 51 : return;
4080 : : }
4081 : 4434 : for (int i = 0; i < TREE_VEC_LENGTH (OMP_FOR_INIT (for_stmt)); i++)
4082 : : {
4083 : 2527 : if (orig_sizes)
4084 : : {
4085 : 1734 : size = tree_to_uhwi (TREE_VALUE (sizes));
4086 : 1734 : sizes = TREE_CHAIN (sizes);
4087 : : }
4088 : 2527 : if (size == 1)
4089 : 428 : continue;
4090 : 2099 : if (OMP_FOR_ORIG_DECLS (for_stmt) == NULL_TREE)
4091 : : {
4092 : 522 : OMP_FOR_ORIG_DECLS (for_stmt)
4093 : 522 : = make_tree_vec (TREE_VEC_LENGTH (OMP_FOR_INIT (for_stmt)));
4094 : 1188 : for (int j = 0; j < TREE_VEC_LENGTH (OMP_FOR_INIT (for_stmt)); j++)
4095 : : {
4096 : 666 : gcc_assert (TREE_CODE (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), j))
4097 : : == MODIFY_EXPR);
4098 : 666 : TREE_VEC_ELT (OMP_FOR_ORIG_DECLS (for_stmt), j)
4099 : 1332 : = TREE_OPERAND (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), j), 0);
4100 : : }
4101 : : }
4102 : 2099 : tree step;
4103 : 2099 : tree iters = omp_loop_number_of_iterations (for_stmt, i, &step);
4104 : 2099 : tree t = TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), i);
4105 : 2099 : tree decl = TREE_OPERAND (t, 0);
4106 : 2099 : tree type = TREE_TYPE (decl);
4107 : 2099 : tree griddecl = create_tmp_var_raw (type);
4108 : 2099 : DECL_CONTEXT (griddecl) = current_function_decl;
4109 : 2099 : t = build1 (DECL_EXPR, void_type_node, griddecl);
4110 : 2099 : append_to_statement_list (t, &OMP_FOR_PRE_BODY (for_stmt));
4111 : 2099 : TREE_OPERAND (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), i), 0) = griddecl;
4112 : 2099 : TREE_PRIVATE (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), i)) = 1;
4113 : 2099 : tree cond = TREE_VEC_ELT (OMP_FOR_COND (for_stmt), i);
4114 : 2099 : TREE_OPERAND (cond, 0) = griddecl;
4115 : 2099 : tree ub = save_expr (TREE_OPERAND (cond, 1));
4116 : 2099 : TREE_OPERAND (cond, 1) = ub;
4117 : 2099 : t = TREE_VEC_ELT (OMP_FOR_INCR (for_stmt), i);
4118 : 2099 : if (TREE_CODE (cond) == NE_EXPR)
4119 : : {
4120 : 163 : tree_code cond_code = TREE_CODE (cond);
4121 : 163 : omp_adjust_for_condition (EXPR_LOCATION (for_stmt), &cond_code,
4122 : : &ub, griddecl, step);
4123 : 163 : TREE_SET_CODE (cond, cond_code);
4124 : : }
4125 : 2099 : step = save_expr (step);
4126 : 2099 : tree gridstep = fold_build2 (MULT_EXPR, TREE_TYPE (step),
4127 : : step, build_int_cst (TREE_TYPE (step),
4128 : : size));
4129 : 2099 : if (POINTER_TYPE_P (type))
4130 : 52 : t = build2 (POINTER_PLUS_EXPR, type, griddecl,
4131 : : fold_convert (sizetype, gridstep));
4132 : : else
4133 : 2047 : t = build2 (PLUS_EXPR, type, griddecl, gridstep);
4134 : 2099 : t = build2 (MODIFY_EXPR, type, griddecl, t);
4135 : 2099 : TREE_VEC_ELT (OMP_FOR_INCR (for_stmt), i) = t;
4136 : 2099 : t = build2 (MODIFY_EXPR, type, decl, griddecl);
4137 : 2099 : append_to_statement_list (t, &pre_body);
4138 : 2099 : if (POINTER_TYPE_P (type))
4139 : 52 : t = build2 (POINTER_PLUS_EXPR, type, griddecl,
4140 : : fold_convert (sizetype, gridstep));
4141 : : else
4142 : 2047 : t = build2 (PLUS_EXPR, type, griddecl, gridstep);
4143 : 2099 : bool minmax_needed = true;
4144 : 2099 : if (TREE_CODE (iters) == INTEGER_CST)
4145 : : {
4146 : 1135 : wide_int witers = wi::to_wide (iters);
4147 : 1135 : wide_int wsize = wide_int::from (size, witers.get_precision (),
4148 : 1135 : TYPE_SIGN (TREE_TYPE (iters)));
4149 : 1135 : if (wi::multiple_of_p (witers, wsize, TYPE_SIGN (TREE_TYPE (iters))))
4150 : 417 : minmax_needed = false;
4151 : 1135 : }
4152 : 1135 : if (minmax_needed)
4153 : 1682 : switch (TREE_CODE (cond))
4154 : : {
4155 : 251 : case LE_EXPR:
4156 : 251 : if (POINTER_TYPE_P (type))
4157 : 4 : t = build2 (MIN_EXPR, type, t,
4158 : 4 : build2 (POINTER_PLUS_EXPR, type, ub, size_int (1)));
4159 : : else
4160 : 247 : t = build2 (MIN_EXPR, type, t,
4161 : : build2 (PLUS_EXPR, type, ub, build_one_cst (type)));
4162 : : break;
4163 : 1375 : case LT_EXPR:
4164 : 1375 : t = build2 (MIN_EXPR, type, t, ub);
4165 : 1375 : break;
4166 : 28 : case GE_EXPR:
4167 : 28 : if (POINTER_TYPE_P (type))
4168 : 8 : t = build2 (MAX_EXPR, type, t,
4169 : 8 : build2 (POINTER_PLUS_EXPR, type, ub, size_int (-1)));
4170 : : else
4171 : 20 : t = build2 (MAX_EXPR, type, t,
4172 : : build2 (PLUS_EXPR, type, ub,
4173 : : build_minus_one_cst (type)));
4174 : : break;
4175 : 28 : case GT_EXPR:
4176 : 28 : t = build2 (MAX_EXPR, type, t, ub);
4177 : 28 : break;
4178 : 0 : default:
4179 : 0 : gcc_unreachable ();
4180 : : }
4181 : 2099 : tree end = create_tmp_var_raw (type);
4182 : 2099 : DECL_CONTEXT (end) = current_function_decl;
4183 : 2099 : end = build4 (TARGET_EXPR, type, end, t, NULL_TREE, NULL_TREE);
4184 : 2099 : TREE_SIDE_EFFECTS (end) = 1;
4185 : 2099 : append_to_statement_list (end, &pre_body);
4186 : 2099 : tree lab1 = create_artificial_label (UNKNOWN_LOCATION);
4187 : 2099 : tree lab2 = create_artificial_label (UNKNOWN_LOCATION);
4188 : 2099 : t = build1 (GOTO_EXPR, void_type_node, lab2);
4189 : 2099 : append_to_statement_list (t, &pre_body);
4190 : 2099 : t = build1 (LABEL_EXPR, void_type_node, lab1);
4191 : 2099 : append_to_statement_list (t, &pre_body);
4192 : 2099 : tree this_post_body = NULL_TREE;
4193 : 2099 : if (POINTER_TYPE_P (type))
4194 : 52 : t = build2 (POINTER_PLUS_EXPR, type, decl,
4195 : : fold_convert (sizetype, step));
4196 : : else
4197 : 2047 : t = build2 (PLUS_EXPR, type, decl, step);
4198 : 2099 : t = build2 (MODIFY_EXPR, type, decl, t);
4199 : 2099 : append_to_statement_list (t, &this_post_body);
4200 : 2099 : t = build1 (LABEL_EXPR, void_type_node, lab2);
4201 : 2099 : append_to_statement_list (t, &this_post_body);
4202 : 2165 : t = build2 ((TREE_CODE (cond) == LT_EXPR || TREE_CODE (cond) == LE_EXPR)
4203 : : ? LT_EXPR : GT_EXPR, boolean_type_node, decl, end);
4204 : 2099 : if (orig_sizes == NULL_TREE)
4205 : : {
4206 : 745 : gcc_assert (i == 0);
4207 : 1490 : t = build3 (ANNOTATE_EXPR, TREE_TYPE (t), t,
4208 : 745 : build_int_cst (integer_type_node,
4209 : 745 : annot_expr_unroll_kind),
4210 : 745 : build_int_cst (integer_type_node, size));
4211 : : }
4212 : 2099 : t = build3 (COND_EXPR, void_type_node, t,
4213 : : build1 (GOTO_EXPR, void_type_node, lab1), NULL_TREE);
4214 : 2099 : append_to_statement_list (t, &this_post_body);
4215 : 2099 : append_to_statement_list (post_body, &this_post_body);
4216 : 2099 : post_body = this_post_body;
4217 : : }
4218 : 1907 : if (pre_body || post_body)
4219 : : {
4220 : 1614 : append_to_statement_list (OMP_FOR_BODY (for_stmt), &pre_body);
4221 : 1614 : append_to_statement_list (post_body, &pre_body);
4222 : 1614 : OMP_FOR_BODY (for_stmt) = pre_body;
4223 : : }
4224 : : }
4225 : :
4226 : : /* Callback for walk_tree to find nested loop transforming construct. */
4227 : :
4228 : : static tree
4229 : 9261 : find_nested_loop_xform (tree *tp, int *walk_subtrees, void *data)
4230 : : {
4231 : 9261 : tree **pdata = (tree **) data;
4232 : 9261 : *walk_subtrees = 0;
4233 : 9261 : switch (TREE_CODE (*tp))
4234 : : {
4235 : 1987 : case OMP_TILE:
4236 : 1987 : case OMP_UNROLL:
4237 : 1987 : pdata[1] = tp;
4238 : 1987 : return *tp;
4239 : 2581 : case BIND_EXPR:
4240 : 2581 : if (BIND_EXPR_VARS (*tp)
4241 : 2581 : || (BIND_EXPR_BLOCK (*tp)
4242 : 1270 : && BLOCK_VARS (BIND_EXPR_BLOCK (*tp))))
4243 : 1311 : pdata[0] = tp;
4244 : 2581 : *walk_subtrees = 1;
4245 : 2581 : break;
4246 : 425 : case STATEMENT_LIST:
4247 : 425 : if (!tsi_one_before_end_p (tsi_start (*tp)))
4248 : 425 : pdata[0] = tp;
4249 : 425 : *walk_subtrees = 1;
4250 : 425 : break;
4251 : 36 : case TRY_FINALLY_EXPR:
4252 : 36 : case CLEANUP_POINT_EXPR:
4253 : 36 : pdata[0] = tp;
4254 : 36 : *walk_subtrees = 1;
4255 : 36 : break;
4256 : : default:
4257 : : break;
4258 : : }
4259 : : return NULL;
4260 : : }
4261 : :
4262 : : /* Main entry point for performing OpenMP loop transformations. */
4263 : :
4264 : : void
4265 : 58521 : omp_maybe_apply_loop_xforms (tree *expr_p, tree for_clauses)
4266 : : {
4267 : 58521 : tree for_stmt = *expr_p;
4268 : :
4269 : 58521 : switch (TREE_CODE (for_stmt))
4270 : : {
4271 : 3923 : case OMP_TILE:
4272 : 3923 : case OMP_UNROLL:
4273 : 3923 : if (OMP_LOOPXFORM_LOWERED (for_stmt))
4274 : : return;
4275 : : break;
4276 : : default:
4277 : : break;
4278 : : }
4279 : :
4280 : : tree *inner_expr_p = expr_p;
4281 : : tree inner_for_stmt = for_stmt;
4282 : 138792 : for (int i = 0; i < TREE_VEC_LENGTH (OMP_FOR_INIT (for_stmt)); i++)
4283 : : {
4284 : : /* If some loop nest needs one or more loops in canonical form
4285 : : from nested loop transforming constructs, first perform the
4286 : : loop transformation on the nested construct and then move over
4287 : : the corresponding loops in canonical form from the inner construct
4288 : : to the outer one. */
4289 : 81993 : if (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), i) == NULL_TREE)
4290 : : {
4291 : 1516 : if (inner_for_stmt == for_stmt
4292 : 3008 : && omp_find_clause (for_clauses ? for_clauses
4293 : 1492 : : OMP_FOR_CLAUSES (for_stmt),
4294 : : OMP_CLAUSE_ORDERED))
4295 : : {
4296 : 73 : error_at (EXPR_LOCATION (for_stmt),
4297 : : "%<ordered%> clause used with generated loops");
4298 : 73 : *expr_p = void_node;
4299 : 73 : return;
4300 : : }
4301 : 1443 : tree *data[2] = { NULL, NULL };
4302 : 1443 : walk_tree (&OMP_FOR_BODY (inner_for_stmt),
4303 : : find_nested_loop_xform, &data, NULL);
4304 : 1443 : gcc_assert (data[1]);
4305 : 1443 : if (data[0])
4306 : : {
4307 : : /* If there is a BIND_EXPR declaring some vars, or statement
4308 : : list with more than one stmt etc., move the intervening
4309 : : code around the outermost loop. */
4310 : 1008 : tree t = *inner_expr_p;
4311 : 1008 : *inner_expr_p = OMP_FOR_BODY (inner_for_stmt);
4312 : 1008 : OMP_FOR_BODY (inner_for_stmt) = *data[1];
4313 : 1008 : *data[1] = t;
4314 : 1008 : inner_expr_p = data[1];
4315 : 1008 : data[1] = &OMP_FOR_BODY (inner_for_stmt);
4316 : : }
4317 : 1443 : inner_for_stmt = *data[1];
4318 : :
4319 : 1443 : omp_maybe_apply_loop_xforms (data[1], NULL_TREE);
4320 : 1443 : if (*data[1] != inner_for_stmt)
4321 : : {
4322 : 544 : tree *data2[2] = { NULL, NULL };
4323 : 544 : walk_tree (data[1], find_nested_loop_xform, &data2, NULL);
4324 : 544 : gcc_assert (data2[1]
4325 : : && *data2[1] == inner_for_stmt
4326 : : && data2[0]);
4327 : 544 : tree t = *inner_expr_p;
4328 : 544 : *inner_expr_p = *data[1];
4329 : 544 : *data[1] = *data2[1];
4330 : 544 : *data2[1] = t;
4331 : 544 : inner_expr_p = data2[1];
4332 : : }
4333 : 1443 : tree clauses = OMP_FOR_CLAUSES (inner_for_stmt);
4334 : 1443 : gcc_checking_assert (TREE_CODE (inner_for_stmt) != OMP_UNROLL
4335 : : || omp_find_clause (clauses,
4336 : : OMP_CLAUSE_PARTIAL));
4337 : 1443 : append_to_statement_list (OMP_FOR_PRE_BODY (inner_for_stmt),
4338 : : &OMP_FOR_PRE_BODY (for_stmt));
4339 : 1443 : OMP_FOR_PRE_BODY (inner_for_stmt) = NULL_TREE;
4340 : 1443 : if (OMP_FOR_ORIG_DECLS (for_stmt) == NULL_TREE
4341 : 1443 : && OMP_FOR_ORIG_DECLS (inner_for_stmt) != NULL_TREE)
4342 : : {
4343 : 624 : OMP_FOR_ORIG_DECLS (for_stmt)
4344 : 624 : = make_tree_vec (TREE_VEC_LENGTH (OMP_FOR_INIT (for_stmt)));
4345 : 1466 : for (int j = 0; j < TREE_VEC_LENGTH (OMP_FOR_INIT (for_stmt));
4346 : : j++)
4347 : : {
4348 : 842 : if (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), j) == NULL_TREE)
4349 : 717 : continue;
4350 : 125 : gcc_assert (TREE_CODE (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt),
4351 : : j)) == MODIFY_EXPR);
4352 : 125 : TREE_VEC_ELT (OMP_FOR_ORIG_DECLS (for_stmt), j)
4353 : 250 : = TREE_OPERAND (TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), j),
4354 : : 0);
4355 : : }
4356 : : }
4357 : 3104 : for (int j = 0; j < TREE_VEC_LENGTH (OMP_FOR_INIT (inner_for_stmt));
4358 : : ++j)
4359 : : {
4360 : 1924 : if (i + j == TREE_VEC_LENGTH (OMP_FOR_INIT (for_stmt)))
4361 : : break;
4362 : 1661 : if (OMP_FOR_ORIG_DECLS (for_stmt))
4363 : : {
4364 : 1576 : if (OMP_FOR_ORIG_DECLS (inner_for_stmt))
4365 : : {
4366 : 1576 : TREE_VEC_ELT (OMP_FOR_ORIG_DECLS (for_stmt), i + j)
4367 : 1576 : = TREE_VEC_ELT (OMP_FOR_ORIG_DECLS (inner_for_stmt),
4368 : : j);
4369 : 1576 : TREE_VEC_ELT (OMP_FOR_ORIG_DECLS (inner_for_stmt), j)
4370 : 1576 : = NULL_TREE;
4371 : : }
4372 : : else
4373 : : {
4374 : 0 : tree t = TREE_VEC_ELT (OMP_FOR_INIT (inner_for_stmt), j);
4375 : 0 : gcc_assert (TREE_CODE (t) == MODIFY_EXPR);
4376 : 0 : TREE_VEC_ELT (OMP_FOR_ORIG_DECLS (for_stmt), i + j)
4377 : 0 : = TREE_OPERAND (t, 0);
4378 : : }
4379 : : }
4380 : 1661 : TREE_VEC_ELT (OMP_FOR_INIT (for_stmt), i + j)
4381 : 1661 : = TREE_VEC_ELT (OMP_FOR_INIT (inner_for_stmt), j);
4382 : 1661 : TREE_VEC_ELT (OMP_FOR_COND (for_stmt), i + j)
4383 : 1661 : = TREE_VEC_ELT (OMP_FOR_COND (inner_for_stmt), j);
4384 : 1661 : TREE_VEC_ELT (OMP_FOR_INCR (for_stmt), i + j)
4385 : 1661 : = TREE_VEC_ELT (OMP_FOR_INCR (inner_for_stmt), j);
4386 : 1661 : TREE_VEC_ELT (OMP_FOR_INIT (inner_for_stmt), j) = NULL_TREE;
4387 : 1661 : TREE_VEC_ELT (OMP_FOR_COND (inner_for_stmt), j) = NULL_TREE;
4388 : 1661 : TREE_VEC_ELT (OMP_FOR_INCR (inner_for_stmt), j) = NULL_TREE;
4389 : : }
4390 : : }
4391 : : }
4392 : :
4393 : 56799 : switch (TREE_CODE (for_stmt))
4394 : : {
4395 : 1165 : case OMP_TILE:
4396 : 1165 : tree sizes;
4397 : 1165 : sizes = omp_find_clause (OMP_FOR_CLAUSES (for_stmt), OMP_CLAUSE_SIZES);
4398 : 1165 : omp_apply_tile (for_stmt, OMP_CLAUSE_SIZES_LIST (sizes), 0);
4399 : 1165 : OMP_LOOPXFORM_LOWERED (for_stmt) = 1;
4400 : 1165 : break;
4401 : 1109 : case OMP_UNROLL:
4402 : 1109 : tree partial;
4403 : 1109 : partial = omp_find_clause (OMP_FOR_CLAUSES (for_stmt),
4404 : : OMP_CLAUSE_PARTIAL);
4405 : 1109 : if (partial)
4406 : 793 : omp_apply_tile (for_stmt, NULL_TREE,
4407 : 793 : OMP_CLAUSE_PARTIAL_EXPR (partial)
4408 : 609 : ? tree_to_shwi (OMP_CLAUSE_PARTIAL_EXPR (partial))
4409 : : : 8);
4410 : 316 : else if (omp_find_clause (OMP_FOR_CLAUSES (for_stmt), OMP_CLAUSE_FULL))
4411 : : {
4412 : 274 : tree iters = omp_loop_number_of_iterations (for_stmt, 0, NULL);
4413 : 274 : if (TREE_CODE (iters) != INTEGER_CST)
4414 : 12 : error_at (EXPR_LOCATION (for_stmt),
4415 : : "non-constant iteration count of %<unroll full%> loop");
4416 : : }
4417 : 1109 : OMP_LOOPXFORM_LOWERED (for_stmt) = 1;
4418 : 1109 : break;
4419 : : default:
4420 : : break;
4421 : : }
4422 : : }
4423 : :
4424 : : #include "gt-omp-general.h"
|
