PR middle-end/84095 - false-positive -Wrestrict warnings for memcpy within array
[official-gcc.git] / gcc / omp-simd-clone.c
blob56832ebf22d17af73b6c25e61eec7c71b83a86bc
1 /* OMP constructs' SIMD clone supporting code.
3 Copyright (C) 2005-2018 Free Software Foundation, Inc.
5 This file is part of GCC.
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.
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.
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/>. */
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-param-manipulation.h"
49 #include "tree-eh.h"
50 #include "varasm.h"
51 #include "stringpool.h"
52 #include "attribs.h"
53 #include "omp-simd-clone.h"
55 /* Return the number of elements in vector type VECTYPE, which is associated
56 with a SIMD clone. At present these always have a constant length. */
58 static unsigned HOST_WIDE_INT
59 simd_clone_subparts (tree vectype)
61 return TYPE_VECTOR_SUBPARTS (vectype).to_constant ();
64 /* Allocate a fresh `simd_clone' and return it. NARGS is the number
65 of arguments to reserve space for. */
67 static struct cgraph_simd_clone *
68 simd_clone_struct_alloc (int nargs)
70 struct cgraph_simd_clone *clone_info;
71 size_t len = (sizeof (struct cgraph_simd_clone)
72 + nargs * sizeof (struct cgraph_simd_clone_arg));
73 clone_info = (struct cgraph_simd_clone *)
74 ggc_internal_cleared_alloc (len);
75 return clone_info;
78 /* Make a copy of the `struct cgraph_simd_clone' in FROM to TO. */
80 static inline void
81 simd_clone_struct_copy (struct cgraph_simd_clone *to,
82 struct cgraph_simd_clone *from)
84 memcpy (to, from, (sizeof (struct cgraph_simd_clone)
85 + ((from->nargs - from->inbranch)
86 * sizeof (struct cgraph_simd_clone_arg))));
89 /* Return vector of parameter types of function FNDECL. This uses
90 TYPE_ARG_TYPES if available, otherwise falls back to types of
91 DECL_ARGUMENTS types. */
93 static vec<tree>
94 simd_clone_vector_of_formal_parm_types (tree fndecl)
96 if (TYPE_ARG_TYPES (TREE_TYPE (fndecl)))
97 return ipa_get_vector_of_formal_parm_types (TREE_TYPE (fndecl));
98 vec<tree> args = ipa_get_vector_of_formal_parms (fndecl);
99 unsigned int i;
100 tree arg;
101 FOR_EACH_VEC_ELT (args, i, arg)
102 args[i] = TREE_TYPE (args[i]);
103 return args;
106 /* Given a simd function in NODE, extract the simd specific
107 information from the OMP clauses passed in CLAUSES, and return
108 the struct cgraph_simd_clone * if it should be cloned. *INBRANCH_SPECIFIED
109 is set to TRUE if the `inbranch' or `notinbranch' clause specified,
110 otherwise set to FALSE. */
112 static struct cgraph_simd_clone *
113 simd_clone_clauses_extract (struct cgraph_node *node, tree clauses,
114 bool *inbranch_specified)
116 vec<tree> args = simd_clone_vector_of_formal_parm_types (node->decl);
117 tree t;
118 int n;
119 *inbranch_specified = false;
121 n = args.length ();
122 if (n > 0 && args.last () == void_type_node)
123 n--;
125 /* Allocate one more than needed just in case this is an in-branch
126 clone which will require a mask argument. */
127 struct cgraph_simd_clone *clone_info = simd_clone_struct_alloc (n + 1);
128 clone_info->nargs = n;
130 if (!clauses)
131 goto out;
133 clauses = TREE_VALUE (clauses);
134 if (!clauses || TREE_CODE (clauses) != OMP_CLAUSE)
135 goto out;
137 for (t = clauses; t; t = OMP_CLAUSE_CHAIN (t))
139 switch (OMP_CLAUSE_CODE (t))
141 case OMP_CLAUSE_INBRANCH:
142 clone_info->inbranch = 1;
143 *inbranch_specified = true;
144 break;
145 case OMP_CLAUSE_NOTINBRANCH:
146 clone_info->inbranch = 0;
147 *inbranch_specified = true;
148 break;
149 case OMP_CLAUSE_SIMDLEN:
150 clone_info->simdlen
151 = TREE_INT_CST_LOW (OMP_CLAUSE_SIMDLEN_EXPR (t));
152 break;
153 case OMP_CLAUSE_LINEAR:
155 tree decl = OMP_CLAUSE_DECL (t);
156 tree step = OMP_CLAUSE_LINEAR_STEP (t);
157 int argno = TREE_INT_CST_LOW (decl);
158 if (OMP_CLAUSE_LINEAR_VARIABLE_STRIDE (t))
160 enum cgraph_simd_clone_arg_type arg_type;
161 if (TREE_CODE (args[argno]) == REFERENCE_TYPE)
162 switch (OMP_CLAUSE_LINEAR_KIND (t))
164 case OMP_CLAUSE_LINEAR_REF:
165 arg_type
166 = SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP;
167 break;
168 case OMP_CLAUSE_LINEAR_UVAL:
169 arg_type
170 = SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP;
171 break;
172 case OMP_CLAUSE_LINEAR_VAL:
173 case OMP_CLAUSE_LINEAR_DEFAULT:
174 arg_type
175 = SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP;
176 break;
177 default:
178 gcc_unreachable ();
180 else
181 arg_type = SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP;
182 clone_info->args[argno].arg_type = arg_type;
183 clone_info->args[argno].linear_step = tree_to_shwi (step);
184 gcc_assert (clone_info->args[argno].linear_step >= 0
185 && clone_info->args[argno].linear_step < n);
187 else
189 if (POINTER_TYPE_P (args[argno]))
190 step = fold_convert (ssizetype, step);
191 if (!tree_fits_shwi_p (step))
193 warning_at (OMP_CLAUSE_LOCATION (t), 0,
194 "ignoring large linear step");
195 args.release ();
196 return NULL;
198 else if (integer_zerop (step))
200 warning_at (OMP_CLAUSE_LOCATION (t), 0,
201 "ignoring zero linear step");
202 args.release ();
203 return NULL;
205 else
207 enum cgraph_simd_clone_arg_type arg_type;
208 if (TREE_CODE (args[argno]) == REFERENCE_TYPE)
209 switch (OMP_CLAUSE_LINEAR_KIND (t))
211 case OMP_CLAUSE_LINEAR_REF:
212 arg_type
213 = SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP;
214 break;
215 case OMP_CLAUSE_LINEAR_UVAL:
216 arg_type
217 = SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP;
218 break;
219 case OMP_CLAUSE_LINEAR_VAL:
220 case OMP_CLAUSE_LINEAR_DEFAULT:
221 arg_type
222 = SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP;
223 break;
224 default:
225 gcc_unreachable ();
227 else
228 arg_type = SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP;
229 clone_info->args[argno].arg_type = arg_type;
230 clone_info->args[argno].linear_step = tree_to_shwi (step);
233 break;
235 case OMP_CLAUSE_UNIFORM:
237 tree decl = OMP_CLAUSE_DECL (t);
238 int argno = tree_to_uhwi (decl);
239 clone_info->args[argno].arg_type
240 = SIMD_CLONE_ARG_TYPE_UNIFORM;
241 break;
243 case OMP_CLAUSE_ALIGNED:
245 tree decl = OMP_CLAUSE_DECL (t);
246 int argno = tree_to_uhwi (decl);
247 clone_info->args[argno].alignment
248 = TREE_INT_CST_LOW (OMP_CLAUSE_ALIGNED_ALIGNMENT (t));
249 break;
251 default:
252 break;
256 out:
257 if (TYPE_ATOMIC (TREE_TYPE (TREE_TYPE (node->decl))))
259 warning_at (DECL_SOURCE_LOCATION (node->decl), 0,
260 "ignoring %<#pragma omp declare simd%> on function "
261 "with %<_Atomic%> qualified return type");
262 args.release ();
263 return NULL;
266 for (unsigned int argno = 0; argno < clone_info->nargs; argno++)
267 if (TYPE_ATOMIC (args[argno])
268 && clone_info->args[argno].arg_type != SIMD_CLONE_ARG_TYPE_UNIFORM)
270 warning_at (DECL_SOURCE_LOCATION (node->decl), 0,
271 "ignoring %<#pragma omp declare simd%> on function "
272 "with %<_Atomic%> qualified non-%<uniform%> argument");
273 args.release ();
274 return NULL;
277 args.release ();
278 return clone_info;
281 /* Given a SIMD clone in NODE, calculate the characteristic data
282 type and return the coresponding type. The characteristic data
283 type is computed as described in the Intel Vector ABI. */
285 static tree
286 simd_clone_compute_base_data_type (struct cgraph_node *node,
287 struct cgraph_simd_clone *clone_info)
289 tree type = integer_type_node;
290 tree fndecl = node->decl;
292 /* a) For non-void function, the characteristic data type is the
293 return type. */
294 if (TREE_CODE (TREE_TYPE (TREE_TYPE (fndecl))) != VOID_TYPE)
295 type = TREE_TYPE (TREE_TYPE (fndecl));
297 /* b) If the function has any non-uniform, non-linear parameters,
298 then the characteristic data type is the type of the first
299 such parameter. */
300 else
302 vec<tree> map = simd_clone_vector_of_formal_parm_types (fndecl);
303 for (unsigned int i = 0; i < clone_info->nargs; ++i)
304 if (clone_info->args[i].arg_type == SIMD_CLONE_ARG_TYPE_VECTOR)
306 type = map[i];
307 break;
309 map.release ();
312 /* c) If the characteristic data type determined by a) or b) above
313 is struct, union, or class type which is pass-by-value (except
314 for the type that maps to the built-in complex data type), the
315 characteristic data type is int. */
316 if (RECORD_OR_UNION_TYPE_P (type)
317 && !aggregate_value_p (type, NULL)
318 && TREE_CODE (type) != COMPLEX_TYPE)
319 return integer_type_node;
321 /* d) If none of the above three classes is applicable, the
322 characteristic data type is int. */
324 return type;
326 /* e) For Intel Xeon Phi native and offload compilation, if the
327 resulting characteristic data type is 8-bit or 16-bit integer
328 data type, the characteristic data type is int. */
329 /* Well, we don't handle Xeon Phi yet. */
332 static tree
333 simd_clone_mangle (struct cgraph_node *node,
334 struct cgraph_simd_clone *clone_info)
336 char vecsize_mangle = clone_info->vecsize_mangle;
337 char mask = clone_info->inbranch ? 'M' : 'N';
338 unsigned int simdlen = clone_info->simdlen;
339 unsigned int n;
340 pretty_printer pp;
342 gcc_assert (vecsize_mangle && simdlen);
344 pp_string (&pp, "_ZGV");
345 pp_character (&pp, vecsize_mangle);
346 pp_character (&pp, mask);
347 pp_decimal_int (&pp, simdlen);
349 for (n = 0; n < clone_info->nargs; ++n)
351 struct cgraph_simd_clone_arg arg = clone_info->args[n];
353 switch (arg.arg_type)
355 case SIMD_CLONE_ARG_TYPE_UNIFORM:
356 pp_character (&pp, 'u');
357 break;
358 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
359 pp_character (&pp, 'l');
360 goto mangle_linear;
361 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP:
362 pp_character (&pp, 'R');
363 goto mangle_linear;
364 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP:
365 pp_character (&pp, 'L');
366 goto mangle_linear;
367 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP:
368 pp_character (&pp, 'U');
369 goto mangle_linear;
370 mangle_linear:
371 gcc_assert (arg.linear_step != 0);
372 if (arg.linear_step > 1)
373 pp_unsigned_wide_integer (&pp, arg.linear_step);
374 else if (arg.linear_step < 0)
376 pp_character (&pp, 'n');
377 pp_unsigned_wide_integer (&pp, (-(unsigned HOST_WIDE_INT)
378 arg.linear_step));
380 break;
381 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
382 pp_string (&pp, "ls");
383 pp_unsigned_wide_integer (&pp, arg.linear_step);
384 break;
385 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP:
386 pp_string (&pp, "Rs");
387 pp_unsigned_wide_integer (&pp, arg.linear_step);
388 break;
389 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
390 pp_string (&pp, "Ls");
391 pp_unsigned_wide_integer (&pp, arg.linear_step);
392 break;
393 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
394 pp_string (&pp, "Us");
395 pp_unsigned_wide_integer (&pp, arg.linear_step);
396 break;
397 default:
398 pp_character (&pp, 'v');
400 if (arg.alignment)
402 pp_character (&pp, 'a');
403 pp_decimal_int (&pp, arg.alignment);
407 pp_underscore (&pp);
408 const char *str = IDENTIFIER_POINTER (DECL_ASSEMBLER_NAME (node->decl));
409 if (*str == '*')
410 ++str;
411 pp_string (&pp, str);
412 str = pp_formatted_text (&pp);
414 /* If there already is a SIMD clone with the same mangled name, don't
415 add another one. This can happen e.g. for
416 #pragma omp declare simd
417 #pragma omp declare simd simdlen(8)
418 int foo (int, int);
419 if the simdlen is assumed to be 8 for the first one, etc. */
420 for (struct cgraph_node *clone = node->simd_clones; clone;
421 clone = clone->simdclone->next_clone)
422 if (id_equal (DECL_ASSEMBLER_NAME (clone->decl), str))
423 return NULL_TREE;
425 return get_identifier (str);
428 /* Create a simd clone of OLD_NODE and return it. */
430 static struct cgraph_node *
431 simd_clone_create (struct cgraph_node *old_node)
433 struct cgraph_node *new_node;
434 if (old_node->definition)
436 if (!old_node->has_gimple_body_p ())
437 return NULL;
438 old_node->get_body ();
439 new_node = old_node->create_version_clone_with_body (vNULL, NULL, NULL,
440 false, NULL, NULL,
441 "simdclone");
443 else
445 tree old_decl = old_node->decl;
446 tree new_decl = copy_node (old_node->decl);
447 DECL_NAME (new_decl) = clone_function_name (old_decl, "simdclone");
448 SET_DECL_ASSEMBLER_NAME (new_decl, DECL_NAME (new_decl));
449 SET_DECL_RTL (new_decl, NULL);
450 DECL_STATIC_CONSTRUCTOR (new_decl) = 0;
451 DECL_STATIC_DESTRUCTOR (new_decl) = 0;
452 new_node = old_node->create_version_clone (new_decl, vNULL, NULL);
453 if (old_node->in_other_partition)
454 new_node->in_other_partition = 1;
456 if (new_node == NULL)
457 return new_node;
459 TREE_PUBLIC (new_node->decl) = TREE_PUBLIC (old_node->decl);
460 DECL_COMDAT (new_node->decl) = DECL_COMDAT (old_node->decl);
461 DECL_WEAK (new_node->decl) = DECL_WEAK (old_node->decl);
462 DECL_EXTERNAL (new_node->decl) = DECL_EXTERNAL (old_node->decl);
463 DECL_VISIBILITY_SPECIFIED (new_node->decl)
464 = DECL_VISIBILITY_SPECIFIED (old_node->decl);
465 DECL_VISIBILITY (new_node->decl) = DECL_VISIBILITY (old_node->decl);
466 DECL_DLLIMPORT_P (new_node->decl) = DECL_DLLIMPORT_P (old_node->decl);
467 if (DECL_ONE_ONLY (old_node->decl))
468 make_decl_one_only (new_node->decl, DECL_ASSEMBLER_NAME (new_node->decl));
470 /* The method cgraph_version_clone_with_body () will force the new
471 symbol local. Undo this, and inherit external visibility from
472 the old node. */
473 new_node->local.local = old_node->local.local;
474 new_node->externally_visible = old_node->externally_visible;
476 return new_node;
479 /* Adjust the return type of the given function to its appropriate
480 vector counterpart. Returns a simd array to be used throughout the
481 function as a return value. */
483 static tree
484 simd_clone_adjust_return_type (struct cgraph_node *node)
486 tree fndecl = node->decl;
487 tree orig_rettype = TREE_TYPE (TREE_TYPE (fndecl));
488 unsigned int veclen;
489 tree t;
491 /* Adjust the function return type. */
492 if (orig_rettype == void_type_node)
493 return NULL_TREE;
494 TREE_TYPE (fndecl) = build_distinct_type_copy (TREE_TYPE (fndecl));
495 t = TREE_TYPE (TREE_TYPE (fndecl));
496 if (INTEGRAL_TYPE_P (t) || POINTER_TYPE_P (t))
497 veclen = node->simdclone->vecsize_int;
498 else
499 veclen = node->simdclone->vecsize_float;
500 veclen /= GET_MODE_BITSIZE (SCALAR_TYPE_MODE (t));
501 if (veclen > node->simdclone->simdlen)
502 veclen = node->simdclone->simdlen;
503 if (POINTER_TYPE_P (t))
504 t = pointer_sized_int_node;
505 if (veclen == node->simdclone->simdlen)
506 t = build_vector_type (t, node->simdclone->simdlen);
507 else
509 t = build_vector_type (t, veclen);
510 t = build_array_type_nelts (t, node->simdclone->simdlen / veclen);
512 TREE_TYPE (TREE_TYPE (fndecl)) = t;
513 if (!node->definition)
514 return NULL_TREE;
516 t = DECL_RESULT (fndecl);
517 /* Adjust the DECL_RESULT. */
518 gcc_assert (TREE_TYPE (t) != void_type_node);
519 TREE_TYPE (t) = TREE_TYPE (TREE_TYPE (fndecl));
520 relayout_decl (t);
522 tree atype = build_array_type_nelts (orig_rettype,
523 node->simdclone->simdlen);
524 if (veclen != node->simdclone->simdlen)
525 return build1 (VIEW_CONVERT_EXPR, atype, t);
527 /* Set up a SIMD array to use as the return value. */
528 tree retval = create_tmp_var_raw (atype, "retval");
529 gimple_add_tmp_var (retval);
530 return retval;
533 /* Each vector argument has a corresponding array to be used locally
534 as part of the eventual loop. Create such temporary array and
535 return it.
537 PREFIX is the prefix to be used for the temporary.
539 TYPE is the inner element type.
541 SIMDLEN is the number of elements. */
543 static tree
544 create_tmp_simd_array (const char *prefix, tree type, int simdlen)
546 tree atype = build_array_type_nelts (type, simdlen);
547 tree avar = create_tmp_var_raw (atype, prefix);
548 gimple_add_tmp_var (avar);
549 return avar;
552 /* Modify the function argument types to their corresponding vector
553 counterparts if appropriate. Also, create one array for each simd
554 argument to be used locally when using the function arguments as
555 part of the loop.
557 NODE is the function whose arguments are to be adjusted.
559 Returns an adjustment vector that will be filled describing how the
560 argument types will be adjusted. */
562 static ipa_parm_adjustment_vec
563 simd_clone_adjust_argument_types (struct cgraph_node *node)
565 vec<tree> args;
566 ipa_parm_adjustment_vec adjustments;
568 if (node->definition)
569 args = ipa_get_vector_of_formal_parms (node->decl);
570 else
571 args = simd_clone_vector_of_formal_parm_types (node->decl);
572 adjustments.create (args.length ());
573 unsigned i, j, veclen;
574 struct ipa_parm_adjustment adj;
575 struct cgraph_simd_clone *sc = node->simdclone;
577 for (i = 0; i < sc->nargs; ++i)
579 memset (&adj, 0, sizeof (adj));
580 tree parm = args[i];
581 tree parm_type = node->definition ? TREE_TYPE (parm) : parm;
582 adj.base_index = i;
583 adj.base = parm;
585 sc->args[i].orig_arg = node->definition ? parm : NULL_TREE;
586 sc->args[i].orig_type = parm_type;
588 switch (sc->args[i].arg_type)
590 default:
591 /* No adjustment necessary for scalar arguments. */
592 adj.op = IPA_PARM_OP_COPY;
593 break;
594 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP:
595 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
596 if (node->definition)
597 sc->args[i].simd_array
598 = create_tmp_simd_array (IDENTIFIER_POINTER (DECL_NAME (parm)),
599 TREE_TYPE (parm_type),
600 sc->simdlen);
601 adj.op = IPA_PARM_OP_COPY;
602 break;
603 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP:
604 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
605 case SIMD_CLONE_ARG_TYPE_VECTOR:
606 if (INTEGRAL_TYPE_P (parm_type) || POINTER_TYPE_P (parm_type))
607 veclen = sc->vecsize_int;
608 else
609 veclen = sc->vecsize_float;
610 veclen /= GET_MODE_BITSIZE (SCALAR_TYPE_MODE (parm_type));
611 if (veclen > sc->simdlen)
612 veclen = sc->simdlen;
613 adj.arg_prefix = "simd";
614 if (POINTER_TYPE_P (parm_type))
615 adj.type = build_vector_type (pointer_sized_int_node, veclen);
616 else
617 adj.type = build_vector_type (parm_type, veclen);
618 sc->args[i].vector_type = adj.type;
619 for (j = veclen; j < sc->simdlen; j += veclen)
621 adjustments.safe_push (adj);
622 if (j == veclen)
624 memset (&adj, 0, sizeof (adj));
625 adj.op = IPA_PARM_OP_NEW;
626 adj.arg_prefix = "simd";
627 adj.base_index = i;
628 adj.type = sc->args[i].vector_type;
632 if (node->definition)
633 sc->args[i].simd_array
634 = create_tmp_simd_array (DECL_NAME (parm)
635 ? IDENTIFIER_POINTER (DECL_NAME (parm))
636 : NULL, parm_type, sc->simdlen);
638 adjustments.safe_push (adj);
641 if (sc->inbranch)
643 tree base_type = simd_clone_compute_base_data_type (sc->origin, sc);
645 memset (&adj, 0, sizeof (adj));
646 adj.op = IPA_PARM_OP_NEW;
647 adj.arg_prefix = "mask";
649 adj.base_index = i;
650 if (INTEGRAL_TYPE_P (base_type) || POINTER_TYPE_P (base_type))
651 veclen = sc->vecsize_int;
652 else
653 veclen = sc->vecsize_float;
654 veclen /= GET_MODE_BITSIZE (SCALAR_TYPE_MODE (base_type));
655 if (veclen > sc->simdlen)
656 veclen = sc->simdlen;
657 if (sc->mask_mode != VOIDmode)
658 adj.type
659 = lang_hooks.types.type_for_mode (sc->mask_mode, 1);
660 else if (POINTER_TYPE_P (base_type))
661 adj.type = build_vector_type (pointer_sized_int_node, veclen);
662 else
663 adj.type = build_vector_type (base_type, veclen);
664 adjustments.safe_push (adj);
666 for (j = veclen; j < sc->simdlen; j += veclen)
667 adjustments.safe_push (adj);
669 /* We have previously allocated one extra entry for the mask. Use
670 it and fill it. */
671 sc->nargs++;
672 if (sc->mask_mode != VOIDmode)
673 base_type = boolean_type_node;
674 if (node->definition)
676 sc->args[i].orig_arg
677 = build_decl (UNKNOWN_LOCATION, PARM_DECL, NULL, base_type);
678 if (sc->mask_mode == VOIDmode)
679 sc->args[i].simd_array
680 = create_tmp_simd_array ("mask", base_type, sc->simdlen);
681 else if (veclen < sc->simdlen)
682 sc->args[i].simd_array
683 = create_tmp_simd_array ("mask", adj.type, sc->simdlen / veclen);
684 else
685 sc->args[i].simd_array = NULL_TREE;
687 sc->args[i].orig_type = base_type;
688 sc->args[i].arg_type = SIMD_CLONE_ARG_TYPE_MASK;
691 if (node->definition)
692 ipa_modify_formal_parameters (node->decl, adjustments);
693 else
695 tree new_arg_types = NULL_TREE, new_reversed;
696 bool last_parm_void = false;
697 if (args.length () > 0 && args.last () == void_type_node)
698 last_parm_void = true;
700 gcc_assert (TYPE_ARG_TYPES (TREE_TYPE (node->decl)));
701 j = adjustments.length ();
702 for (i = 0; i < j; i++)
704 struct ipa_parm_adjustment *adj = &adjustments[i];
705 tree ptype;
706 if (adj->op == IPA_PARM_OP_COPY)
707 ptype = args[adj->base_index];
708 else
709 ptype = adj->type;
710 new_arg_types = tree_cons (NULL_TREE, ptype, new_arg_types);
712 new_reversed = nreverse (new_arg_types);
713 if (last_parm_void)
715 if (new_reversed)
716 TREE_CHAIN (new_arg_types) = void_list_node;
717 else
718 new_reversed = void_list_node;
721 tree new_type = build_distinct_type_copy (TREE_TYPE (node->decl));
722 TYPE_ARG_TYPES (new_type) = new_reversed;
723 TREE_TYPE (node->decl) = new_type;
725 adjustments.release ();
727 args.release ();
728 return adjustments;
731 /* Initialize and copy the function arguments in NODE to their
732 corresponding local simd arrays. Returns a fresh gimple_seq with
733 the instruction sequence generated. */
735 static gimple_seq
736 simd_clone_init_simd_arrays (struct cgraph_node *node,
737 ipa_parm_adjustment_vec adjustments)
739 gimple_seq seq = NULL;
740 unsigned i = 0, j = 0, k;
742 for (tree arg = DECL_ARGUMENTS (node->decl);
743 arg;
744 arg = DECL_CHAIN (arg), i++, j++)
746 if (adjustments[j].op == IPA_PARM_OP_COPY
747 || POINTER_TYPE_P (TREE_TYPE (arg)))
748 continue;
750 node->simdclone->args[i].vector_arg = arg;
752 tree array = node->simdclone->args[i].simd_array;
753 if (node->simdclone->mask_mode != VOIDmode
754 && node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_MASK)
756 if (array == NULL_TREE)
757 continue;
758 unsigned int l
759 = tree_to_uhwi (TYPE_MAX_VALUE (TYPE_DOMAIN (TREE_TYPE (array))));
760 for (k = 0; k <= l; k++)
762 if (k)
764 arg = DECL_CHAIN (arg);
765 j++;
767 tree t = build4 (ARRAY_REF, TREE_TYPE (TREE_TYPE (array)),
768 array, size_int (k), NULL, NULL);
769 t = build2 (MODIFY_EXPR, TREE_TYPE (t), t, arg);
770 gimplify_and_add (t, &seq);
772 continue;
774 if (simd_clone_subparts (TREE_TYPE (arg)) == node->simdclone->simdlen)
776 tree ptype = build_pointer_type (TREE_TYPE (TREE_TYPE (array)));
777 tree ptr = build_fold_addr_expr (array);
778 tree t = build2 (MEM_REF, TREE_TYPE (arg), ptr,
779 build_int_cst (ptype, 0));
780 t = build2 (MODIFY_EXPR, TREE_TYPE (t), t, arg);
781 gimplify_and_add (t, &seq);
783 else
785 unsigned int simdlen = simd_clone_subparts (TREE_TYPE (arg));
786 tree ptype = build_pointer_type (TREE_TYPE (TREE_TYPE (array)));
787 for (k = 0; k < node->simdclone->simdlen; k += simdlen)
789 tree ptr = build_fold_addr_expr (array);
790 int elemsize;
791 if (k)
793 arg = DECL_CHAIN (arg);
794 j++;
796 tree elemtype = TREE_TYPE (TREE_TYPE (arg));
797 elemsize = GET_MODE_SIZE (SCALAR_TYPE_MODE (elemtype));
798 tree t = build2 (MEM_REF, TREE_TYPE (arg), ptr,
799 build_int_cst (ptype, k * elemsize));
800 t = build2 (MODIFY_EXPR, TREE_TYPE (t), t, arg);
801 gimplify_and_add (t, &seq);
805 return seq;
808 /* Callback info for ipa_simd_modify_stmt_ops below. */
810 struct modify_stmt_info {
811 ipa_parm_adjustment_vec adjustments;
812 gimple *stmt;
813 /* True if the parent statement was modified by
814 ipa_simd_modify_stmt_ops. */
815 bool modified;
818 /* Callback for walk_gimple_op.
820 Adjust operands from a given statement as specified in the
821 adjustments vector in the callback data. */
823 static tree
824 ipa_simd_modify_stmt_ops (tree *tp, int *walk_subtrees, void *data)
826 struct walk_stmt_info *wi = (struct walk_stmt_info *) data;
827 struct modify_stmt_info *info = (struct modify_stmt_info *) wi->info;
828 tree *orig_tp = tp;
829 if (TREE_CODE (*tp) == ADDR_EXPR)
830 tp = &TREE_OPERAND (*tp, 0);
831 struct ipa_parm_adjustment *cand = NULL;
832 if (TREE_CODE (*tp) == PARM_DECL)
833 cand = ipa_get_adjustment_candidate (&tp, NULL, info->adjustments, true);
834 else
836 if (TYPE_P (*tp))
837 *walk_subtrees = 0;
840 tree repl = NULL_TREE;
841 if (cand)
842 repl = unshare_expr (cand->new_decl);
843 else
845 if (tp != orig_tp)
847 *walk_subtrees = 0;
848 bool modified = info->modified;
849 info->modified = false;
850 walk_tree (tp, ipa_simd_modify_stmt_ops, wi, wi->pset);
851 if (!info->modified)
853 info->modified = modified;
854 return NULL_TREE;
856 info->modified = modified;
857 repl = *tp;
859 else
860 return NULL_TREE;
863 if (tp != orig_tp)
865 repl = build_fold_addr_expr (repl);
866 gimple *stmt;
867 if (is_gimple_debug (info->stmt))
869 tree vexpr = make_node (DEBUG_EXPR_DECL);
870 stmt = gimple_build_debug_source_bind (vexpr, repl, NULL);
871 DECL_ARTIFICIAL (vexpr) = 1;
872 TREE_TYPE (vexpr) = TREE_TYPE (repl);
873 SET_DECL_MODE (vexpr, TYPE_MODE (TREE_TYPE (repl)));
874 repl = vexpr;
876 else
878 stmt = gimple_build_assign (make_ssa_name (TREE_TYPE (repl)), repl);
879 repl = gimple_assign_lhs (stmt);
881 gimple_stmt_iterator gsi = gsi_for_stmt (info->stmt);
882 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
883 *orig_tp = repl;
885 else if (!useless_type_conversion_p (TREE_TYPE (*tp), TREE_TYPE (repl)))
887 tree vce = build1 (VIEW_CONVERT_EXPR, TREE_TYPE (*tp), repl);
888 *tp = vce;
890 else
891 *tp = repl;
893 info->modified = true;
894 return NULL_TREE;
897 /* Traverse the function body and perform all modifications as
898 described in ADJUSTMENTS. At function return, ADJUSTMENTS will be
899 modified such that the replacement/reduction value will now be an
900 offset into the corresponding simd_array.
902 This function will replace all function argument uses with their
903 corresponding simd array elements, and ajust the return values
904 accordingly. */
906 static void
907 ipa_simd_modify_function_body (struct cgraph_node *node,
908 ipa_parm_adjustment_vec adjustments,
909 tree retval_array, tree iter)
911 basic_block bb;
912 unsigned int i, j, l;
914 /* Re-use the adjustments array, but this time use it to replace
915 every function argument use to an offset into the corresponding
916 simd_array. */
917 for (i = 0, j = 0; i < node->simdclone->nargs; ++i, ++j)
919 if (!node->simdclone->args[i].vector_arg)
920 continue;
922 tree basetype = TREE_TYPE (node->simdclone->args[i].orig_arg);
923 tree vectype = TREE_TYPE (node->simdclone->args[i].vector_arg);
924 adjustments[j].new_decl
925 = build4 (ARRAY_REF,
926 basetype,
927 node->simdclone->args[i].simd_array,
928 iter,
929 NULL_TREE, NULL_TREE);
930 if (adjustments[j].op == IPA_PARM_OP_NONE
931 && simd_clone_subparts (vectype) < node->simdclone->simdlen)
932 j += node->simdclone->simdlen / simd_clone_subparts (vectype) - 1;
935 l = adjustments.length ();
936 tree name;
938 FOR_EACH_SSA_NAME (i, name, cfun)
940 if (SSA_NAME_VAR (name)
941 && TREE_CODE (SSA_NAME_VAR (name)) == PARM_DECL)
943 for (j = 0; j < l; j++)
944 if (SSA_NAME_VAR (name) == adjustments[j].base
945 && adjustments[j].new_decl)
947 tree base_var;
948 if (adjustments[j].new_ssa_base == NULL_TREE)
950 base_var
951 = copy_var_decl (adjustments[j].base,
952 DECL_NAME (adjustments[j].base),
953 TREE_TYPE (adjustments[j].base));
954 adjustments[j].new_ssa_base = base_var;
956 else
957 base_var = adjustments[j].new_ssa_base;
958 if (SSA_NAME_IS_DEFAULT_DEF (name))
960 bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
961 gimple_stmt_iterator gsi = gsi_after_labels (bb);
962 tree new_decl = unshare_expr (adjustments[j].new_decl);
963 set_ssa_default_def (cfun, adjustments[j].base, NULL_TREE);
964 SET_SSA_NAME_VAR_OR_IDENTIFIER (name, base_var);
965 SSA_NAME_IS_DEFAULT_DEF (name) = 0;
966 gimple *stmt = gimple_build_assign (name, new_decl);
967 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
969 else
970 SET_SSA_NAME_VAR_OR_IDENTIFIER (name, base_var);
975 struct modify_stmt_info info;
976 info.adjustments = adjustments;
978 FOR_EACH_BB_FN (bb, DECL_STRUCT_FUNCTION (node->decl))
980 gimple_stmt_iterator gsi;
982 gsi = gsi_start_bb (bb);
983 while (!gsi_end_p (gsi))
985 gimple *stmt = gsi_stmt (gsi);
986 info.stmt = stmt;
987 struct walk_stmt_info wi;
989 memset (&wi, 0, sizeof (wi));
990 info.modified = false;
991 wi.info = &info;
992 walk_gimple_op (stmt, ipa_simd_modify_stmt_ops, &wi);
994 if (greturn *return_stmt = dyn_cast <greturn *> (stmt))
996 tree retval = gimple_return_retval (return_stmt);
997 if (!retval)
999 gsi_remove (&gsi, true);
1000 continue;
1003 /* Replace `return foo' with `retval_array[iter] = foo'. */
1004 tree ref = build4 (ARRAY_REF, TREE_TYPE (retval),
1005 retval_array, iter, NULL, NULL);
1006 stmt = gimple_build_assign (ref, retval);
1007 gsi_replace (&gsi, stmt, true);
1008 info.modified = true;
1011 if (info.modified)
1013 update_stmt (stmt);
1014 if (maybe_clean_eh_stmt (stmt))
1015 gimple_purge_dead_eh_edges (gimple_bb (stmt));
1017 gsi_next (&gsi);
1022 /* Helper function of simd_clone_adjust, return linear step addend
1023 of Ith argument. */
1025 static tree
1026 simd_clone_linear_addend (struct cgraph_node *node, unsigned int i,
1027 tree addtype, basic_block entry_bb)
1029 tree ptype = NULL_TREE;
1030 switch (node->simdclone->args[i].arg_type)
1032 case SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP:
1033 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP:
1034 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_CONSTANT_STEP:
1035 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP:
1036 return build_int_cst (addtype, node->simdclone->args[i].linear_step);
1037 case SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP:
1038 case SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP:
1039 ptype = TREE_TYPE (node->simdclone->args[i].orig_arg);
1040 break;
1041 case SIMD_CLONE_ARG_TYPE_LINEAR_VAL_VARIABLE_STEP:
1042 case SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP:
1043 ptype = TREE_TYPE (TREE_TYPE (node->simdclone->args[i].orig_arg));
1044 break;
1045 default:
1046 gcc_unreachable ();
1049 unsigned int idx = node->simdclone->args[i].linear_step;
1050 tree arg = node->simdclone->args[idx].orig_arg;
1051 gcc_assert (is_gimple_reg_type (TREE_TYPE (arg)));
1052 gimple_stmt_iterator gsi = gsi_after_labels (entry_bb);
1053 gimple *g;
1054 tree ret;
1055 if (is_gimple_reg (arg))
1056 ret = get_or_create_ssa_default_def (cfun, arg);
1057 else
1059 g = gimple_build_assign (make_ssa_name (TREE_TYPE (arg)), arg);
1060 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1061 ret = gimple_assign_lhs (g);
1063 if (TREE_CODE (TREE_TYPE (arg)) == REFERENCE_TYPE)
1065 g = gimple_build_assign (make_ssa_name (TREE_TYPE (TREE_TYPE (arg))),
1066 build_simple_mem_ref (ret));
1067 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1068 ret = gimple_assign_lhs (g);
1070 if (!useless_type_conversion_p (addtype, TREE_TYPE (ret)))
1072 g = gimple_build_assign (make_ssa_name (addtype), NOP_EXPR, ret);
1073 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1074 ret = gimple_assign_lhs (g);
1076 if (POINTER_TYPE_P (ptype))
1078 tree size = TYPE_SIZE_UNIT (TREE_TYPE (ptype));
1079 if (size && TREE_CODE (size) == INTEGER_CST)
1081 g = gimple_build_assign (make_ssa_name (addtype), MULT_EXPR,
1082 ret, fold_convert (addtype, size));
1083 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1084 ret = gimple_assign_lhs (g);
1087 return ret;
1090 /* Adjust the argument types in NODE to their appropriate vector
1091 counterparts. */
1093 static void
1094 simd_clone_adjust (struct cgraph_node *node)
1096 push_cfun (DECL_STRUCT_FUNCTION (node->decl));
1098 targetm.simd_clone.adjust (node);
1100 tree retval = simd_clone_adjust_return_type (node);
1101 ipa_parm_adjustment_vec adjustments
1102 = simd_clone_adjust_argument_types (node);
1104 push_gimplify_context ();
1106 gimple_seq seq = simd_clone_init_simd_arrays (node, adjustments);
1108 /* Adjust all uses of vector arguments accordingly. Adjust all
1109 return values accordingly. */
1110 tree iter = create_tmp_var (unsigned_type_node, "iter");
1111 tree iter1 = make_ssa_name (iter);
1112 tree iter2 = NULL_TREE;
1113 ipa_simd_modify_function_body (node, adjustments, retval, iter1);
1114 adjustments.release ();
1116 /* Initialize the iteration variable. */
1117 basic_block entry_bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1118 basic_block body_bb = split_block_after_labels (entry_bb)->dest;
1119 gimple_stmt_iterator gsi = gsi_after_labels (entry_bb);
1120 /* Insert the SIMD array and iv initialization at function
1121 entry. */
1122 gsi_insert_seq_before (&gsi, seq, GSI_NEW_STMT);
1124 pop_gimplify_context (NULL);
1126 gimple *g;
1127 basic_block incr_bb = NULL;
1128 struct loop *loop = NULL;
1130 /* Create a new BB right before the original exit BB, to hold the
1131 iteration increment and the condition/branch. */
1132 if (EDGE_COUNT (EXIT_BLOCK_PTR_FOR_FN (cfun)->preds))
1134 basic_block orig_exit = EDGE_PRED (EXIT_BLOCK_PTR_FOR_FN (cfun), 0)->src;
1135 incr_bb = create_empty_bb (orig_exit);
1136 incr_bb->count = profile_count::zero ();
1137 add_bb_to_loop (incr_bb, body_bb->loop_father);
1138 /* The succ of orig_exit was EXIT_BLOCK_PTR_FOR_FN (cfun), with an empty
1139 flag. Set it now to be a FALLTHRU_EDGE. */
1140 gcc_assert (EDGE_COUNT (orig_exit->succs) == 1);
1141 EDGE_SUCC (orig_exit, 0)->flags |= EDGE_FALLTHRU;
1142 for (unsigned i = 0;
1143 i < EDGE_COUNT (EXIT_BLOCK_PTR_FOR_FN (cfun)->preds); ++i)
1145 edge e = EDGE_PRED (EXIT_BLOCK_PTR_FOR_FN (cfun), i);
1146 redirect_edge_succ (e, incr_bb);
1147 incr_bb->count += e->count ();
1150 else if (node->simdclone->inbranch)
1152 incr_bb = create_empty_bb (entry_bb);
1153 incr_bb->count = profile_count::zero ();
1154 add_bb_to_loop (incr_bb, body_bb->loop_father);
1157 if (incr_bb)
1159 make_single_succ_edge (incr_bb, EXIT_BLOCK_PTR_FOR_FN (cfun), 0);
1160 gsi = gsi_last_bb (incr_bb);
1161 iter2 = make_ssa_name (iter);
1162 g = gimple_build_assign (iter2, PLUS_EXPR, iter1,
1163 build_int_cst (unsigned_type_node, 1));
1164 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1166 /* Mostly annotate the loop for the vectorizer (the rest is done
1167 below). */
1168 loop = alloc_loop ();
1169 cfun->has_force_vectorize_loops = true;
1170 loop->safelen = node->simdclone->simdlen;
1171 loop->force_vectorize = true;
1172 loop->header = body_bb;
1175 /* Branch around the body if the mask applies. */
1176 if (node->simdclone->inbranch)
1178 gsi = gsi_last_bb (loop->header);
1179 tree mask_array
1180 = node->simdclone->args[node->simdclone->nargs - 1].simd_array;
1181 tree mask;
1182 if (node->simdclone->mask_mode != VOIDmode)
1184 tree shift_cnt;
1185 if (mask_array == NULL_TREE)
1187 tree arg = node->simdclone->args[node->simdclone->nargs
1188 - 1].vector_arg;
1189 mask = get_or_create_ssa_default_def (cfun, arg);
1190 shift_cnt = iter1;
1192 else
1194 tree maskt = TREE_TYPE (mask_array);
1195 int c = tree_to_uhwi (TYPE_MAX_VALUE (TYPE_DOMAIN (maskt)));
1196 c = node->simdclone->simdlen / (c + 1);
1197 int s = exact_log2 (c);
1198 gcc_assert (s > 0);
1199 c--;
1200 tree idx = make_ssa_name (TREE_TYPE (iter1));
1201 g = gimple_build_assign (idx, RSHIFT_EXPR, iter1,
1202 build_int_cst (NULL_TREE, s));
1203 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1204 mask = make_ssa_name (TREE_TYPE (TREE_TYPE (mask_array)));
1205 tree aref = build4 (ARRAY_REF,
1206 TREE_TYPE (TREE_TYPE (mask_array)),
1207 mask_array, idx, NULL, NULL);
1208 g = gimple_build_assign (mask, aref);
1209 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1210 shift_cnt = make_ssa_name (TREE_TYPE (iter1));
1211 g = gimple_build_assign (shift_cnt, BIT_AND_EXPR, iter1,
1212 build_int_cst (TREE_TYPE (iter1), c));
1213 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1215 g = gimple_build_assign (make_ssa_name (TREE_TYPE (mask)),
1216 RSHIFT_EXPR, mask, shift_cnt);
1217 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1218 mask = gimple_assign_lhs (g);
1219 g = gimple_build_assign (make_ssa_name (TREE_TYPE (mask)),
1220 BIT_AND_EXPR, mask,
1221 build_int_cst (TREE_TYPE (mask), 1));
1222 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1223 mask = gimple_assign_lhs (g);
1225 else
1227 mask = make_ssa_name (TREE_TYPE (TREE_TYPE (mask_array)));
1228 tree aref = build4 (ARRAY_REF,
1229 TREE_TYPE (TREE_TYPE (mask_array)),
1230 mask_array, iter1, NULL, NULL);
1231 g = gimple_build_assign (mask, aref);
1232 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1233 int bitsize = GET_MODE_BITSIZE (SCALAR_TYPE_MODE (TREE_TYPE (aref)));
1234 if (!INTEGRAL_TYPE_P (TREE_TYPE (aref)))
1236 aref = build1 (VIEW_CONVERT_EXPR,
1237 build_nonstandard_integer_type (bitsize, 0),
1238 mask);
1239 mask = make_ssa_name (TREE_TYPE (aref));
1240 g = gimple_build_assign (mask, aref);
1241 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1245 g = gimple_build_cond (EQ_EXPR, mask, build_zero_cst (TREE_TYPE (mask)),
1246 NULL, NULL);
1247 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1248 edge e = make_edge (loop->header, incr_bb, EDGE_TRUE_VALUE);
1249 e->probability = profile_probability::unlikely ().guessed ();
1250 incr_bb->count += e->count ();
1251 edge fallthru = FALLTHRU_EDGE (loop->header);
1252 fallthru->flags = EDGE_FALSE_VALUE;
1253 fallthru->probability = profile_probability::likely ().guessed ();
1256 basic_block latch_bb = NULL;
1257 basic_block new_exit_bb = NULL;
1259 /* Generate the condition. */
1260 if (incr_bb)
1262 gsi = gsi_last_bb (incr_bb);
1263 g = gimple_build_cond (LT_EXPR, iter2,
1264 build_int_cst (unsigned_type_node,
1265 node->simdclone->simdlen),
1266 NULL, NULL);
1267 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1268 edge e = split_block (incr_bb, gsi_stmt (gsi));
1269 latch_bb = e->dest;
1270 new_exit_bb = split_block_after_labels (latch_bb)->dest;
1271 loop->latch = latch_bb;
1273 redirect_edge_succ (FALLTHRU_EDGE (latch_bb), body_bb);
1275 edge new_e = make_edge (incr_bb, new_exit_bb, EDGE_FALSE_VALUE);
1277 /* FIXME: Do we need to distribute probabilities for the conditional? */
1278 new_e->probability = profile_probability::guessed_never ();
1279 /* The successor of incr_bb is already pointing to latch_bb; just
1280 change the flags.
1281 make_edge (incr_bb, latch_bb, EDGE_TRUE_VALUE); */
1282 FALLTHRU_EDGE (incr_bb)->flags = EDGE_TRUE_VALUE;
1285 gphi *phi = create_phi_node (iter1, body_bb);
1286 edge preheader_edge = find_edge (entry_bb, body_bb);
1287 edge latch_edge = NULL;
1288 add_phi_arg (phi, build_zero_cst (unsigned_type_node), preheader_edge,
1289 UNKNOWN_LOCATION);
1290 if (incr_bb)
1292 latch_edge = single_succ_edge (latch_bb);
1293 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1295 /* Generate the new return. */
1296 gsi = gsi_last_bb (new_exit_bb);
1297 if (retval
1298 && TREE_CODE (retval) == VIEW_CONVERT_EXPR
1299 && TREE_CODE (TREE_OPERAND (retval, 0)) == RESULT_DECL)
1300 retval = TREE_OPERAND (retval, 0);
1301 else if (retval)
1303 retval = build1 (VIEW_CONVERT_EXPR,
1304 TREE_TYPE (TREE_TYPE (node->decl)),
1305 retval);
1306 retval = force_gimple_operand_gsi (&gsi, retval, true, NULL,
1307 false, GSI_CONTINUE_LINKING);
1309 g = gimple_build_return (retval);
1310 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1313 /* Handle aligned clauses by replacing default defs of the aligned
1314 uniform args with __builtin_assume_aligned (arg_N(D), alignment)
1315 lhs. Handle linear by adding PHIs. */
1316 for (unsigned i = 0; i < node->simdclone->nargs; i++)
1317 if (node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_UNIFORM
1318 && (TREE_ADDRESSABLE (node->simdclone->args[i].orig_arg)
1319 || !is_gimple_reg_type
1320 (TREE_TYPE (node->simdclone->args[i].orig_arg))))
1322 tree orig_arg = node->simdclone->args[i].orig_arg;
1323 if (is_gimple_reg_type (TREE_TYPE (orig_arg)))
1324 iter1 = make_ssa_name (TREE_TYPE (orig_arg));
1325 else
1327 iter1 = create_tmp_var_raw (TREE_TYPE (orig_arg));
1328 gimple_add_tmp_var (iter1);
1330 gsi = gsi_after_labels (entry_bb);
1331 g = gimple_build_assign (iter1, orig_arg);
1332 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1333 gsi = gsi_after_labels (body_bb);
1334 g = gimple_build_assign (orig_arg, iter1);
1335 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1337 else if (node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_UNIFORM
1338 && DECL_BY_REFERENCE (node->simdclone->args[i].orig_arg)
1339 && TREE_CODE (TREE_TYPE (node->simdclone->args[i].orig_arg))
1340 == REFERENCE_TYPE
1341 && TREE_ADDRESSABLE
1342 (TREE_TYPE (TREE_TYPE (node->simdclone->args[i].orig_arg))))
1344 tree orig_arg = node->simdclone->args[i].orig_arg;
1345 tree def = ssa_default_def (cfun, orig_arg);
1346 if (def && !has_zero_uses (def))
1348 iter1 = create_tmp_var_raw (TREE_TYPE (TREE_TYPE (orig_arg)));
1349 gimple_add_tmp_var (iter1);
1350 gsi = gsi_after_labels (entry_bb);
1351 g = gimple_build_assign (iter1, build_simple_mem_ref (def));
1352 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1353 gsi = gsi_after_labels (body_bb);
1354 g = gimple_build_assign (build_simple_mem_ref (def), iter1);
1355 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1358 else if (node->simdclone->args[i].alignment
1359 && node->simdclone->args[i].arg_type
1360 == SIMD_CLONE_ARG_TYPE_UNIFORM
1361 && (node->simdclone->args[i].alignment
1362 & (node->simdclone->args[i].alignment - 1)) == 0
1363 && TREE_CODE (TREE_TYPE (node->simdclone->args[i].orig_arg))
1364 == POINTER_TYPE)
1366 unsigned int alignment = node->simdclone->args[i].alignment;
1367 tree orig_arg = node->simdclone->args[i].orig_arg;
1368 tree def = ssa_default_def (cfun, orig_arg);
1369 if (def && !has_zero_uses (def))
1371 tree fn = builtin_decl_explicit (BUILT_IN_ASSUME_ALIGNED);
1372 gimple_seq seq = NULL;
1373 bool need_cvt = false;
1374 gcall *call
1375 = gimple_build_call (fn, 2, def, size_int (alignment));
1376 g = call;
1377 if (!useless_type_conversion_p (TREE_TYPE (orig_arg),
1378 ptr_type_node))
1379 need_cvt = true;
1380 tree t = make_ssa_name (need_cvt ? ptr_type_node : orig_arg);
1381 gimple_call_set_lhs (g, t);
1382 gimple_seq_add_stmt_without_update (&seq, g);
1383 if (need_cvt)
1385 t = make_ssa_name (orig_arg);
1386 g = gimple_build_assign (t, NOP_EXPR, gimple_call_lhs (g));
1387 gimple_seq_add_stmt_without_update (&seq, g);
1389 gsi_insert_seq_on_edge_immediate
1390 (single_succ_edge (ENTRY_BLOCK_PTR_FOR_FN (cfun)), seq);
1392 entry_bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1393 node->create_edge (cgraph_node::get_create (fn),
1394 call, entry_bb->count);
1396 imm_use_iterator iter;
1397 use_operand_p use_p;
1398 gimple *use_stmt;
1399 tree repl = gimple_get_lhs (g);
1400 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1401 if (is_gimple_debug (use_stmt) || use_stmt == call)
1402 continue;
1403 else
1404 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1405 SET_USE (use_p, repl);
1408 else if ((node->simdclone->args[i].arg_type
1409 == SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP)
1410 || (node->simdclone->args[i].arg_type
1411 == SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP)
1412 || (node->simdclone->args[i].arg_type
1413 == SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP)
1414 || (node->simdclone->args[i].arg_type
1415 == SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP))
1417 tree orig_arg = node->simdclone->args[i].orig_arg;
1418 gcc_assert (INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1419 || POINTER_TYPE_P (TREE_TYPE (orig_arg)));
1420 tree def = NULL_TREE;
1421 if (TREE_ADDRESSABLE (orig_arg))
1423 def = make_ssa_name (TREE_TYPE (orig_arg));
1424 iter1 = make_ssa_name (TREE_TYPE (orig_arg));
1425 if (incr_bb)
1426 iter2 = make_ssa_name (TREE_TYPE (orig_arg));
1427 gsi = gsi_after_labels (entry_bb);
1428 g = gimple_build_assign (def, orig_arg);
1429 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1431 else
1433 def = ssa_default_def (cfun, orig_arg);
1434 if (!def || has_zero_uses (def))
1435 def = NULL_TREE;
1436 else
1438 iter1 = make_ssa_name (orig_arg);
1439 if (incr_bb)
1440 iter2 = make_ssa_name (orig_arg);
1443 if (def)
1445 phi = create_phi_node (iter1, body_bb);
1446 add_phi_arg (phi, def, preheader_edge, UNKNOWN_LOCATION);
1447 if (incr_bb)
1449 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1450 enum tree_code code = INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1451 ? PLUS_EXPR : POINTER_PLUS_EXPR;
1452 tree addtype = INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1453 ? TREE_TYPE (orig_arg) : sizetype;
1454 tree addcst = simd_clone_linear_addend (node, i, addtype,
1455 entry_bb);
1456 gsi = gsi_last_bb (incr_bb);
1457 g = gimple_build_assign (iter2, code, iter1, addcst);
1458 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1461 imm_use_iterator iter;
1462 use_operand_p use_p;
1463 gimple *use_stmt;
1464 if (TREE_ADDRESSABLE (orig_arg))
1466 gsi = gsi_after_labels (body_bb);
1467 g = gimple_build_assign (orig_arg, iter1);
1468 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1470 else
1471 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1472 if (use_stmt == phi)
1473 continue;
1474 else
1475 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1476 SET_USE (use_p, iter1);
1479 else if (node->simdclone->args[i].arg_type
1480 == SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP
1481 || (node->simdclone->args[i].arg_type
1482 == SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP))
1484 tree orig_arg = node->simdclone->args[i].orig_arg;
1485 tree def = ssa_default_def (cfun, orig_arg);
1486 gcc_assert (!TREE_ADDRESSABLE (orig_arg)
1487 && TREE_CODE (TREE_TYPE (orig_arg)) == REFERENCE_TYPE);
1488 if (def && !has_zero_uses (def))
1490 tree rtype = TREE_TYPE (TREE_TYPE (orig_arg));
1491 iter1 = make_ssa_name (orig_arg);
1492 if (incr_bb)
1493 iter2 = make_ssa_name (orig_arg);
1494 tree iter3 = make_ssa_name (rtype);
1495 tree iter4 = make_ssa_name (rtype);
1496 tree iter5 = incr_bb ? make_ssa_name (rtype) : NULL_TREE;
1497 gsi = gsi_after_labels (entry_bb);
1498 gimple *load
1499 = gimple_build_assign (iter3, build_simple_mem_ref (def));
1500 gsi_insert_before (&gsi, load, GSI_NEW_STMT);
1502 tree array = node->simdclone->args[i].simd_array;
1503 TREE_ADDRESSABLE (array) = 1;
1504 tree ptr = build_fold_addr_expr (array);
1505 phi = create_phi_node (iter1, body_bb);
1506 add_phi_arg (phi, ptr, preheader_edge, UNKNOWN_LOCATION);
1507 if (incr_bb)
1509 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1510 g = gimple_build_assign (iter2, POINTER_PLUS_EXPR, iter1,
1511 TYPE_SIZE_UNIT (TREE_TYPE (iter3)));
1512 gsi = gsi_last_bb (incr_bb);
1513 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1516 phi = create_phi_node (iter4, body_bb);
1517 add_phi_arg (phi, iter3, preheader_edge, UNKNOWN_LOCATION);
1518 if (incr_bb)
1520 add_phi_arg (phi, iter5, latch_edge, UNKNOWN_LOCATION);
1521 enum tree_code code = INTEGRAL_TYPE_P (TREE_TYPE (iter3))
1522 ? PLUS_EXPR : POINTER_PLUS_EXPR;
1523 tree addtype = INTEGRAL_TYPE_P (TREE_TYPE (iter3))
1524 ? TREE_TYPE (iter3) : sizetype;
1525 tree addcst = simd_clone_linear_addend (node, i, addtype,
1526 entry_bb);
1527 g = gimple_build_assign (iter5, code, iter4, addcst);
1528 gsi = gsi_last_bb (incr_bb);
1529 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1532 g = gimple_build_assign (build_simple_mem_ref (iter1), iter4);
1533 gsi = gsi_after_labels (body_bb);
1534 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1536 imm_use_iterator iter;
1537 use_operand_p use_p;
1538 gimple *use_stmt;
1539 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1540 if (use_stmt == load)
1541 continue;
1542 else
1543 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1544 SET_USE (use_p, iter1);
1546 if (!TYPE_READONLY (rtype) && incr_bb)
1548 tree v = make_ssa_name (rtype);
1549 tree aref = build4 (ARRAY_REF, rtype, array,
1550 size_zero_node, NULL_TREE,
1551 NULL_TREE);
1552 gsi = gsi_after_labels (new_exit_bb);
1553 g = gimple_build_assign (v, aref);
1554 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1555 g = gimple_build_assign (build_simple_mem_ref (def), v);
1556 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1561 calculate_dominance_info (CDI_DOMINATORS);
1562 if (loop)
1563 add_loop (loop, loop->header->loop_father);
1564 update_ssa (TODO_update_ssa);
1566 pop_cfun ();
1569 /* If the function in NODE is tagged as an elemental SIMD function,
1570 create the appropriate SIMD clones. */
1572 void
1573 expand_simd_clones (struct cgraph_node *node)
1575 tree attr = lookup_attribute ("omp declare simd",
1576 DECL_ATTRIBUTES (node->decl));
1577 if (attr == NULL_TREE
1578 || node->global.inlined_to
1579 || lookup_attribute ("noclone", DECL_ATTRIBUTES (node->decl)))
1580 return;
1582 /* Ignore
1583 #pragma omp declare simd
1584 extern int foo ();
1585 in C, there we don't know the argument types at all. */
1586 if (!node->definition
1587 && TYPE_ARG_TYPES (TREE_TYPE (node->decl)) == NULL_TREE)
1588 return;
1590 /* Call this before creating clone_info, as it might ggc_collect. */
1591 if (node->definition && node->has_gimple_body_p ())
1592 node->get_body ();
1596 /* Start with parsing the "omp declare simd" attribute(s). */
1597 bool inbranch_clause_specified;
1598 struct cgraph_simd_clone *clone_info
1599 = simd_clone_clauses_extract (node, TREE_VALUE (attr),
1600 &inbranch_clause_specified);
1601 if (clone_info == NULL)
1602 continue;
1604 int orig_simdlen = clone_info->simdlen;
1605 tree base_type = simd_clone_compute_base_data_type (node, clone_info);
1606 /* The target can return 0 (no simd clones should be created),
1607 1 (just one ISA of simd clones should be created) or higher
1608 count of ISA variants. In that case, clone_info is initialized
1609 for the first ISA variant. */
1610 int count
1611 = targetm.simd_clone.compute_vecsize_and_simdlen (node, clone_info,
1612 base_type, 0);
1613 if (count == 0)
1614 continue;
1616 /* Loop over all COUNT ISA variants, and if !INBRANCH_CLAUSE_SPECIFIED,
1617 also create one inbranch and one !inbranch clone of it. */
1618 for (int i = 0; i < count * 2; i++)
1620 struct cgraph_simd_clone *clone = clone_info;
1621 if (inbranch_clause_specified && (i & 1) != 0)
1622 continue;
1624 if (i != 0)
1626 clone = simd_clone_struct_alloc (clone_info->nargs
1627 + ((i & 1) != 0));
1628 simd_clone_struct_copy (clone, clone_info);
1629 /* Undo changes targetm.simd_clone.compute_vecsize_and_simdlen
1630 and simd_clone_adjust_argument_types did to the first
1631 clone's info. */
1632 clone->nargs -= clone_info->inbranch;
1633 clone->simdlen = orig_simdlen;
1634 /* And call the target hook again to get the right ISA. */
1635 targetm.simd_clone.compute_vecsize_and_simdlen (node, clone,
1636 base_type,
1637 i / 2);
1638 if ((i & 1) != 0)
1639 clone->inbranch = 1;
1642 /* simd_clone_mangle might fail if such a clone has been created
1643 already. */
1644 tree id = simd_clone_mangle (node, clone);
1645 if (id == NULL_TREE)
1646 continue;
1648 /* Only when we are sure we want to create the clone actually
1649 clone the function (or definitions) or create another
1650 extern FUNCTION_DECL (for prototypes without definitions). */
1651 struct cgraph_node *n = simd_clone_create (node);
1652 if (n == NULL)
1653 continue;
1655 n->simdclone = clone;
1656 clone->origin = node;
1657 clone->next_clone = NULL;
1658 if (node->simd_clones == NULL)
1660 clone->prev_clone = n;
1661 node->simd_clones = n;
1663 else
1665 clone->prev_clone = node->simd_clones->simdclone->prev_clone;
1666 clone->prev_clone->simdclone->next_clone = n;
1667 node->simd_clones->simdclone->prev_clone = n;
1669 symtab->change_decl_assembler_name (n->decl, id);
1670 /* And finally adjust the return type, parameters and for
1671 definitions also function body. */
1672 if (node->definition)
1673 simd_clone_adjust (n);
1674 else
1676 simd_clone_adjust_return_type (n);
1677 simd_clone_adjust_argument_types (n);
1681 while ((attr = lookup_attribute ("omp declare simd", TREE_CHAIN (attr))));
1684 /* Entry point for IPA simd clone creation pass. */
1686 static unsigned int
1687 ipa_omp_simd_clone (void)
1689 struct cgraph_node *node;
1690 FOR_EACH_FUNCTION (node)
1691 expand_simd_clones (node);
1692 return 0;
1695 namespace {
1697 const pass_data pass_data_omp_simd_clone =
1699 SIMPLE_IPA_PASS, /* type */
1700 "simdclone", /* name */
1701 OPTGROUP_OMP, /* optinfo_flags */
1702 TV_NONE, /* tv_id */
1703 ( PROP_ssa | PROP_cfg ), /* properties_required */
1704 0, /* properties_provided */
1705 0, /* properties_destroyed */
1706 0, /* todo_flags_start */
1707 0, /* todo_flags_finish */
1710 class pass_omp_simd_clone : public simple_ipa_opt_pass
1712 public:
1713 pass_omp_simd_clone(gcc::context *ctxt)
1714 : simple_ipa_opt_pass(pass_data_omp_simd_clone, ctxt)
1717 /* opt_pass methods: */
1718 virtual bool gate (function *);
1719 virtual unsigned int execute (function *) { return ipa_omp_simd_clone (); }
1722 bool
1723 pass_omp_simd_clone::gate (function *)
1725 return targetm.simd_clone.compute_vecsize_and_simdlen != NULL;
1728 } // anon namespace
1730 simple_ipa_opt_pass *
1731 make_pass_omp_simd_clone (gcc::context *ctxt)
1733 return new pass_omp_simd_clone (ctxt);