]> git.ipfire.org Git - thirdparty/gcc.git/blob - gcc/omp-simd-clone.c
Fix memory leak in omp-simd-clone.c
[thirdparty/gcc.git] / gcc / omp-simd-clone.c
1 /* OMP constructs' SIMD clone supporting code.
2
3 Copyright (C) 2005-2016 Free Software Foundation, Inc.
4
5 This file is part of GCC.
6
7 GCC is free software; you can redistribute it and/or modify it under
8 the terms of the GNU General Public License as published by the Free
9 Software Foundation; either version 3, or (at your option) any later
10 version.
11
12 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
15 for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with GCC; see the file COPYING3. If not see
19 <http://www.gnu.org/licenses/>. */
20
21 #include "config.h"
22 #include "system.h"
23 #include "coretypes.h"
24 #include "backend.h"
25 #include "target.h"
26 #include "tree.h"
27 #include "gimple.h"
28 #include "cfghooks.h"
29 #include "alloc-pool.h"
30 #include "tree-pass.h"
31 #include "ssa.h"
32 #include "cgraph.h"
33 #include "pretty-print.h"
34 #include "diagnostic-core.h"
35 #include "fold-const.h"
36 #include "stor-layout.h"
37 #include "cfganal.h"
38 #include "gimplify.h"
39 #include "gimple-iterator.h"
40 #include "gimplify-me.h"
41 #include "gimple-walk.h"
42 #include "langhooks.h"
43 #include "tree-cfg.h"
44 #include "tree-into-ssa.h"
45 #include "tree-dfa.h"
46 #include "cfgloop.h"
47 #include "symbol-summary.h"
48 #include "ipa-prop.h"
49 #include "tree-eh.h"
50
51
52 /* Allocate a fresh `simd_clone' and return it. NARGS is the number
53 of arguments to reserve space for. */
54
55 static struct cgraph_simd_clone *
56 simd_clone_struct_alloc (int nargs)
57 {
58 struct cgraph_simd_clone *clone_info;
59 size_t len = (sizeof (struct cgraph_simd_clone)
60 + nargs * sizeof (struct cgraph_simd_clone_arg));
61 clone_info = (struct cgraph_simd_clone *)
62 ggc_internal_cleared_alloc (len);
63 return clone_info;
64 }
65
66 /* Make a copy of the `struct cgraph_simd_clone' in FROM to TO. */
67
68 static inline void
69 simd_clone_struct_copy (struct cgraph_simd_clone *to,
70 struct cgraph_simd_clone *from)
71 {
72 memcpy (to, from, (sizeof (struct cgraph_simd_clone)
73 + ((from->nargs - from->inbranch)
74 * sizeof (struct cgraph_simd_clone_arg))));
75 }
76
77 /* Return vector of parameter types of function FNDECL. This uses
78 TYPE_ARG_TYPES if available, otherwise falls back to types of
79 DECL_ARGUMENTS types. */
80
81 static vec<tree>
82 simd_clone_vector_of_formal_parm_types (tree fndecl)
83 {
84 if (TYPE_ARG_TYPES (TREE_TYPE (fndecl)))
85 return ipa_get_vector_of_formal_parm_types (TREE_TYPE (fndecl));
86 vec<tree> args = ipa_get_vector_of_formal_parms (fndecl);
87 unsigned int i;
88 tree arg;
89 FOR_EACH_VEC_ELT (args, i, arg)
90 args[i] = TREE_TYPE (args[i]);
91 return args;
92 }
93
94 /* Given a simd function in NODE, extract the simd specific
95 information from the OMP clauses passed in CLAUSES, and return
96 the struct cgraph_simd_clone * if it should be cloned. *INBRANCH_SPECIFIED
97 is set to TRUE if the `inbranch' or `notinbranch' clause specified,
98 otherwise set to FALSE. */
99
100 static struct cgraph_simd_clone *
101 simd_clone_clauses_extract (struct cgraph_node *node, tree clauses,
102 bool *inbranch_specified)
103 {
104 vec<tree> args = simd_clone_vector_of_formal_parm_types (node->decl);
105 tree t;
106 int n;
107 *inbranch_specified = false;
108
109 n = args.length ();
110 if (n > 0 && args.last () == void_type_node)
111 n--;
112
113 /* To distinguish from an OpenMP simd clone, Cilk Plus functions to
114 be cloned have a distinctive artificial label in addition to "omp
115 declare simd". */
116 bool cilk_clone
117 = (flag_cilkplus
118 && lookup_attribute ("cilk simd function",
119 DECL_ATTRIBUTES (node->decl)));
120
121 /* Allocate one more than needed just in case this is an in-branch
122 clone which will require a mask argument. */
123 struct cgraph_simd_clone *clone_info = simd_clone_struct_alloc (n + 1);
124 clone_info->nargs = n;
125 clone_info->cilk_elemental = cilk_clone;
126
127 if (!clauses)
128 {
129 args.release ();
130 return clone_info;
131 }
132 clauses = TREE_VALUE (clauses);
133 if (!clauses || TREE_CODE (clauses) != OMP_CLAUSE)
134 return clone_info;
135
136 for (t = clauses; t; t = OMP_CLAUSE_CHAIN (t))
137 {
138 switch (OMP_CLAUSE_CODE (t))
139 {
140 case OMP_CLAUSE_INBRANCH:
141 clone_info->inbranch = 1;
142 *inbranch_specified = true;
143 break;
144 case OMP_CLAUSE_NOTINBRANCH:
145 clone_info->inbranch = 0;
146 *inbranch_specified = true;
147 break;
148 case OMP_CLAUSE_SIMDLEN:
149 clone_info->simdlen
150 = TREE_INT_CST_LOW (OMP_CLAUSE_SIMDLEN_EXPR (t));
151 break;
152 case OMP_CLAUSE_LINEAR:
153 {
154 tree decl = OMP_CLAUSE_DECL (t);
155 tree step = OMP_CLAUSE_LINEAR_STEP (t);
156 int argno = TREE_INT_CST_LOW (decl);
157 if (OMP_CLAUSE_LINEAR_VARIABLE_STRIDE (t))
158 {
159 enum cgraph_simd_clone_arg_type arg_type;
160 if (TREE_CODE (args[argno]) == REFERENCE_TYPE)
161 switch (OMP_CLAUSE_LINEAR_KIND (t))
162 {
163 case OMP_CLAUSE_LINEAR_REF:
164 arg_type
165 = SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP;
166 break;
167 case OMP_CLAUSE_LINEAR_UVAL:
168 arg_type
169 = SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP;
170 break;
171 case OMP_CLAUSE_LINEAR_VAL:
172 case OMP_CLAUSE_LINEAR_DEFAULT:
173 arg_type
174 = SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP;
175 break;
176 default:
177 gcc_unreachable ();
178 }
179 else
180 arg_type = SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP;
181 clone_info->args[argno].arg_type = arg_type;
182 clone_info->args[argno].linear_step = tree_to_shwi (step);
183 gcc_assert (clone_info->args[argno].linear_step >= 0
184 && clone_info->args[argno].linear_step < n);
185 }
186 else
187 {
188 if (POINTER_TYPE_P (args[argno]))
189 step = fold_convert (ssizetype, step);
190 if (!tree_fits_shwi_p (step))
191 {
192 warning_at (OMP_CLAUSE_LOCATION (t), 0,
193 "ignoring large linear step");
194 args.release ();
195 return NULL;
196 }
197 else if (integer_zerop (step))
198 {
199 warning_at (OMP_CLAUSE_LOCATION (t), 0,
200 "ignoring zero linear step");
201 args.release ();
202 return NULL;
203 }
204 else
205 {
206 enum cgraph_simd_clone_arg_type arg_type;
207 if (TREE_CODE (args[argno]) == REFERENCE_TYPE)
208 switch (OMP_CLAUSE_LINEAR_KIND (t))
209 {
210 case OMP_CLAUSE_LINEAR_REF:
211 arg_type
212 = SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP;
213 break;
214 case OMP_CLAUSE_LINEAR_UVAL:
215 arg_type
216 = SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP;
217 break;
218 case OMP_CLAUSE_LINEAR_VAL:
219 case OMP_CLAUSE_LINEAR_DEFAULT:
220 arg_type
221 = SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP;
222 break;
223 default:
224 gcc_unreachable ();
225 }
226 else
227 arg_type = SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP;
228 clone_info->args[argno].arg_type = arg_type;
229 clone_info->args[argno].linear_step = tree_to_shwi (step);
230 }
231 }
232 break;
233 }
234 case OMP_CLAUSE_UNIFORM:
235 {
236 tree decl = OMP_CLAUSE_DECL (t);
237 int argno = tree_to_uhwi (decl);
238 clone_info->args[argno].arg_type
239 = SIMD_CLONE_ARG_TYPE_UNIFORM;
240 break;
241 }
242 case OMP_CLAUSE_ALIGNED:
243 {
244 tree decl = OMP_CLAUSE_DECL (t);
245 int argno = tree_to_uhwi (decl);
246 clone_info->args[argno].alignment
247 = TREE_INT_CST_LOW (OMP_CLAUSE_ALIGNED_ALIGNMENT (t));
248 break;
249 }
250 default:
251 break;
252 }
253 }
254 args.release ();
255 return clone_info;
256 }
257
258 /* Given a SIMD clone in NODE, calculate the characteristic data
259 type and return the coresponding type. The characteristic data
260 type is computed as described in the Intel Vector ABI. */
261
262 static tree
263 simd_clone_compute_base_data_type (struct cgraph_node *node,
264 struct cgraph_simd_clone *clone_info)
265 {
266 tree type = integer_type_node;
267 tree fndecl = node->decl;
268
269 /* a) For non-void function, the characteristic data type is the
270 return type. */
271 if (TREE_CODE (TREE_TYPE (TREE_TYPE (fndecl))) != VOID_TYPE)
272 type = TREE_TYPE (TREE_TYPE (fndecl));
273
274 /* b) If the function has any non-uniform, non-linear parameters,
275 then the characteristic data type is the type of the first
276 such parameter. */
277 else
278 {
279 vec<tree> map = simd_clone_vector_of_formal_parm_types (fndecl);
280 for (unsigned int i = 0; i < clone_info->nargs; ++i)
281 if (clone_info->args[i].arg_type == SIMD_CLONE_ARG_TYPE_VECTOR)
282 {
283 type = map[i];
284 break;
285 }
286 map.release ();
287 }
288
289 /* c) If the characteristic data type determined by a) or b) above
290 is struct, union, or class type which is pass-by-value (except
291 for the type that maps to the built-in complex data type), the
292 characteristic data type is int. */
293 if (RECORD_OR_UNION_TYPE_P (type)
294 && !aggregate_value_p (type, NULL)
295 && TREE_CODE (type) != COMPLEX_TYPE)
296 return integer_type_node;
297
298 /* d) If none of the above three classes is applicable, the
299 characteristic data type is int. */
300
301 return type;
302
303 /* e) For Intel Xeon Phi native and offload compilation, if the
304 resulting characteristic data type is 8-bit or 16-bit integer
305 data type, the characteristic data type is int. */
306 /* Well, we don't handle Xeon Phi yet. */
307 }
308
309 static tree
310 simd_clone_mangle (struct cgraph_node *node,
311 struct cgraph_simd_clone *clone_info)
312 {
313 char vecsize_mangle = clone_info->vecsize_mangle;
314 char mask = clone_info->inbranch ? 'M' : 'N';
315 unsigned int simdlen = clone_info->simdlen;
316 unsigned int n;
317 pretty_printer pp;
318
319 gcc_assert (vecsize_mangle && simdlen);
320
321 pp_string (&pp, "_ZGV");
322 pp_character (&pp, vecsize_mangle);
323 pp_character (&pp, mask);
324 pp_decimal_int (&pp, simdlen);
325
326 for (n = 0; n < clone_info->nargs; ++n)
327 {
328 struct cgraph_simd_clone_arg arg = clone_info->args[n];
329
330 switch (arg.arg_type)
331 {
332 case SIMD_CLONE_ARG_TYPE_UNIFORM:
333 pp_character (&pp, 'u');
334 break;
335 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
336 pp_character (&pp, 'l');
337 goto mangle_linear;
338 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP:
339 pp_character (&pp, 'R');
340 goto mangle_linear;
341 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP:
342 pp_character (&pp, 'L');
343 goto mangle_linear;
344 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP:
345 pp_character (&pp, 'U');
346 goto mangle_linear;
347 mangle_linear:
348 gcc_assert (arg.linear_step != 0);
349 if (arg.linear_step > 1)
350 pp_unsigned_wide_integer (&pp, arg.linear_step);
351 else if (arg.linear_step < 0)
352 {
353 pp_character (&pp, 'n');
354 pp_unsigned_wide_integer (&pp, (-(unsigned HOST_WIDE_INT)
355 arg.linear_step));
356 }
357 break;
358 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
359 pp_string (&pp, "ls");
360 pp_unsigned_wide_integer (&pp, arg.linear_step);
361 break;
362 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP:
363 pp_string (&pp, "Rs");
364 pp_unsigned_wide_integer (&pp, arg.linear_step);
365 break;
366 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
367 pp_string (&pp, "Ls");
368 pp_unsigned_wide_integer (&pp, arg.linear_step);
369 break;
370 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
371 pp_string (&pp, "Us");
372 pp_unsigned_wide_integer (&pp, arg.linear_step);
373 break;
374 default:
375 pp_character (&pp, 'v');
376 }
377 if (arg.alignment)
378 {
379 pp_character (&pp, 'a');
380 pp_decimal_int (&pp, arg.alignment);
381 }
382 }
383
384 pp_underscore (&pp);
385 const char *str = IDENTIFIER_POINTER (DECL_ASSEMBLER_NAME (node->decl));
386 if (*str == '*')
387 ++str;
388 pp_string (&pp, str);
389 str = pp_formatted_text (&pp);
390
391 /* If there already is a SIMD clone with the same mangled name, don't
392 add another one. This can happen e.g. for
393 #pragma omp declare simd
394 #pragma omp declare simd simdlen(8)
395 int foo (int, int);
396 if the simdlen is assumed to be 8 for the first one, etc. */
397 for (struct cgraph_node *clone = node->simd_clones; clone;
398 clone = clone->simdclone->next_clone)
399 if (strcmp (IDENTIFIER_POINTER (DECL_ASSEMBLER_NAME (clone->decl)),
400 str) == 0)
401 return NULL_TREE;
402
403 return get_identifier (str);
404 }
405
406 /* Create a simd clone of OLD_NODE and return it. */
407
408 static struct cgraph_node *
409 simd_clone_create (struct cgraph_node *old_node)
410 {
411 struct cgraph_node *new_node;
412 if (old_node->definition)
413 {
414 if (!old_node->has_gimple_body_p ())
415 return NULL;
416 old_node->get_body ();
417 new_node = old_node->create_version_clone_with_body (vNULL, NULL, NULL,
418 false, NULL, NULL,
419 "simdclone");
420 }
421 else
422 {
423 tree old_decl = old_node->decl;
424 tree new_decl = copy_node (old_node->decl);
425 DECL_NAME (new_decl) = clone_function_name (old_decl, "simdclone");
426 SET_DECL_ASSEMBLER_NAME (new_decl, DECL_NAME (new_decl));
427 SET_DECL_RTL (new_decl, NULL);
428 DECL_STATIC_CONSTRUCTOR (new_decl) = 0;
429 DECL_STATIC_DESTRUCTOR (new_decl) = 0;
430 new_node = old_node->create_version_clone (new_decl, vNULL, NULL);
431 if (old_node->in_other_partition)
432 new_node->in_other_partition = 1;
433 }
434 if (new_node == NULL)
435 return new_node;
436
437 TREE_PUBLIC (new_node->decl) = TREE_PUBLIC (old_node->decl);
438
439 /* The function cgraph_function_versioning () will force the new
440 symbol local. Undo this, and inherit external visability from
441 the old node. */
442 new_node->local.local = old_node->local.local;
443 new_node->externally_visible = old_node->externally_visible;
444
445 return new_node;
446 }
447
448 /* Adjust the return type of the given function to its appropriate
449 vector counterpart. Returns a simd array to be used throughout the
450 function as a return value. */
451
452 static tree
453 simd_clone_adjust_return_type (struct cgraph_node *node)
454 {
455 tree fndecl = node->decl;
456 tree orig_rettype = TREE_TYPE (TREE_TYPE (fndecl));
457 unsigned int veclen;
458 tree t;
459
460 /* Adjust the function return type. */
461 if (orig_rettype == void_type_node)
462 return NULL_TREE;
463 TREE_TYPE (fndecl) = build_distinct_type_copy (TREE_TYPE (fndecl));
464 t = TREE_TYPE (TREE_TYPE (fndecl));
465 if (INTEGRAL_TYPE_P (t) || POINTER_TYPE_P (t))
466 veclen = node->simdclone->vecsize_int;
467 else
468 veclen = node->simdclone->vecsize_float;
469 veclen /= GET_MODE_BITSIZE (TYPE_MODE (t));
470 if (veclen > node->simdclone->simdlen)
471 veclen = node->simdclone->simdlen;
472 if (POINTER_TYPE_P (t))
473 t = pointer_sized_int_node;
474 if (veclen == node->simdclone->simdlen)
475 t = build_vector_type (t, node->simdclone->simdlen);
476 else
477 {
478 t = build_vector_type (t, veclen);
479 t = build_array_type_nelts (t, node->simdclone->simdlen / veclen);
480 }
481 TREE_TYPE (TREE_TYPE (fndecl)) = t;
482 if (!node->definition)
483 return NULL_TREE;
484
485 t = DECL_RESULT (fndecl);
486 /* Adjust the DECL_RESULT. */
487 gcc_assert (TREE_TYPE (t) != void_type_node);
488 TREE_TYPE (t) = TREE_TYPE (TREE_TYPE (fndecl));
489 relayout_decl (t);
490
491 tree atype = build_array_type_nelts (orig_rettype,
492 node->simdclone->simdlen);
493 if (veclen != node->simdclone->simdlen)
494 return build1 (VIEW_CONVERT_EXPR, atype, t);
495
496 /* Set up a SIMD array to use as the return value. */
497 tree retval = create_tmp_var_raw (atype, "retval");
498 gimple_add_tmp_var (retval);
499 return retval;
500 }
501
502 /* Each vector argument has a corresponding array to be used locally
503 as part of the eventual loop. Create such temporary array and
504 return it.
505
506 PREFIX is the prefix to be used for the temporary.
507
508 TYPE is the inner element type.
509
510 SIMDLEN is the number of elements. */
511
512 static tree
513 create_tmp_simd_array (const char *prefix, tree type, int simdlen)
514 {
515 tree atype = build_array_type_nelts (type, simdlen);
516 tree avar = create_tmp_var_raw (atype, prefix);
517 gimple_add_tmp_var (avar);
518 return avar;
519 }
520
521 /* Modify the function argument types to their corresponding vector
522 counterparts if appropriate. Also, create one array for each simd
523 argument to be used locally when using the function arguments as
524 part of the loop.
525
526 NODE is the function whose arguments are to be adjusted.
527
528 Returns an adjustment vector that will be filled describing how the
529 argument types will be adjusted. */
530
531 static ipa_parm_adjustment_vec
532 simd_clone_adjust_argument_types (struct cgraph_node *node)
533 {
534 vec<tree> args;
535 ipa_parm_adjustment_vec adjustments;
536
537 if (node->definition)
538 args = ipa_get_vector_of_formal_parms (node->decl);
539 else
540 args = simd_clone_vector_of_formal_parm_types (node->decl);
541 adjustments.create (args.length ());
542 unsigned i, j, veclen;
543 struct ipa_parm_adjustment adj;
544 struct cgraph_simd_clone *sc = node->simdclone;
545
546 for (i = 0; i < sc->nargs; ++i)
547 {
548 memset (&adj, 0, sizeof (adj));
549 tree parm = args[i];
550 tree parm_type = node->definition ? TREE_TYPE (parm) : parm;
551 adj.base_index = i;
552 adj.base = parm;
553
554 sc->args[i].orig_arg = node->definition ? parm : NULL_TREE;
555 sc->args[i].orig_type = parm_type;
556
557 switch (sc->args[i].arg_type)
558 {
559 default:
560 /* No adjustment necessary for scalar arguments. */
561 adj.op = IPA_PARM_OP_COPY;
562 break;
563 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP:
564 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
565 if (node->definition)
566 sc->args[i].simd_array
567 = create_tmp_simd_array (IDENTIFIER_POINTER (DECL_NAME (parm)),
568 TREE_TYPE (parm_type),
569 sc->simdlen);
570 adj.op = IPA_PARM_OP_COPY;
571 break;
572 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP:
573 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
574 case SIMD_CLONE_ARG_TYPE_VECTOR:
575 if (INTEGRAL_TYPE_P (parm_type) || POINTER_TYPE_P (parm_type))
576 veclen = sc->vecsize_int;
577 else
578 veclen = sc->vecsize_float;
579 veclen /= GET_MODE_BITSIZE (TYPE_MODE (parm_type));
580 if (veclen > sc->simdlen)
581 veclen = sc->simdlen;
582 adj.arg_prefix = "simd";
583 if (POINTER_TYPE_P (parm_type))
584 adj.type = build_vector_type (pointer_sized_int_node, veclen);
585 else
586 adj.type = build_vector_type (parm_type, veclen);
587 sc->args[i].vector_type = adj.type;
588 for (j = veclen; j < sc->simdlen; j += veclen)
589 {
590 adjustments.safe_push (adj);
591 if (j == veclen)
592 {
593 memset (&adj, 0, sizeof (adj));
594 adj.op = IPA_PARM_OP_NEW;
595 adj.arg_prefix = "simd";
596 adj.base_index = i;
597 adj.type = sc->args[i].vector_type;
598 }
599 }
600
601 if (node->definition)
602 sc->args[i].simd_array
603 = create_tmp_simd_array (IDENTIFIER_POINTER (DECL_NAME (parm)),
604 parm_type, sc->simdlen);
605 }
606 adjustments.safe_push (adj);
607 }
608
609 if (sc->inbranch)
610 {
611 tree base_type = simd_clone_compute_base_data_type (sc->origin, sc);
612
613 memset (&adj, 0, sizeof (adj));
614 adj.op = IPA_PARM_OP_NEW;
615 adj.arg_prefix = "mask";
616
617 adj.base_index = i;
618 if (INTEGRAL_TYPE_P (base_type) || POINTER_TYPE_P (base_type))
619 veclen = sc->vecsize_int;
620 else
621 veclen = sc->vecsize_float;
622 veclen /= GET_MODE_BITSIZE (TYPE_MODE (base_type));
623 if (veclen > sc->simdlen)
624 veclen = sc->simdlen;
625 if (sc->mask_mode != VOIDmode)
626 adj.type
627 = lang_hooks.types.type_for_mode (sc->mask_mode, 1);
628 else if (POINTER_TYPE_P (base_type))
629 adj.type = build_vector_type (pointer_sized_int_node, veclen);
630 else
631 adj.type = build_vector_type (base_type, veclen);
632 adjustments.safe_push (adj);
633
634 for (j = veclen; j < sc->simdlen; j += veclen)
635 adjustments.safe_push (adj);
636
637 /* We have previously allocated one extra entry for the mask. Use
638 it and fill it. */
639 sc->nargs++;
640 if (sc->mask_mode != VOIDmode)
641 base_type = boolean_type_node;
642 if (node->definition)
643 {
644 sc->args[i].orig_arg
645 = build_decl (UNKNOWN_LOCATION, PARM_DECL, NULL, base_type);
646 if (sc->mask_mode == VOIDmode)
647 sc->args[i].simd_array
648 = create_tmp_simd_array ("mask", base_type, sc->simdlen);
649 else if (veclen < sc->simdlen)
650 sc->args[i].simd_array
651 = create_tmp_simd_array ("mask", adj.type, sc->simdlen / veclen);
652 else
653 sc->args[i].simd_array = NULL_TREE;
654 }
655 sc->args[i].orig_type = base_type;
656 sc->args[i].arg_type = SIMD_CLONE_ARG_TYPE_MASK;
657 }
658
659 if (node->definition)
660 ipa_modify_formal_parameters (node->decl, adjustments);
661 else
662 {
663 tree new_arg_types = NULL_TREE, new_reversed;
664 bool last_parm_void = false;
665 if (args.length () > 0 && args.last () == void_type_node)
666 last_parm_void = true;
667
668 gcc_assert (TYPE_ARG_TYPES (TREE_TYPE (node->decl)));
669 j = adjustments.length ();
670 for (i = 0; i < j; i++)
671 {
672 struct ipa_parm_adjustment *adj = &adjustments[i];
673 tree ptype;
674 if (adj->op == IPA_PARM_OP_COPY)
675 ptype = args[adj->base_index];
676 else
677 ptype = adj->type;
678 new_arg_types = tree_cons (NULL_TREE, ptype, new_arg_types);
679 }
680 new_reversed = nreverse (new_arg_types);
681 if (last_parm_void)
682 {
683 if (new_reversed)
684 TREE_CHAIN (new_arg_types) = void_list_node;
685 else
686 new_reversed = void_list_node;
687 }
688
689 tree new_type = build_distinct_type_copy (TREE_TYPE (node->decl));
690 TYPE_ARG_TYPES (new_type) = new_reversed;
691 TREE_TYPE (node->decl) = new_type;
692
693 adjustments.release ();
694 }
695 args.release ();
696 return adjustments;
697 }
698
699 /* Initialize and copy the function arguments in NODE to their
700 corresponding local simd arrays. Returns a fresh gimple_seq with
701 the instruction sequence generated. */
702
703 static gimple_seq
704 simd_clone_init_simd_arrays (struct cgraph_node *node,
705 ipa_parm_adjustment_vec adjustments)
706 {
707 gimple_seq seq = NULL;
708 unsigned i = 0, j = 0, k;
709
710 for (tree arg = DECL_ARGUMENTS (node->decl);
711 arg;
712 arg = DECL_CHAIN (arg), i++, j++)
713 {
714 if (adjustments[j].op == IPA_PARM_OP_COPY
715 || POINTER_TYPE_P (TREE_TYPE (arg)))
716 continue;
717
718 node->simdclone->args[i].vector_arg = arg;
719
720 tree array = node->simdclone->args[i].simd_array;
721 if (node->simdclone->mask_mode != VOIDmode
722 && node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_MASK)
723 {
724 if (array == NULL_TREE)
725 continue;
726 unsigned int l
727 = tree_to_uhwi (TYPE_MAX_VALUE (TYPE_DOMAIN (TREE_TYPE (array))));
728 for (k = 0; k <= l; k++)
729 {
730 if (k)
731 {
732 arg = DECL_CHAIN (arg);
733 j++;
734 }
735 tree t = build4 (ARRAY_REF, TREE_TYPE (TREE_TYPE (array)),
736 array, size_int (k), NULL, NULL);
737 t = build2 (MODIFY_EXPR, TREE_TYPE (t), t, arg);
738 gimplify_and_add (t, &seq);
739 }
740 continue;
741 }
742 if (TYPE_VECTOR_SUBPARTS (TREE_TYPE (arg)) == node->simdclone->simdlen)
743 {
744 tree ptype = build_pointer_type (TREE_TYPE (TREE_TYPE (array)));
745 tree ptr = build_fold_addr_expr (array);
746 tree t = build2 (MEM_REF, TREE_TYPE (arg), ptr,
747 build_int_cst (ptype, 0));
748 t = build2 (MODIFY_EXPR, TREE_TYPE (t), t, arg);
749 gimplify_and_add (t, &seq);
750 }
751 else
752 {
753 unsigned int simdlen = TYPE_VECTOR_SUBPARTS (TREE_TYPE (arg));
754 tree ptype = build_pointer_type (TREE_TYPE (TREE_TYPE (array)));
755 for (k = 0; k < node->simdclone->simdlen; k += simdlen)
756 {
757 tree ptr = build_fold_addr_expr (array);
758 int elemsize;
759 if (k)
760 {
761 arg = DECL_CHAIN (arg);
762 j++;
763 }
764 elemsize
765 = GET_MODE_SIZE (TYPE_MODE (TREE_TYPE (TREE_TYPE (arg))));
766 tree t = build2 (MEM_REF, TREE_TYPE (arg), ptr,
767 build_int_cst (ptype, k * elemsize));
768 t = build2 (MODIFY_EXPR, TREE_TYPE (t), t, arg);
769 gimplify_and_add (t, &seq);
770 }
771 }
772 }
773 return seq;
774 }
775
776 /* Callback info for ipa_simd_modify_stmt_ops below. */
777
778 struct modify_stmt_info {
779 ipa_parm_adjustment_vec adjustments;
780 gimple *stmt;
781 /* True if the parent statement was modified by
782 ipa_simd_modify_stmt_ops. */
783 bool modified;
784 };
785
786 /* Callback for walk_gimple_op.
787
788 Adjust operands from a given statement as specified in the
789 adjustments vector in the callback data. */
790
791 static tree
792 ipa_simd_modify_stmt_ops (tree *tp, int *walk_subtrees, void *data)
793 {
794 struct walk_stmt_info *wi = (struct walk_stmt_info *) data;
795 struct modify_stmt_info *info = (struct modify_stmt_info *) wi->info;
796 tree *orig_tp = tp;
797 if (TREE_CODE (*tp) == ADDR_EXPR)
798 tp = &TREE_OPERAND (*tp, 0);
799 struct ipa_parm_adjustment *cand = NULL;
800 if (TREE_CODE (*tp) == PARM_DECL)
801 cand = ipa_get_adjustment_candidate (&tp, NULL, info->adjustments, true);
802 else
803 {
804 if (TYPE_P (*tp))
805 *walk_subtrees = 0;
806 }
807
808 tree repl = NULL_TREE;
809 if (cand)
810 repl = unshare_expr (cand->new_decl);
811 else
812 {
813 if (tp != orig_tp)
814 {
815 *walk_subtrees = 0;
816 bool modified = info->modified;
817 info->modified = false;
818 walk_tree (tp, ipa_simd_modify_stmt_ops, wi, wi->pset);
819 if (!info->modified)
820 {
821 info->modified = modified;
822 return NULL_TREE;
823 }
824 info->modified = modified;
825 repl = *tp;
826 }
827 else
828 return NULL_TREE;
829 }
830
831 if (tp != orig_tp)
832 {
833 repl = build_fold_addr_expr (repl);
834 gimple *stmt;
835 if (is_gimple_debug (info->stmt))
836 {
837 tree vexpr = make_node (DEBUG_EXPR_DECL);
838 stmt = gimple_build_debug_source_bind (vexpr, repl, NULL);
839 DECL_ARTIFICIAL (vexpr) = 1;
840 TREE_TYPE (vexpr) = TREE_TYPE (repl);
841 DECL_MODE (vexpr) = TYPE_MODE (TREE_TYPE (repl));
842 repl = vexpr;
843 }
844 else
845 {
846 stmt = gimple_build_assign (make_ssa_name (TREE_TYPE (repl)), repl);
847 repl = gimple_assign_lhs (stmt);
848 }
849 gimple_stmt_iterator gsi = gsi_for_stmt (info->stmt);
850 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
851 *orig_tp = repl;
852 }
853 else if (!useless_type_conversion_p (TREE_TYPE (*tp), TREE_TYPE (repl)))
854 {
855 tree vce = build1 (VIEW_CONVERT_EXPR, TREE_TYPE (*tp), repl);
856 *tp = vce;
857 }
858 else
859 *tp = repl;
860
861 info->modified = true;
862 return NULL_TREE;
863 }
864
865 /* Traverse the function body and perform all modifications as
866 described in ADJUSTMENTS. At function return, ADJUSTMENTS will be
867 modified such that the replacement/reduction value will now be an
868 offset into the corresponding simd_array.
869
870 This function will replace all function argument uses with their
871 corresponding simd array elements, and ajust the return values
872 accordingly. */
873
874 static void
875 ipa_simd_modify_function_body (struct cgraph_node *node,
876 ipa_parm_adjustment_vec adjustments,
877 tree retval_array, tree iter)
878 {
879 basic_block bb;
880 unsigned int i, j, l;
881
882 /* Re-use the adjustments array, but this time use it to replace
883 every function argument use to an offset into the corresponding
884 simd_array. */
885 for (i = 0, j = 0; i < node->simdclone->nargs; ++i, ++j)
886 {
887 if (!node->simdclone->args[i].vector_arg)
888 continue;
889
890 tree basetype = TREE_TYPE (node->simdclone->args[i].orig_arg);
891 tree vectype = TREE_TYPE (node->simdclone->args[i].vector_arg);
892 adjustments[j].new_decl
893 = build4 (ARRAY_REF,
894 basetype,
895 node->simdclone->args[i].simd_array,
896 iter,
897 NULL_TREE, NULL_TREE);
898 if (adjustments[j].op == IPA_PARM_OP_NONE
899 && TYPE_VECTOR_SUBPARTS (vectype) < node->simdclone->simdlen)
900 j += node->simdclone->simdlen / TYPE_VECTOR_SUBPARTS (vectype) - 1;
901 }
902
903 l = adjustments.length ();
904 for (i = 1; i < num_ssa_names; i++)
905 {
906 tree name = ssa_name (i);
907 if (name
908 && SSA_NAME_VAR (name)
909 && TREE_CODE (SSA_NAME_VAR (name)) == PARM_DECL)
910 {
911 for (j = 0; j < l; j++)
912 if (SSA_NAME_VAR (name) == adjustments[j].base
913 && adjustments[j].new_decl)
914 {
915 tree base_var;
916 if (adjustments[j].new_ssa_base == NULL_TREE)
917 {
918 base_var
919 = copy_var_decl (adjustments[j].base,
920 DECL_NAME (adjustments[j].base),
921 TREE_TYPE (adjustments[j].base));
922 adjustments[j].new_ssa_base = base_var;
923 }
924 else
925 base_var = adjustments[j].new_ssa_base;
926 if (SSA_NAME_IS_DEFAULT_DEF (name))
927 {
928 bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
929 gimple_stmt_iterator gsi = gsi_after_labels (bb);
930 tree new_decl = unshare_expr (adjustments[j].new_decl);
931 set_ssa_default_def (cfun, adjustments[j].base, NULL_TREE);
932 SET_SSA_NAME_VAR_OR_IDENTIFIER (name, base_var);
933 SSA_NAME_IS_DEFAULT_DEF (name) = 0;
934 gimple *stmt = gimple_build_assign (name, new_decl);
935 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
936 }
937 else
938 SET_SSA_NAME_VAR_OR_IDENTIFIER (name, base_var);
939 }
940 }
941 }
942
943 struct modify_stmt_info info;
944 info.adjustments = adjustments;
945
946 FOR_EACH_BB_FN (bb, DECL_STRUCT_FUNCTION (node->decl))
947 {
948 gimple_stmt_iterator gsi;
949
950 gsi = gsi_start_bb (bb);
951 while (!gsi_end_p (gsi))
952 {
953 gimple *stmt = gsi_stmt (gsi);
954 info.stmt = stmt;
955 struct walk_stmt_info wi;
956
957 memset (&wi, 0, sizeof (wi));
958 info.modified = false;
959 wi.info = &info;
960 walk_gimple_op (stmt, ipa_simd_modify_stmt_ops, &wi);
961
962 if (greturn *return_stmt = dyn_cast <greturn *> (stmt))
963 {
964 tree retval = gimple_return_retval (return_stmt);
965 if (!retval)
966 {
967 gsi_remove (&gsi, true);
968 continue;
969 }
970
971 /* Replace `return foo' with `retval_array[iter] = foo'. */
972 tree ref = build4 (ARRAY_REF, TREE_TYPE (retval),
973 retval_array, iter, NULL, NULL);
974 stmt = gimple_build_assign (ref, retval);
975 gsi_replace (&gsi, stmt, true);
976 info.modified = true;
977 }
978
979 if (info.modified)
980 {
981 update_stmt (stmt);
982 if (maybe_clean_eh_stmt (stmt))
983 gimple_purge_dead_eh_edges (gimple_bb (stmt));
984 }
985 gsi_next (&gsi);
986 }
987 }
988 }
989
990 /* Helper function of simd_clone_adjust, return linear step addend
991 of Ith argument. */
992
993 static tree
994 simd_clone_linear_addend (struct cgraph_node *node, unsigned int i,
995 tree addtype, basic_block entry_bb)
996 {
997 tree ptype = NULL_TREE;
998 switch (node->simdclone->args[i].arg_type)
999 {
1000 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
1001 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP:
1002 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP:
1003 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP:
1004 return build_int_cst (addtype, node->simdclone->args[i].linear_step);
1005 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
1006 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP:
1007 ptype = TREE_TYPE (node->simdclone->args[i].orig_arg);
1008 break;
1009 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
1010 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
1011 ptype = TREE_TYPE (TREE_TYPE (node->simdclone->args[i].orig_arg));
1012 break;
1013 default:
1014 gcc_unreachable ();
1015 }
1016
1017 unsigned int idx = node->simdclone->args[i].linear_step;
1018 tree arg = node->simdclone->args[idx].orig_arg;
1019 gcc_assert (is_gimple_reg_type (TREE_TYPE (arg)));
1020 gimple_stmt_iterator gsi = gsi_after_labels (entry_bb);
1021 gimple *g;
1022 tree ret;
1023 if (is_gimple_reg (arg))
1024 ret = get_or_create_ssa_default_def (cfun, arg);
1025 else
1026 {
1027 g = gimple_build_assign (make_ssa_name (TREE_TYPE (arg)), arg);
1028 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1029 ret = gimple_assign_lhs (g);
1030 }
1031 if (TREE_CODE (TREE_TYPE (arg)) == REFERENCE_TYPE)
1032 {
1033 g = gimple_build_assign (make_ssa_name (TREE_TYPE (TREE_TYPE (arg))),
1034 build_simple_mem_ref (ret));
1035 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1036 ret = gimple_assign_lhs (g);
1037 }
1038 if (!useless_type_conversion_p (addtype, TREE_TYPE (ret)))
1039 {
1040 g = gimple_build_assign (make_ssa_name (addtype), NOP_EXPR, ret);
1041 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1042 ret = gimple_assign_lhs (g);
1043 }
1044 if (POINTER_TYPE_P (ptype))
1045 {
1046 tree size = TYPE_SIZE_UNIT (TREE_TYPE (ptype));
1047 if (size && TREE_CODE (size) == INTEGER_CST)
1048 {
1049 g = gimple_build_assign (make_ssa_name (addtype), MULT_EXPR,
1050 ret, fold_convert (addtype, size));
1051 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1052 ret = gimple_assign_lhs (g);
1053 }
1054 }
1055 return ret;
1056 }
1057
1058 /* Adjust the argument types in NODE to their appropriate vector
1059 counterparts. */
1060
1061 static void
1062 simd_clone_adjust (struct cgraph_node *node)
1063 {
1064 push_cfun (DECL_STRUCT_FUNCTION (node->decl));
1065
1066 targetm.simd_clone.adjust (node);
1067
1068 tree retval = simd_clone_adjust_return_type (node);
1069 ipa_parm_adjustment_vec adjustments
1070 = simd_clone_adjust_argument_types (node);
1071
1072 push_gimplify_context ();
1073
1074 gimple_seq seq = simd_clone_init_simd_arrays (node, adjustments);
1075
1076 /* Adjust all uses of vector arguments accordingly. Adjust all
1077 return values accordingly. */
1078 tree iter = create_tmp_var (unsigned_type_node, "iter");
1079 tree iter1 = make_ssa_name (iter);
1080 tree iter2 = make_ssa_name (iter);
1081 ipa_simd_modify_function_body (node, adjustments, retval, iter1);
1082 adjustments.release ();
1083
1084 /* Initialize the iteration variable. */
1085 basic_block entry_bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1086 basic_block body_bb = split_block_after_labels (entry_bb)->dest;
1087 gimple_stmt_iterator gsi = gsi_after_labels (entry_bb);
1088 /* Insert the SIMD array and iv initialization at function
1089 entry. */
1090 gsi_insert_seq_before (&gsi, seq, GSI_NEW_STMT);
1091
1092 pop_gimplify_context (NULL);
1093
1094 /* Create a new BB right before the original exit BB, to hold the
1095 iteration increment and the condition/branch. */
1096 basic_block orig_exit = EDGE_PRED (EXIT_BLOCK_PTR_FOR_FN (cfun), 0)->src;
1097 basic_block incr_bb = create_empty_bb (orig_exit);
1098 add_bb_to_loop (incr_bb, body_bb->loop_father);
1099 /* The succ of orig_exit was EXIT_BLOCK_PTR_FOR_FN (cfun), with an empty
1100 flag. Set it now to be a FALLTHRU_EDGE. */
1101 gcc_assert (EDGE_COUNT (orig_exit->succs) == 1);
1102 EDGE_SUCC (orig_exit, 0)->flags |= EDGE_FALLTHRU;
1103 for (unsigned i = 0;
1104 i < EDGE_COUNT (EXIT_BLOCK_PTR_FOR_FN (cfun)->preds); ++i)
1105 {
1106 edge e = EDGE_PRED (EXIT_BLOCK_PTR_FOR_FN (cfun), i);
1107 redirect_edge_succ (e, incr_bb);
1108 }
1109 edge e = make_edge (incr_bb, EXIT_BLOCK_PTR_FOR_FN (cfun), 0);
1110 e->probability = REG_BR_PROB_BASE;
1111 gsi = gsi_last_bb (incr_bb);
1112 gimple *g = gimple_build_assign (iter2, PLUS_EXPR, iter1,
1113 build_int_cst (unsigned_type_node, 1));
1114 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1115
1116 /* Mostly annotate the loop for the vectorizer (the rest is done below). */
1117 struct loop *loop = alloc_loop ();
1118 cfun->has_force_vectorize_loops = true;
1119 loop->safelen = node->simdclone->simdlen;
1120 loop->force_vectorize = true;
1121 loop->header = body_bb;
1122
1123 /* Branch around the body if the mask applies. */
1124 if (node->simdclone->inbranch)
1125 {
1126 gimple_stmt_iterator gsi = gsi_last_bb (loop->header);
1127 tree mask_array
1128 = node->simdclone->args[node->simdclone->nargs - 1].simd_array;
1129 tree mask;
1130 if (node->simdclone->mask_mode != VOIDmode)
1131 {
1132 tree shift_cnt;
1133 if (mask_array == NULL_TREE)
1134 {
1135 tree arg = node->simdclone->args[node->simdclone->nargs
1136 - 1].vector_arg;
1137 mask = get_or_create_ssa_default_def (cfun, arg);
1138 shift_cnt = iter1;
1139 }
1140 else
1141 {
1142 tree maskt = TREE_TYPE (mask_array);
1143 int c = tree_to_uhwi (TYPE_MAX_VALUE (TYPE_DOMAIN (maskt)));
1144 c = node->simdclone->simdlen / (c + 1);
1145 int s = exact_log2 (c);
1146 gcc_assert (s > 0);
1147 c--;
1148 tree idx = make_ssa_name (TREE_TYPE (iter1));
1149 g = gimple_build_assign (idx, RSHIFT_EXPR, iter1,
1150 build_int_cst (NULL_TREE, s));
1151 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1152 mask = make_ssa_name (TREE_TYPE (TREE_TYPE (mask_array)));
1153 tree aref = build4 (ARRAY_REF,
1154 TREE_TYPE (TREE_TYPE (mask_array)),
1155 mask_array, idx, NULL, NULL);
1156 g = gimple_build_assign (mask, aref);
1157 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1158 shift_cnt = make_ssa_name (TREE_TYPE (iter1));
1159 g = gimple_build_assign (shift_cnt, BIT_AND_EXPR, iter1,
1160 build_int_cst (TREE_TYPE (iter1), c));
1161 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1162 }
1163 g = gimple_build_assign (make_ssa_name (TREE_TYPE (mask)),
1164 RSHIFT_EXPR, mask, shift_cnt);
1165 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1166 mask = gimple_assign_lhs (g);
1167 g = gimple_build_assign (make_ssa_name (TREE_TYPE (mask)),
1168 BIT_AND_EXPR, mask,
1169 build_int_cst (TREE_TYPE (mask), 1));
1170 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1171 mask = gimple_assign_lhs (g);
1172 }
1173 else
1174 {
1175 mask = make_ssa_name (TREE_TYPE (TREE_TYPE (mask_array)));
1176 tree aref = build4 (ARRAY_REF,
1177 TREE_TYPE (TREE_TYPE (mask_array)),
1178 mask_array, iter1, NULL, NULL);
1179 g = gimple_build_assign (mask, aref);
1180 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1181 int bitsize = GET_MODE_BITSIZE (TYPE_MODE (TREE_TYPE (aref)));
1182 if (!INTEGRAL_TYPE_P (TREE_TYPE (aref)))
1183 {
1184 aref = build1 (VIEW_CONVERT_EXPR,
1185 build_nonstandard_integer_type (bitsize, 0),
1186 mask);
1187 mask = make_ssa_name (TREE_TYPE (aref));
1188 g = gimple_build_assign (mask, aref);
1189 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1190 }
1191 }
1192
1193 g = gimple_build_cond (EQ_EXPR, mask, build_zero_cst (TREE_TYPE (mask)),
1194 NULL, NULL);
1195 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1196 make_edge (loop->header, incr_bb, EDGE_TRUE_VALUE);
1197 FALLTHRU_EDGE (loop->header)->flags = EDGE_FALSE_VALUE;
1198 }
1199
1200 /* Generate the condition. */
1201 g = gimple_build_cond (LT_EXPR,
1202 iter2,
1203 build_int_cst (unsigned_type_node,
1204 node->simdclone->simdlen),
1205 NULL, NULL);
1206 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1207 e = split_block (incr_bb, gsi_stmt (gsi));
1208 basic_block latch_bb = e->dest;
1209 basic_block new_exit_bb;
1210 new_exit_bb = split_block_after_labels (latch_bb)->dest;
1211 loop->latch = latch_bb;
1212
1213 redirect_edge_succ (FALLTHRU_EDGE (latch_bb), body_bb);
1214
1215 make_edge (incr_bb, new_exit_bb, EDGE_FALSE_VALUE);
1216 /* The successor of incr_bb is already pointing to latch_bb; just
1217 change the flags.
1218 make_edge (incr_bb, latch_bb, EDGE_TRUE_VALUE); */
1219 FALLTHRU_EDGE (incr_bb)->flags = EDGE_TRUE_VALUE;
1220
1221 gphi *phi = create_phi_node (iter1, body_bb);
1222 edge preheader_edge = find_edge (entry_bb, body_bb);
1223 edge latch_edge = single_succ_edge (latch_bb);
1224 add_phi_arg (phi, build_zero_cst (unsigned_type_node), preheader_edge,
1225 UNKNOWN_LOCATION);
1226 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1227
1228 /* Generate the new return. */
1229 gsi = gsi_last_bb (new_exit_bb);
1230 if (retval
1231 && TREE_CODE (retval) == VIEW_CONVERT_EXPR
1232 && TREE_CODE (TREE_OPERAND (retval, 0)) == RESULT_DECL)
1233 retval = TREE_OPERAND (retval, 0);
1234 else if (retval)
1235 {
1236 retval = build1 (VIEW_CONVERT_EXPR,
1237 TREE_TYPE (TREE_TYPE (node->decl)),
1238 retval);
1239 retval = force_gimple_operand_gsi (&gsi, retval, true, NULL,
1240 false, GSI_CONTINUE_LINKING);
1241 }
1242 g = gimple_build_return (retval);
1243 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1244
1245 /* Handle aligned clauses by replacing default defs of the aligned
1246 uniform args with __builtin_assume_aligned (arg_N(D), alignment)
1247 lhs. Handle linear by adding PHIs. */
1248 for (unsigned i = 0; i < node->simdclone->nargs; i++)
1249 if (node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_UNIFORM
1250 && (TREE_ADDRESSABLE (node->simdclone->args[i].orig_arg)
1251 || !is_gimple_reg_type
1252 (TREE_TYPE (node->simdclone->args[i].orig_arg))))
1253 {
1254 tree orig_arg = node->simdclone->args[i].orig_arg;
1255 if (is_gimple_reg_type (TREE_TYPE (orig_arg)))
1256 iter1 = make_ssa_name (TREE_TYPE (orig_arg));
1257 else
1258 {
1259 iter1 = create_tmp_var_raw (TREE_TYPE (orig_arg));
1260 gimple_add_tmp_var (iter1);
1261 }
1262 gsi = gsi_after_labels (entry_bb);
1263 g = gimple_build_assign (iter1, orig_arg);
1264 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1265 gsi = gsi_after_labels (body_bb);
1266 g = gimple_build_assign (orig_arg, iter1);
1267 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1268 }
1269 else if (node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_UNIFORM
1270 && DECL_BY_REFERENCE (node->simdclone->args[i].orig_arg)
1271 && TREE_CODE (TREE_TYPE (node->simdclone->args[i].orig_arg))
1272 == REFERENCE_TYPE
1273 && TREE_ADDRESSABLE
1274 (TREE_TYPE (TREE_TYPE (node->simdclone->args[i].orig_arg))))
1275 {
1276 tree orig_arg = node->simdclone->args[i].orig_arg;
1277 tree def = ssa_default_def (cfun, orig_arg);
1278 if (def && !has_zero_uses (def))
1279 {
1280 iter1 = create_tmp_var_raw (TREE_TYPE (TREE_TYPE (orig_arg)));
1281 gimple_add_tmp_var (iter1);
1282 gsi = gsi_after_labels (entry_bb);
1283 g = gimple_build_assign (iter1, build_simple_mem_ref (def));
1284 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1285 gsi = gsi_after_labels (body_bb);
1286 g = gimple_build_assign (build_simple_mem_ref (def), iter1);
1287 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1288 }
1289 }
1290 else if (node->simdclone->args[i].alignment
1291 && node->simdclone->args[i].arg_type
1292 == SIMD_CLONE_ARG_TYPE_UNIFORM
1293 && (node->simdclone->args[i].alignment
1294 & (node->simdclone->args[i].alignment - 1)) == 0
1295 && TREE_CODE (TREE_TYPE (node->simdclone->args[i].orig_arg))
1296 == POINTER_TYPE)
1297 {
1298 unsigned int alignment = node->simdclone->args[i].alignment;
1299 tree orig_arg = node->simdclone->args[i].orig_arg;
1300 tree def = ssa_default_def (cfun, orig_arg);
1301 if (def && !has_zero_uses (def))
1302 {
1303 tree fn = builtin_decl_explicit (BUILT_IN_ASSUME_ALIGNED);
1304 gimple_seq seq = NULL;
1305 bool need_cvt = false;
1306 gcall *call
1307 = gimple_build_call (fn, 2, def, size_int (alignment));
1308 g = call;
1309 if (!useless_type_conversion_p (TREE_TYPE (orig_arg),
1310 ptr_type_node))
1311 need_cvt = true;
1312 tree t = make_ssa_name (need_cvt ? ptr_type_node : orig_arg);
1313 gimple_call_set_lhs (g, t);
1314 gimple_seq_add_stmt_without_update (&seq, g);
1315 if (need_cvt)
1316 {
1317 t = make_ssa_name (orig_arg);
1318 g = gimple_build_assign (t, NOP_EXPR, gimple_call_lhs (g));
1319 gimple_seq_add_stmt_without_update (&seq, g);
1320 }
1321 gsi_insert_seq_on_edge_immediate
1322 (single_succ_edge (ENTRY_BLOCK_PTR_FOR_FN (cfun)), seq);
1323
1324 entry_bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1325 int freq = compute_call_stmt_bb_frequency (current_function_decl,
1326 entry_bb);
1327 node->create_edge (cgraph_node::get_create (fn),
1328 call, entry_bb->count, freq);
1329
1330 imm_use_iterator iter;
1331 use_operand_p use_p;
1332 gimple *use_stmt;
1333 tree repl = gimple_get_lhs (g);
1334 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1335 if (is_gimple_debug (use_stmt) || use_stmt == call)
1336 continue;
1337 else
1338 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1339 SET_USE (use_p, repl);
1340 }
1341 }
1342 else if ((node->simdclone->args[i].arg_type
1343 == SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP)
1344 || (node->simdclone->args[i].arg_type
1345 == SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP)
1346 || (node->simdclone->args[i].arg_type
1347 == SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP)
1348 || (node->simdclone->args[i].arg_type
1349 == SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP))
1350 {
1351 tree orig_arg = node->simdclone->args[i].orig_arg;
1352 gcc_assert (INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1353 || POINTER_TYPE_P (TREE_TYPE (orig_arg)));
1354 tree def = NULL_TREE;
1355 if (TREE_ADDRESSABLE (orig_arg))
1356 {
1357 def = make_ssa_name (TREE_TYPE (orig_arg));
1358 iter1 = make_ssa_name (TREE_TYPE (orig_arg));
1359 iter2 = make_ssa_name (TREE_TYPE (orig_arg));
1360 gsi = gsi_after_labels (entry_bb);
1361 g = gimple_build_assign (def, orig_arg);
1362 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1363 }
1364 else
1365 {
1366 def = ssa_default_def (cfun, orig_arg);
1367 if (!def || has_zero_uses (def))
1368 def = NULL_TREE;
1369 else
1370 {
1371 iter1 = make_ssa_name (orig_arg);
1372 iter2 = make_ssa_name (orig_arg);
1373 }
1374 }
1375 if (def)
1376 {
1377 phi = create_phi_node (iter1, body_bb);
1378 add_phi_arg (phi, def, preheader_edge, UNKNOWN_LOCATION);
1379 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1380 enum tree_code code = INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1381 ? PLUS_EXPR : POINTER_PLUS_EXPR;
1382 tree addtype = INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1383 ? TREE_TYPE (orig_arg) : sizetype;
1384 tree addcst = simd_clone_linear_addend (node, i, addtype,
1385 entry_bb);
1386 gsi = gsi_last_bb (incr_bb);
1387 g = gimple_build_assign (iter2, code, iter1, addcst);
1388 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1389
1390 imm_use_iterator iter;
1391 use_operand_p use_p;
1392 gimple *use_stmt;
1393 if (TREE_ADDRESSABLE (orig_arg))
1394 {
1395 gsi = gsi_after_labels (body_bb);
1396 g = gimple_build_assign (orig_arg, iter1);
1397 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1398 }
1399 else
1400 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1401 if (use_stmt == phi)
1402 continue;
1403 else
1404 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1405 SET_USE (use_p, iter1);
1406 }
1407 }
1408 else if (node->simdclone->args[i].arg_type
1409 == SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP
1410 || (node->simdclone->args[i].arg_type
1411 == SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP))
1412 {
1413 tree orig_arg = node->simdclone->args[i].orig_arg;
1414 tree def = ssa_default_def (cfun, orig_arg);
1415 gcc_assert (!TREE_ADDRESSABLE (orig_arg)
1416 && TREE_CODE (TREE_TYPE (orig_arg)) == REFERENCE_TYPE);
1417 if (def && !has_zero_uses (def))
1418 {
1419 tree rtype = TREE_TYPE (TREE_TYPE (orig_arg));
1420 iter1 = make_ssa_name (orig_arg);
1421 iter2 = make_ssa_name (orig_arg);
1422 tree iter3 = make_ssa_name (rtype);
1423 tree iter4 = make_ssa_name (rtype);
1424 tree iter5 = make_ssa_name (rtype);
1425 gsi = gsi_after_labels (entry_bb);
1426 gimple *load
1427 = gimple_build_assign (iter3, build_simple_mem_ref (def));
1428 gsi_insert_before (&gsi, load, GSI_NEW_STMT);
1429
1430 tree array = node->simdclone->args[i].simd_array;
1431 TREE_ADDRESSABLE (array) = 1;
1432 tree ptr = build_fold_addr_expr (array);
1433 phi = create_phi_node (iter1, body_bb);
1434 add_phi_arg (phi, ptr, preheader_edge, UNKNOWN_LOCATION);
1435 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1436 g = gimple_build_assign (iter2, POINTER_PLUS_EXPR, iter1,
1437 TYPE_SIZE_UNIT (TREE_TYPE (iter3)));
1438 gsi = gsi_last_bb (incr_bb);
1439 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1440
1441 phi = create_phi_node (iter4, body_bb);
1442 add_phi_arg (phi, iter3, preheader_edge, UNKNOWN_LOCATION);
1443 add_phi_arg (phi, iter5, latch_edge, UNKNOWN_LOCATION);
1444 enum tree_code code = INTEGRAL_TYPE_P (TREE_TYPE (iter3))
1445 ? PLUS_EXPR : POINTER_PLUS_EXPR;
1446 tree addtype = INTEGRAL_TYPE_P (TREE_TYPE (iter3))
1447 ? TREE_TYPE (iter3) : sizetype;
1448 tree addcst = simd_clone_linear_addend (node, i, addtype,
1449 entry_bb);
1450 g = gimple_build_assign (iter5, code, iter4, addcst);
1451 gsi = gsi_last_bb (incr_bb);
1452 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1453
1454 g = gimple_build_assign (build_simple_mem_ref (iter1), iter4);
1455 gsi = gsi_after_labels (body_bb);
1456 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1457
1458 imm_use_iterator iter;
1459 use_operand_p use_p;
1460 gimple *use_stmt;
1461 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1462 if (use_stmt == load)
1463 continue;
1464 else
1465 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1466 SET_USE (use_p, iter1);
1467
1468 if (!TYPE_READONLY (rtype))
1469 {
1470 tree v = make_ssa_name (rtype);
1471 tree aref = build4 (ARRAY_REF, rtype, array,
1472 size_zero_node, NULL_TREE,
1473 NULL_TREE);
1474 gsi = gsi_after_labels (new_exit_bb);
1475 g = gimple_build_assign (v, aref);
1476 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1477 g = gimple_build_assign (build_simple_mem_ref (def), v);
1478 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1479 }
1480 }
1481 }
1482
1483 calculate_dominance_info (CDI_DOMINATORS);
1484 add_loop (loop, loop->header->loop_father);
1485 update_ssa (TODO_update_ssa);
1486
1487 pop_cfun ();
1488 }
1489
1490 /* If the function in NODE is tagged as an elemental SIMD function,
1491 create the appropriate SIMD clones. */
1492
1493 static void
1494 expand_simd_clones (struct cgraph_node *node)
1495 {
1496 tree attr = lookup_attribute ("omp declare simd",
1497 DECL_ATTRIBUTES (node->decl));
1498 if (attr == NULL_TREE
1499 || node->global.inlined_to
1500 || lookup_attribute ("noclone", DECL_ATTRIBUTES (node->decl)))
1501 return;
1502
1503 /* Ignore
1504 #pragma omp declare simd
1505 extern int foo ();
1506 in C, there we don't know the argument types at all. */
1507 if (!node->definition
1508 && TYPE_ARG_TYPES (TREE_TYPE (node->decl)) == NULL_TREE)
1509 return;
1510
1511 /* Call this before creating clone_info, as it might ggc_collect. */
1512 if (node->definition && node->has_gimple_body_p ())
1513 node->get_body ();
1514
1515 do
1516 {
1517 /* Start with parsing the "omp declare simd" attribute(s). */
1518 bool inbranch_clause_specified;
1519 struct cgraph_simd_clone *clone_info
1520 = simd_clone_clauses_extract (node, TREE_VALUE (attr),
1521 &inbranch_clause_specified);
1522 if (clone_info == NULL)
1523 continue;
1524
1525 int orig_simdlen = clone_info->simdlen;
1526 tree base_type = simd_clone_compute_base_data_type (node, clone_info);
1527 /* The target can return 0 (no simd clones should be created),
1528 1 (just one ISA of simd clones should be created) or higher
1529 count of ISA variants. In that case, clone_info is initialized
1530 for the first ISA variant. */
1531 int count
1532 = targetm.simd_clone.compute_vecsize_and_simdlen (node, clone_info,
1533 base_type, 0);
1534 if (count == 0)
1535 continue;
1536
1537 /* Loop over all COUNT ISA variants, and if !INBRANCH_CLAUSE_SPECIFIED,
1538 also create one inbranch and one !inbranch clone of it. */
1539 for (int i = 0; i < count * 2; i++)
1540 {
1541 struct cgraph_simd_clone *clone = clone_info;
1542 if (inbranch_clause_specified && (i & 1) != 0)
1543 continue;
1544
1545 if (i != 0)
1546 {
1547 clone = simd_clone_struct_alloc (clone_info->nargs
1548 + ((i & 1) != 0));
1549 simd_clone_struct_copy (clone, clone_info);
1550 /* Undo changes targetm.simd_clone.compute_vecsize_and_simdlen
1551 and simd_clone_adjust_argument_types did to the first
1552 clone's info. */
1553 clone->nargs -= clone_info->inbranch;
1554 clone->simdlen = orig_simdlen;
1555 /* And call the target hook again to get the right ISA. */
1556 targetm.simd_clone.compute_vecsize_and_simdlen (node, clone,
1557 base_type,
1558 i / 2);
1559 if ((i & 1) != 0)
1560 clone->inbranch = 1;
1561 }
1562
1563 /* simd_clone_mangle might fail if such a clone has been created
1564 already. */
1565 tree id = simd_clone_mangle (node, clone);
1566 if (id == NULL_TREE)
1567 continue;
1568
1569 /* Only when we are sure we want to create the clone actually
1570 clone the function (or definitions) or create another
1571 extern FUNCTION_DECL (for prototypes without definitions). */
1572 struct cgraph_node *n = simd_clone_create (node);
1573 if (n == NULL)
1574 continue;
1575
1576 n->simdclone = clone;
1577 clone->origin = node;
1578 clone->next_clone = NULL;
1579 if (node->simd_clones == NULL)
1580 {
1581 clone->prev_clone = n;
1582 node->simd_clones = n;
1583 }
1584 else
1585 {
1586 clone->prev_clone = node->simd_clones->simdclone->prev_clone;
1587 clone->prev_clone->simdclone->next_clone = n;
1588 node->simd_clones->simdclone->prev_clone = n;
1589 }
1590 symtab->change_decl_assembler_name (n->decl, id);
1591 /* And finally adjust the return type, parameters and for
1592 definitions also function body. */
1593 if (node->definition)
1594 simd_clone_adjust (n);
1595 else
1596 {
1597 simd_clone_adjust_return_type (n);
1598 simd_clone_adjust_argument_types (n);
1599 }
1600 }
1601 }
1602 while ((attr = lookup_attribute ("omp declare simd", TREE_CHAIN (attr))));
1603 }
1604
1605 /* Entry point for IPA simd clone creation pass. */
1606
1607 static unsigned int
1608 ipa_omp_simd_clone (void)
1609 {
1610 struct cgraph_node *node;
1611 FOR_EACH_FUNCTION (node)
1612 expand_simd_clones (node);
1613 return 0;
1614 }
1615
1616 namespace {
1617
1618 const pass_data pass_data_omp_simd_clone =
1619 {
1620 SIMPLE_IPA_PASS, /* type */
1621 "simdclone", /* name */
1622 OPTGROUP_NONE, /* optinfo_flags */
1623 TV_NONE, /* tv_id */
1624 ( PROP_ssa | PROP_cfg ), /* properties_required */
1625 0, /* properties_provided */
1626 0, /* properties_destroyed */
1627 0, /* todo_flags_start */
1628 0, /* todo_flags_finish */
1629 };
1630
1631 class pass_omp_simd_clone : public simple_ipa_opt_pass
1632 {
1633 public:
1634 pass_omp_simd_clone(gcc::context *ctxt)
1635 : simple_ipa_opt_pass(pass_data_omp_simd_clone, ctxt)
1636 {}
1637
1638 /* opt_pass methods: */
1639 virtual bool gate (function *);
1640 virtual unsigned int execute (function *) { return ipa_omp_simd_clone (); }
1641 };
1642
1643 bool
1644 pass_omp_simd_clone::gate (function *)
1645 {
1646 return targetm.simd_clone.compute_vecsize_and_simdlen != NULL;
1647 }
1648
1649 } // anon namespace
1650
1651 simple_ipa_opt_pass *
1652 make_pass_omp_simd_clone (gcc::context *ctxt)
1653 {
1654 return new pass_omp_simd_clone (ctxt);
1655 }