2016-04-27 Hristian Kirtchev <kirtchev@adacore.com>
[official-gcc.git] / gcc / omp-simd-clone.c
blobfa6ffecb4c280c7c4e1b4f26008ef4eb80ea6a86
1 /* OMP constructs' SIMD clone supporting code.
3 Copyright (C) 2005-2016 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-prop.h"
49 #include "tree-eh.h"
52 /* Allocate a fresh `simd_clone' and return it. NARGS is the number
53 of arguments to reserve space for. */
55 static struct cgraph_simd_clone *
56 simd_clone_struct_alloc (int nargs)
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;
66 /* Make a copy of the `struct cgraph_simd_clone' in FROM to TO. */
68 static inline void
69 simd_clone_struct_copy (struct cgraph_simd_clone *to,
70 struct cgraph_simd_clone *from)
72 memcpy (to, from, (sizeof (struct cgraph_simd_clone)
73 + ((from->nargs - from->inbranch)
74 * sizeof (struct cgraph_simd_clone_arg))));
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. */
81 static vec<tree>
82 simd_clone_vector_of_formal_parm_types (tree fndecl)
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;
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. */
100 static struct cgraph_simd_clone *
101 simd_clone_clauses_extract (struct cgraph_node *node, tree clauses,
102 bool *inbranch_specified)
104 vec<tree> args = simd_clone_vector_of_formal_parm_types (node->decl);
105 tree t;
106 int n;
107 *inbranch_specified = false;
109 n = args.length ();
110 if (n > 0 && args.last () == void_type_node)
111 n--;
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)));
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;
127 if (!clauses)
129 args.release ();
130 return clone_info;
132 clauses = TREE_VALUE (clauses);
133 if (!clauses || TREE_CODE (clauses) != OMP_CLAUSE)
134 return clone_info;
136 for (t = clauses; t; t = OMP_CLAUSE_CHAIN (t))
138 switch (OMP_CLAUSE_CODE (t))
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:
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))
159 enum cgraph_simd_clone_arg_type arg_type;
160 if (TREE_CODE (args[argno]) == REFERENCE_TYPE)
161 switch (OMP_CLAUSE_LINEAR_KIND (t))
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 ();
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);
186 else
188 if (POINTER_TYPE_P (args[argno]))
189 step = fold_convert (ssizetype, step);
190 if (!tree_fits_shwi_p (step))
192 warning_at (OMP_CLAUSE_LOCATION (t), 0,
193 "ignoring large linear step");
194 args.release ();
195 return NULL;
197 else if (integer_zerop (step))
199 warning_at (OMP_CLAUSE_LOCATION (t), 0,
200 "ignoring zero linear step");
201 args.release ();
202 return NULL;
204 else
206 enum cgraph_simd_clone_arg_type arg_type;
207 if (TREE_CODE (args[argno]) == REFERENCE_TYPE)
208 switch (OMP_CLAUSE_LINEAR_KIND (t))
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 ();
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);
232 break;
234 case OMP_CLAUSE_UNIFORM:
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;
242 case OMP_CLAUSE_ALIGNED:
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;
250 default:
251 break;
254 args.release ();
255 return clone_info;
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. */
262 static tree
263 simd_clone_compute_base_data_type (struct cgraph_node *node,
264 struct cgraph_simd_clone *clone_info)
266 tree type = integer_type_node;
267 tree fndecl = node->decl;
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));
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
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)
283 type = map[i];
284 break;
286 map.release ();
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;
298 /* d) If none of the above three classes is applicable, the
299 characteristic data type is int. */
301 return type;
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. */
309 static tree
310 simd_clone_mangle (struct cgraph_node *node,
311 struct cgraph_simd_clone *clone_info)
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;
319 gcc_assert (vecsize_mangle && simdlen);
321 pp_string (&pp, "_ZGV");
322 pp_character (&pp, vecsize_mangle);
323 pp_character (&pp, mask);
324 pp_decimal_int (&pp, simdlen);
326 for (n = 0; n < clone_info->nargs; ++n)
328 struct cgraph_simd_clone_arg arg = clone_info->args[n];
330 switch (arg.arg_type)
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)
353 pp_character (&pp, 'n');
354 pp_unsigned_wide_integer (&pp, (-(unsigned HOST_WIDE_INT)
355 arg.linear_step));
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');
377 if (arg.alignment)
379 pp_character (&pp, 'a');
380 pp_decimal_int (&pp, arg.alignment);
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);
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;
403 return get_identifier (str);
406 /* Create a simd clone of OLD_NODE and return it. */
408 static struct cgraph_node *
409 simd_clone_create (struct cgraph_node *old_node)
411 struct cgraph_node *new_node;
412 if (old_node->definition)
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");
421 else
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;
434 if (new_node == NULL)
435 return new_node;
437 TREE_PUBLIC (new_node->decl) = TREE_PUBLIC (old_node->decl);
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;
445 return new_node;
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. */
452 static tree
453 simd_clone_adjust_return_type (struct cgraph_node *node)
455 tree fndecl = node->decl;
456 tree orig_rettype = TREE_TYPE (TREE_TYPE (fndecl));
457 unsigned int veclen;
458 tree t;
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
478 t = build_vector_type (t, veclen);
479 t = build_array_type_nelts (t, node->simdclone->simdlen / veclen);
481 TREE_TYPE (TREE_TYPE (fndecl)) = t;
482 if (!node->definition)
483 return NULL_TREE;
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);
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);
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;
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.
506 PREFIX is the prefix to be used for the temporary.
508 TYPE is the inner element type.
510 SIMDLEN is the number of elements. */
512 static tree
513 create_tmp_simd_array (const char *prefix, tree type, int simdlen)
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;
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.
526 NODE is the function whose arguments are to be adjusted.
528 Returns an adjustment vector that will be filled describing how the
529 argument types will be adjusted. */
531 static ipa_parm_adjustment_vec
532 simd_clone_adjust_argument_types (struct cgraph_node *node)
534 vec<tree> args;
535 ipa_parm_adjustment_vec adjustments;
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;
546 for (i = 0; i < sc->nargs; ++i)
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;
554 sc->args[i].orig_arg = node->definition ? parm : NULL_TREE;
555 sc->args[i].orig_type = parm_type;
557 switch (sc->args[i].arg_type)
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)
590 adjustments.safe_push (adj);
591 if (j == veclen)
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;
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);
606 adjustments.safe_push (adj);
609 if (sc->inbranch)
611 tree base_type = simd_clone_compute_base_data_type (sc->origin, sc);
613 memset (&adj, 0, sizeof (adj));
614 adj.op = IPA_PARM_OP_NEW;
615 adj.arg_prefix = "mask";
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);
634 for (j = veclen; j < sc->simdlen; j += veclen)
635 adjustments.safe_push (adj);
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)
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;
655 sc->args[i].orig_type = base_type;
656 sc->args[i].arg_type = SIMD_CLONE_ARG_TYPE_MASK;
659 if (node->definition)
660 ipa_modify_formal_parameters (node->decl, adjustments);
661 else
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;
668 gcc_assert (TYPE_ARG_TYPES (TREE_TYPE (node->decl)));
669 j = adjustments.length ();
670 for (i = 0; i < j; i++)
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);
680 new_reversed = nreverse (new_arg_types);
681 if (last_parm_void)
683 if (new_reversed)
684 TREE_CHAIN (new_arg_types) = void_list_node;
685 else
686 new_reversed = void_list_node;
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;
693 adjustments.release ();
695 args.release ();
696 return adjustments;
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. */
703 static gimple_seq
704 simd_clone_init_simd_arrays (struct cgraph_node *node,
705 ipa_parm_adjustment_vec adjustments)
707 gimple_seq seq = NULL;
708 unsigned i = 0, j = 0, k;
710 for (tree arg = DECL_ARGUMENTS (node->decl);
711 arg;
712 arg = DECL_CHAIN (arg), i++, j++)
714 if (adjustments[j].op == IPA_PARM_OP_COPY
715 || POINTER_TYPE_P (TREE_TYPE (arg)))
716 continue;
718 node->simdclone->args[i].vector_arg = arg;
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)
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++)
730 if (k)
732 arg = DECL_CHAIN (arg);
733 j++;
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);
740 continue;
742 if (TYPE_VECTOR_SUBPARTS (TREE_TYPE (arg)) == node->simdclone->simdlen)
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);
751 else
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)
757 tree ptr = build_fold_addr_expr (array);
758 int elemsize;
759 if (k)
761 arg = DECL_CHAIN (arg);
762 j++;
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);
773 return seq;
776 /* Callback info for ipa_simd_modify_stmt_ops below. */
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;
786 /* Callback for walk_gimple_op.
788 Adjust operands from a given statement as specified in the
789 adjustments vector in the callback data. */
791 static tree
792 ipa_simd_modify_stmt_ops (tree *tp, int *walk_subtrees, void *data)
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
804 if (TYPE_P (*tp))
805 *walk_subtrees = 0;
808 tree repl = NULL_TREE;
809 if (cand)
810 repl = unshare_expr (cand->new_decl);
811 else
813 if (tp != orig_tp)
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)
821 info->modified = modified;
822 return NULL_TREE;
824 info->modified = modified;
825 repl = *tp;
827 else
828 return NULL_TREE;
831 if (tp != orig_tp)
833 repl = build_fold_addr_expr (repl);
834 gimple *stmt;
835 if (is_gimple_debug (info->stmt))
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;
844 else
846 stmt = gimple_build_assign (make_ssa_name (TREE_TYPE (repl)), repl);
847 repl = gimple_assign_lhs (stmt);
849 gimple_stmt_iterator gsi = gsi_for_stmt (info->stmt);
850 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
851 *orig_tp = repl;
853 else if (!useless_type_conversion_p (TREE_TYPE (*tp), TREE_TYPE (repl)))
855 tree vce = build1 (VIEW_CONVERT_EXPR, TREE_TYPE (*tp), repl);
856 *tp = vce;
858 else
859 *tp = repl;
861 info->modified = true;
862 return NULL_TREE;
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.
870 This function will replace all function argument uses with their
871 corresponding simd array elements, and ajust the return values
872 accordingly. */
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)
879 basic_block bb;
880 unsigned int i, j, l;
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)
887 if (!node->simdclone->args[i].vector_arg)
888 continue;
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;
903 l = adjustments.length ();
904 for (i = 1; i < num_ssa_names; i++)
906 tree name = ssa_name (i);
907 if (name
908 && SSA_NAME_VAR (name)
909 && TREE_CODE (SSA_NAME_VAR (name)) == PARM_DECL)
911 for (j = 0; j < l; j++)
912 if (SSA_NAME_VAR (name) == adjustments[j].base
913 && adjustments[j].new_decl)
915 tree base_var;
916 if (adjustments[j].new_ssa_base == NULL_TREE)
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;
924 else
925 base_var = adjustments[j].new_ssa_base;
926 if (SSA_NAME_IS_DEFAULT_DEF (name))
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);
937 else
938 SET_SSA_NAME_VAR_OR_IDENTIFIER (name, base_var);
943 struct modify_stmt_info info;
944 info.adjustments = adjustments;
946 FOR_EACH_BB_FN (bb, DECL_STRUCT_FUNCTION (node->decl))
948 gimple_stmt_iterator gsi;
950 gsi = gsi_start_bb (bb);
951 while (!gsi_end_p (gsi))
953 gimple *stmt = gsi_stmt (gsi);
954 info.stmt = stmt;
955 struct walk_stmt_info wi;
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);
962 if (greturn *return_stmt = dyn_cast <greturn *> (stmt))
964 tree retval = gimple_return_retval (return_stmt);
965 if (!retval)
967 gsi_remove (&gsi, true);
968 continue;
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;
979 if (info.modified)
981 update_stmt (stmt);
982 if (maybe_clean_eh_stmt (stmt))
983 gimple_purge_dead_eh_edges (gimple_bb (stmt));
985 gsi_next (&gsi);
990 /* Helper function of simd_clone_adjust, return linear step addend
991 of Ith argument. */
993 static tree
994 simd_clone_linear_addend (struct cgraph_node *node, unsigned int i,
995 tree addtype, basic_block entry_bb)
997 tree ptype = NULL_TREE;
998 switch (node->simdclone->args[i].arg_type)
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 ();
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
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);
1031 if (TREE_CODE (TREE_TYPE (arg)) == REFERENCE_TYPE)
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);
1038 if (!useless_type_conversion_p (addtype, TREE_TYPE (ret)))
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);
1044 if (POINTER_TYPE_P (ptype))
1046 tree size = TYPE_SIZE_UNIT (TREE_TYPE (ptype));
1047 if (size && TREE_CODE (size) == INTEGER_CST)
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);
1055 return ret;
1058 /* Adjust the argument types in NODE to their appropriate vector
1059 counterparts. */
1061 static void
1062 simd_clone_adjust (struct cgraph_node *node)
1064 push_cfun (DECL_STRUCT_FUNCTION (node->decl));
1066 targetm.simd_clone.adjust (node);
1068 tree retval = simd_clone_adjust_return_type (node);
1069 ipa_parm_adjustment_vec adjustments
1070 = simd_clone_adjust_argument_types (node);
1072 push_gimplify_context ();
1074 gimple_seq seq = simd_clone_init_simd_arrays (node, adjustments);
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);
1083 /* Initialize the iteration variable. */
1084 basic_block entry_bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1085 basic_block body_bb = split_block_after_labels (entry_bb)->dest;
1086 gimple_stmt_iterator gsi = gsi_after_labels (entry_bb);
1087 /* Insert the SIMD array and iv initialization at function
1088 entry. */
1089 gsi_insert_seq_before (&gsi, seq, GSI_NEW_STMT);
1091 pop_gimplify_context (NULL);
1093 /* Create a new BB right before the original exit BB, to hold the
1094 iteration increment and the condition/branch. */
1095 basic_block orig_exit = EDGE_PRED (EXIT_BLOCK_PTR_FOR_FN (cfun), 0)->src;
1096 basic_block incr_bb = create_empty_bb (orig_exit);
1097 add_bb_to_loop (incr_bb, body_bb->loop_father);
1098 /* The succ of orig_exit was EXIT_BLOCK_PTR_FOR_FN (cfun), with an empty
1099 flag. Set it now to be a FALLTHRU_EDGE. */
1100 gcc_assert (EDGE_COUNT (orig_exit->succs) == 1);
1101 EDGE_SUCC (orig_exit, 0)->flags |= EDGE_FALLTHRU;
1102 for (unsigned i = 0;
1103 i < EDGE_COUNT (EXIT_BLOCK_PTR_FOR_FN (cfun)->preds); ++i)
1105 edge e = EDGE_PRED (EXIT_BLOCK_PTR_FOR_FN (cfun), i);
1106 redirect_edge_succ (e, incr_bb);
1108 edge e = make_edge (incr_bb, EXIT_BLOCK_PTR_FOR_FN (cfun), 0);
1109 e->probability = REG_BR_PROB_BASE;
1110 gsi = gsi_last_bb (incr_bb);
1111 gimple *g = gimple_build_assign (iter2, PLUS_EXPR, iter1,
1112 build_int_cst (unsigned_type_node, 1));
1113 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1115 /* Mostly annotate the loop for the vectorizer (the rest is done below). */
1116 struct loop *loop = alloc_loop ();
1117 cfun->has_force_vectorize_loops = true;
1118 loop->safelen = node->simdclone->simdlen;
1119 loop->force_vectorize = true;
1120 loop->header = body_bb;
1122 /* Branch around the body if the mask applies. */
1123 if (node->simdclone->inbranch)
1125 gimple_stmt_iterator gsi = gsi_last_bb (loop->header);
1126 tree mask_array
1127 = node->simdclone->args[node->simdclone->nargs - 1].simd_array;
1128 tree mask;
1129 if (node->simdclone->mask_mode != VOIDmode)
1131 tree shift_cnt;
1132 if (mask_array == NULL_TREE)
1134 tree arg = node->simdclone->args[node->simdclone->nargs
1135 - 1].vector_arg;
1136 mask = get_or_create_ssa_default_def (cfun, arg);
1137 shift_cnt = iter1;
1139 else
1141 tree maskt = TREE_TYPE (mask_array);
1142 int c = tree_to_uhwi (TYPE_MAX_VALUE (TYPE_DOMAIN (maskt)));
1143 c = node->simdclone->simdlen / (c + 1);
1144 int s = exact_log2 (c);
1145 gcc_assert (s > 0);
1146 c--;
1147 tree idx = make_ssa_name (TREE_TYPE (iter1));
1148 g = gimple_build_assign (idx, RSHIFT_EXPR, iter1,
1149 build_int_cst (NULL_TREE, s));
1150 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1151 mask = make_ssa_name (TREE_TYPE (TREE_TYPE (mask_array)));
1152 tree aref = build4 (ARRAY_REF,
1153 TREE_TYPE (TREE_TYPE (mask_array)),
1154 mask_array, idx, NULL, NULL);
1155 g = gimple_build_assign (mask, aref);
1156 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1157 shift_cnt = make_ssa_name (TREE_TYPE (iter1));
1158 g = gimple_build_assign (shift_cnt, BIT_AND_EXPR, iter1,
1159 build_int_cst (TREE_TYPE (iter1), c));
1160 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1162 g = gimple_build_assign (make_ssa_name (TREE_TYPE (mask)),
1163 RSHIFT_EXPR, mask, shift_cnt);
1164 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1165 mask = gimple_assign_lhs (g);
1166 g = gimple_build_assign (make_ssa_name (TREE_TYPE (mask)),
1167 BIT_AND_EXPR, mask,
1168 build_int_cst (TREE_TYPE (mask), 1));
1169 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1170 mask = gimple_assign_lhs (g);
1172 else
1174 mask = make_ssa_name (TREE_TYPE (TREE_TYPE (mask_array)));
1175 tree aref = build4 (ARRAY_REF,
1176 TREE_TYPE (TREE_TYPE (mask_array)),
1177 mask_array, iter1, NULL, NULL);
1178 g = gimple_build_assign (mask, aref);
1179 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1180 int bitsize = GET_MODE_BITSIZE (TYPE_MODE (TREE_TYPE (aref)));
1181 if (!INTEGRAL_TYPE_P (TREE_TYPE (aref)))
1183 aref = build1 (VIEW_CONVERT_EXPR,
1184 build_nonstandard_integer_type (bitsize, 0),
1185 mask);
1186 mask = make_ssa_name (TREE_TYPE (aref));
1187 g = gimple_build_assign (mask, aref);
1188 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1192 g = gimple_build_cond (EQ_EXPR, mask, build_zero_cst (TREE_TYPE (mask)),
1193 NULL, NULL);
1194 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1195 make_edge (loop->header, incr_bb, EDGE_TRUE_VALUE);
1196 FALLTHRU_EDGE (loop->header)->flags = EDGE_FALSE_VALUE;
1199 /* Generate the condition. */
1200 g = gimple_build_cond (LT_EXPR,
1201 iter2,
1202 build_int_cst (unsigned_type_node,
1203 node->simdclone->simdlen),
1204 NULL, NULL);
1205 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1206 e = split_block (incr_bb, gsi_stmt (gsi));
1207 basic_block latch_bb = e->dest;
1208 basic_block new_exit_bb;
1209 new_exit_bb = split_block_after_labels (latch_bb)->dest;
1210 loop->latch = latch_bb;
1212 redirect_edge_succ (FALLTHRU_EDGE (latch_bb), body_bb);
1214 make_edge (incr_bb, new_exit_bb, EDGE_FALSE_VALUE);
1215 /* The successor of incr_bb is already pointing to latch_bb; just
1216 change the flags.
1217 make_edge (incr_bb, latch_bb, EDGE_TRUE_VALUE); */
1218 FALLTHRU_EDGE (incr_bb)->flags = EDGE_TRUE_VALUE;
1220 gphi *phi = create_phi_node (iter1, body_bb);
1221 edge preheader_edge = find_edge (entry_bb, body_bb);
1222 edge latch_edge = single_succ_edge (latch_bb);
1223 add_phi_arg (phi, build_zero_cst (unsigned_type_node), preheader_edge,
1224 UNKNOWN_LOCATION);
1225 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1227 /* Generate the new return. */
1228 gsi = gsi_last_bb (new_exit_bb);
1229 if (retval
1230 && TREE_CODE (retval) == VIEW_CONVERT_EXPR
1231 && TREE_CODE (TREE_OPERAND (retval, 0)) == RESULT_DECL)
1232 retval = TREE_OPERAND (retval, 0);
1233 else if (retval)
1235 retval = build1 (VIEW_CONVERT_EXPR,
1236 TREE_TYPE (TREE_TYPE (node->decl)),
1237 retval);
1238 retval = force_gimple_operand_gsi (&gsi, retval, true, NULL,
1239 false, GSI_CONTINUE_LINKING);
1241 g = gimple_build_return (retval);
1242 gsi_insert_after (&gsi, g, GSI_CONTINUE_LINKING);
1244 /* Handle aligned clauses by replacing default defs of the aligned
1245 uniform args with __builtin_assume_aligned (arg_N(D), alignment)
1246 lhs. Handle linear by adding PHIs. */
1247 for (unsigned i = 0; i < node->simdclone->nargs; i++)
1248 if (node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_UNIFORM
1249 && (TREE_ADDRESSABLE (node->simdclone->args[i].orig_arg)
1250 || !is_gimple_reg_type
1251 (TREE_TYPE (node->simdclone->args[i].orig_arg))))
1253 tree orig_arg = node->simdclone->args[i].orig_arg;
1254 if (is_gimple_reg_type (TREE_TYPE (orig_arg)))
1255 iter1 = make_ssa_name (TREE_TYPE (orig_arg));
1256 else
1258 iter1 = create_tmp_var_raw (TREE_TYPE (orig_arg));
1259 gimple_add_tmp_var (iter1);
1261 gsi = gsi_after_labels (entry_bb);
1262 g = gimple_build_assign (iter1, orig_arg);
1263 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1264 gsi = gsi_after_labels (body_bb);
1265 g = gimple_build_assign (orig_arg, iter1);
1266 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1268 else if (node->simdclone->args[i].arg_type == SIMD_CLONE_ARG_TYPE_UNIFORM
1269 && DECL_BY_REFERENCE (node->simdclone->args[i].orig_arg)
1270 && TREE_CODE (TREE_TYPE (node->simdclone->args[i].orig_arg))
1271 == REFERENCE_TYPE
1272 && TREE_ADDRESSABLE
1273 (TREE_TYPE (TREE_TYPE (node->simdclone->args[i].orig_arg))))
1275 tree orig_arg = node->simdclone->args[i].orig_arg;
1276 tree def = ssa_default_def (cfun, orig_arg);
1277 if (def && !has_zero_uses (def))
1279 iter1 = create_tmp_var_raw (TREE_TYPE (TREE_TYPE (orig_arg)));
1280 gimple_add_tmp_var (iter1);
1281 gsi = gsi_after_labels (entry_bb);
1282 g = gimple_build_assign (iter1, build_simple_mem_ref (def));
1283 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1284 gsi = gsi_after_labels (body_bb);
1285 g = gimple_build_assign (build_simple_mem_ref (def), iter1);
1286 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1289 else if (node->simdclone->args[i].alignment
1290 && node->simdclone->args[i].arg_type
1291 == SIMD_CLONE_ARG_TYPE_UNIFORM
1292 && (node->simdclone->args[i].alignment
1293 & (node->simdclone->args[i].alignment - 1)) == 0
1294 && TREE_CODE (TREE_TYPE (node->simdclone->args[i].orig_arg))
1295 == POINTER_TYPE)
1297 unsigned int alignment = node->simdclone->args[i].alignment;
1298 tree orig_arg = node->simdclone->args[i].orig_arg;
1299 tree def = ssa_default_def (cfun, orig_arg);
1300 if (def && !has_zero_uses (def))
1302 tree fn = builtin_decl_explicit (BUILT_IN_ASSUME_ALIGNED);
1303 gimple_seq seq = NULL;
1304 bool need_cvt = false;
1305 gcall *call
1306 = gimple_build_call (fn, 2, def, size_int (alignment));
1307 g = call;
1308 if (!useless_type_conversion_p (TREE_TYPE (orig_arg),
1309 ptr_type_node))
1310 need_cvt = true;
1311 tree t = make_ssa_name (need_cvt ? ptr_type_node : orig_arg);
1312 gimple_call_set_lhs (g, t);
1313 gimple_seq_add_stmt_without_update (&seq, g);
1314 if (need_cvt)
1316 t = make_ssa_name (orig_arg);
1317 g = gimple_build_assign (t, NOP_EXPR, gimple_call_lhs (g));
1318 gimple_seq_add_stmt_without_update (&seq, g);
1320 gsi_insert_seq_on_edge_immediate
1321 (single_succ_edge (ENTRY_BLOCK_PTR_FOR_FN (cfun)), seq);
1323 entry_bb = single_succ (ENTRY_BLOCK_PTR_FOR_FN (cfun));
1324 int freq = compute_call_stmt_bb_frequency (current_function_decl,
1325 entry_bb);
1326 node->create_edge (cgraph_node::get_create (fn),
1327 call, entry_bb->count, freq);
1329 imm_use_iterator iter;
1330 use_operand_p use_p;
1331 gimple *use_stmt;
1332 tree repl = gimple_get_lhs (g);
1333 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1334 if (is_gimple_debug (use_stmt) || use_stmt == call)
1335 continue;
1336 else
1337 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1338 SET_USE (use_p, repl);
1341 else if ((node->simdclone->args[i].arg_type
1342 == SIMD_CLONE_ARG_TYPE_LINEAR_CONSTANT_STEP)
1343 || (node->simdclone->args[i].arg_type
1344 == SIMD_CLONE_ARG_TYPE_LINEAR_REF_CONSTANT_STEP)
1345 || (node->simdclone->args[i].arg_type
1346 == SIMD_CLONE_ARG_TYPE_LINEAR_VARIABLE_STEP)
1347 || (node->simdclone->args[i].arg_type
1348 == SIMD_CLONE_ARG_TYPE_LINEAR_REF_VARIABLE_STEP))
1350 tree orig_arg = node->simdclone->args[i].orig_arg;
1351 gcc_assert (INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1352 || POINTER_TYPE_P (TREE_TYPE (orig_arg)));
1353 tree def = NULL_TREE;
1354 if (TREE_ADDRESSABLE (orig_arg))
1356 def = make_ssa_name (TREE_TYPE (orig_arg));
1357 iter1 = make_ssa_name (TREE_TYPE (orig_arg));
1358 iter2 = make_ssa_name (TREE_TYPE (orig_arg));
1359 gsi = gsi_after_labels (entry_bb);
1360 g = gimple_build_assign (def, orig_arg);
1361 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1363 else
1365 def = ssa_default_def (cfun, orig_arg);
1366 if (!def || has_zero_uses (def))
1367 def = NULL_TREE;
1368 else
1370 iter1 = make_ssa_name (orig_arg);
1371 iter2 = make_ssa_name (orig_arg);
1374 if (def)
1376 phi = create_phi_node (iter1, body_bb);
1377 add_phi_arg (phi, def, preheader_edge, UNKNOWN_LOCATION);
1378 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1379 enum tree_code code = INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1380 ? PLUS_EXPR : POINTER_PLUS_EXPR;
1381 tree addtype = INTEGRAL_TYPE_P (TREE_TYPE (orig_arg))
1382 ? TREE_TYPE (orig_arg) : sizetype;
1383 tree addcst = simd_clone_linear_addend (node, i, addtype,
1384 entry_bb);
1385 gsi = gsi_last_bb (incr_bb);
1386 g = gimple_build_assign (iter2, code, iter1, addcst);
1387 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1389 imm_use_iterator iter;
1390 use_operand_p use_p;
1391 gimple *use_stmt;
1392 if (TREE_ADDRESSABLE (orig_arg))
1394 gsi = gsi_after_labels (body_bb);
1395 g = gimple_build_assign (orig_arg, iter1);
1396 gsi_insert_before (&gsi, g, GSI_NEW_STMT);
1398 else
1399 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1400 if (use_stmt == phi)
1401 continue;
1402 else
1403 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1404 SET_USE (use_p, iter1);
1407 else if (node->simdclone->args[i].arg_type
1408 == SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_CONSTANT_STEP
1409 || (node->simdclone->args[i].arg_type
1410 == SIMD_CLONE_ARG_TYPE_LINEAR_UVAL_VARIABLE_STEP))
1412 tree orig_arg = node->simdclone->args[i].orig_arg;
1413 tree def = ssa_default_def (cfun, orig_arg);
1414 gcc_assert (!TREE_ADDRESSABLE (orig_arg)
1415 && TREE_CODE (TREE_TYPE (orig_arg)) == REFERENCE_TYPE);
1416 if (def && !has_zero_uses (def))
1418 tree rtype = TREE_TYPE (TREE_TYPE (orig_arg));
1419 iter1 = make_ssa_name (orig_arg);
1420 iter2 = make_ssa_name (orig_arg);
1421 tree iter3 = make_ssa_name (rtype);
1422 tree iter4 = make_ssa_name (rtype);
1423 tree iter5 = make_ssa_name (rtype);
1424 gsi = gsi_after_labels (entry_bb);
1425 gimple *load
1426 = gimple_build_assign (iter3, build_simple_mem_ref (def));
1427 gsi_insert_before (&gsi, load, GSI_NEW_STMT);
1429 tree array = node->simdclone->args[i].simd_array;
1430 TREE_ADDRESSABLE (array) = 1;
1431 tree ptr = build_fold_addr_expr (array);
1432 phi = create_phi_node (iter1, body_bb);
1433 add_phi_arg (phi, ptr, preheader_edge, UNKNOWN_LOCATION);
1434 add_phi_arg (phi, iter2, latch_edge, UNKNOWN_LOCATION);
1435 g = gimple_build_assign (iter2, POINTER_PLUS_EXPR, iter1,
1436 TYPE_SIZE_UNIT (TREE_TYPE (iter3)));
1437 gsi = gsi_last_bb (incr_bb);
1438 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1440 phi = create_phi_node (iter4, body_bb);
1441 add_phi_arg (phi, iter3, preheader_edge, UNKNOWN_LOCATION);
1442 add_phi_arg (phi, iter5, latch_edge, UNKNOWN_LOCATION);
1443 enum tree_code code = INTEGRAL_TYPE_P (TREE_TYPE (iter3))
1444 ? PLUS_EXPR : POINTER_PLUS_EXPR;
1445 tree addtype = INTEGRAL_TYPE_P (TREE_TYPE (iter3))
1446 ? TREE_TYPE (iter3) : sizetype;
1447 tree addcst = simd_clone_linear_addend (node, i, addtype,
1448 entry_bb);
1449 g = gimple_build_assign (iter5, code, iter4, addcst);
1450 gsi = gsi_last_bb (incr_bb);
1451 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1453 g = gimple_build_assign (build_simple_mem_ref (iter1), iter4);
1454 gsi = gsi_after_labels (body_bb);
1455 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1457 imm_use_iterator iter;
1458 use_operand_p use_p;
1459 gimple *use_stmt;
1460 FOR_EACH_IMM_USE_STMT (use_stmt, iter, def)
1461 if (use_stmt == load)
1462 continue;
1463 else
1464 FOR_EACH_IMM_USE_ON_STMT (use_p, iter)
1465 SET_USE (use_p, iter1);
1467 if (!TYPE_READONLY (rtype))
1469 tree v = make_ssa_name (rtype);
1470 tree aref = build4 (ARRAY_REF, rtype, array,
1471 size_zero_node, NULL_TREE,
1472 NULL_TREE);
1473 gsi = gsi_after_labels (new_exit_bb);
1474 g = gimple_build_assign (v, aref);
1475 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1476 g = gimple_build_assign (build_simple_mem_ref (def), v);
1477 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
1482 calculate_dominance_info (CDI_DOMINATORS);
1483 add_loop (loop, loop->header->loop_father);
1484 update_ssa (TODO_update_ssa);
1486 pop_cfun ();
1489 /* If the function in NODE is tagged as an elemental SIMD function,
1490 create the appropriate SIMD clones. */
1492 static void
1493 expand_simd_clones (struct cgraph_node *node)
1495 tree attr = lookup_attribute ("omp declare simd",
1496 DECL_ATTRIBUTES (node->decl));
1497 if (attr == NULL_TREE
1498 || node->global.inlined_to
1499 || lookup_attribute ("noclone", DECL_ATTRIBUTES (node->decl)))
1500 return;
1502 /* Ignore
1503 #pragma omp declare simd
1504 extern int foo ();
1505 in C, there we don't know the argument types at all. */
1506 if (!node->definition
1507 && TYPE_ARG_TYPES (TREE_TYPE (node->decl)) == NULL_TREE)
1508 return;
1510 /* Call this before creating clone_info, as it might ggc_collect. */
1511 if (node->definition && node->has_gimple_body_p ())
1512 node->get_body ();
1516 /* Start with parsing the "omp declare simd" attribute(s). */
1517 bool inbranch_clause_specified;
1518 struct cgraph_simd_clone *clone_info
1519 = simd_clone_clauses_extract (node, TREE_VALUE (attr),
1520 &inbranch_clause_specified);
1521 if (clone_info == NULL)
1522 continue;
1524 int orig_simdlen = clone_info->simdlen;
1525 tree base_type = simd_clone_compute_base_data_type (node, clone_info);
1526 /* The target can return 0 (no simd clones should be created),
1527 1 (just one ISA of simd clones should be created) or higher
1528 count of ISA variants. In that case, clone_info is initialized
1529 for the first ISA variant. */
1530 int count
1531 = targetm.simd_clone.compute_vecsize_and_simdlen (node, clone_info,
1532 base_type, 0);
1533 if (count == 0)
1534 continue;
1536 /* Loop over all COUNT ISA variants, and if !INBRANCH_CLAUSE_SPECIFIED,
1537 also create one inbranch and one !inbranch clone of it. */
1538 for (int i = 0; i < count * 2; i++)
1540 struct cgraph_simd_clone *clone = clone_info;
1541 if (inbranch_clause_specified && (i & 1) != 0)
1542 continue;
1544 if (i != 0)
1546 clone = simd_clone_struct_alloc (clone_info->nargs
1547 + ((i & 1) != 0));
1548 simd_clone_struct_copy (clone, clone_info);
1549 /* Undo changes targetm.simd_clone.compute_vecsize_and_simdlen
1550 and simd_clone_adjust_argument_types did to the first
1551 clone's info. */
1552 clone->nargs -= clone_info->inbranch;
1553 clone->simdlen = orig_simdlen;
1554 /* And call the target hook again to get the right ISA. */
1555 targetm.simd_clone.compute_vecsize_and_simdlen (node, clone,
1556 base_type,
1557 i / 2);
1558 if ((i & 1) != 0)
1559 clone->inbranch = 1;
1562 /* simd_clone_mangle might fail if such a clone has been created
1563 already. */
1564 tree id = simd_clone_mangle (node, clone);
1565 if (id == NULL_TREE)
1566 continue;
1568 /* Only when we are sure we want to create the clone actually
1569 clone the function (or definitions) or create another
1570 extern FUNCTION_DECL (for prototypes without definitions). */
1571 struct cgraph_node *n = simd_clone_create (node);
1572 if (n == NULL)
1573 continue;
1575 n->simdclone = clone;
1576 clone->origin = node;
1577 clone->next_clone = NULL;
1578 if (node->simd_clones == NULL)
1580 clone->prev_clone = n;
1581 node->simd_clones = n;
1583 else
1585 clone->prev_clone = node->simd_clones->simdclone->prev_clone;
1586 clone->prev_clone->simdclone->next_clone = n;
1587 node->simd_clones->simdclone->prev_clone = n;
1589 symtab->change_decl_assembler_name (n->decl, id);
1590 /* And finally adjust the return type, parameters and for
1591 definitions also function body. */
1592 if (node->definition)
1593 simd_clone_adjust (n);
1594 else
1596 simd_clone_adjust_return_type (n);
1597 simd_clone_adjust_argument_types (n);
1601 while ((attr = lookup_attribute ("omp declare simd", TREE_CHAIN (attr))));
1604 /* Entry point for IPA simd clone creation pass. */
1606 static unsigned int
1607 ipa_omp_simd_clone (void)
1609 struct cgraph_node *node;
1610 FOR_EACH_FUNCTION (node)
1611 expand_simd_clones (node);
1612 return 0;
1615 namespace {
1617 const pass_data pass_data_omp_simd_clone =
1619 SIMPLE_IPA_PASS, /* type */
1620 "simdclone", /* name */
1621 OPTGROUP_NONE, /* optinfo_flags */
1622 TV_NONE, /* tv_id */
1623 ( PROP_ssa | PROP_cfg ), /* properties_required */
1624 0, /* properties_provided */
1625 0, /* properties_destroyed */
1626 0, /* todo_flags_start */
1627 0, /* todo_flags_finish */
1630 class pass_omp_simd_clone : public simple_ipa_opt_pass
1632 public:
1633 pass_omp_simd_clone(gcc::context *ctxt)
1634 : simple_ipa_opt_pass(pass_data_omp_simd_clone, ctxt)
1637 /* opt_pass methods: */
1638 virtual bool gate (function *);
1639 virtual unsigned int execute (function *) { return ipa_omp_simd_clone (); }
1642 bool
1643 pass_omp_simd_clone::gate (function *)
1645 return targetm.simd_clone.compute_vecsize_and_simdlen != NULL;
1648 } // anon namespace
1650 simple_ipa_opt_pass *
1651 make_pass_omp_simd_clone (gcc::context *ctxt)
1653 return new pass_omp_simd_clone (ctxt);