blob: 2c095200d5b55724f8406b1385bedf4986ae64c5 [file] [log] [blame]
/* General types and functions that are useful for processing of OpenMP,
OpenACC and similar directives at various stages of compilation.
Copyright (C) 2005-2024 Free Software Foundation, Inc.
This file is part of GCC.
GCC is free software; you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by the Free
Software Foundation; either version 3, or (at your option) any later
version.
GCC is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License
along with GCC; see the file COPYING3. If not see
<http://www.gnu.org/licenses/>. */
#include "config.h"
#include "system.h"
#include "coretypes.h"
#include "backend.h"
#include "target.h"
#include "tree.h"
#include "gimple.h"
#include "ssa.h"
#include "diagnostic-core.h"
#include "fold-const.h"
#include "langhooks.h"
#include "omp-general.h"
#include "stringpool.h"
#include "attribs.h"
#include "gimplify.h"
#include "cgraph.h"
#include "alloc-pool.h"
#include "symbol-summary.h"
#include "tree-pass.h"
#include "omp-device-properties.h"
#include "tree-iterator.h"
#include "data-streamer.h"
#include "streamer-hooks.h"
#include "opts.h"
#include "tree-pretty-print.h"
enum omp_requires omp_requires_mask;
/* Find an OMP clause of type KIND within CLAUSES. */
tree
omp_find_clause (tree clauses, enum omp_clause_code kind)
{
for (; clauses ; clauses = OMP_CLAUSE_CHAIN (clauses))
if (OMP_CLAUSE_CODE (clauses) == kind)
return clauses;
return NULL_TREE;
}
/* True if OpenMP should regard this DECL as being a scalar which has Fortran's
allocatable or pointer attribute. */
bool
omp_is_allocatable_or_ptr (tree decl)
{
return lang_hooks.decls.omp_is_allocatable_or_ptr (decl);
}
/* Check whether this DECL belongs to a Fortran optional argument.
With 'for_present_check' set to false, decls which are optional parameters
themselve are returned as tree - or a NULL_TREE otherwise. Those decls are
always pointers. With 'for_present_check' set to true, the decl for checking
whether an argument is present is returned; for arguments with value
attribute this is the hidden argument and of BOOLEAN_TYPE. If the decl is
unrelated to optional arguments, NULL_TREE is returned. */
tree
omp_check_optional_argument (tree decl, bool for_present_check)
{
return lang_hooks.decls.omp_check_optional_argument (decl, for_present_check);
}
/* Return true if TYPE is an OpenMP mappable type. */
bool
omp_mappable_type (tree type)
{
/* Mappable type has to be complete. */
if (type == error_mark_node || !COMPLETE_TYPE_P (type))
return false;
return true;
}
/* True if OpenMP should privatize what this DECL points to rather
than the DECL itself. */
bool
omp_privatize_by_reference (tree decl)
{
return lang_hooks.decls.omp_privatize_by_reference (decl);
}
/* Adjust *COND_CODE and *N2 so that the former is either LT_EXPR or GT_EXPR,
given that V is the loop index variable and STEP is loop step. */
void
omp_adjust_for_condition (location_t loc, enum tree_code *cond_code, tree *n2,
tree v, tree step)
{
switch (*cond_code)
{
case LT_EXPR:
case GT_EXPR:
break;
case NE_EXPR:
gcc_assert (TREE_CODE (step) == INTEGER_CST);
if (TREE_CODE (TREE_TYPE (v)) == INTEGER_TYPE
|| TREE_CODE (TREE_TYPE (v)) == BITINT_TYPE)
{
if (integer_onep (step))
*cond_code = LT_EXPR;
else
{
gcc_assert (integer_minus_onep (step));
*cond_code = GT_EXPR;
}
}
else
{
tree unit = TYPE_SIZE_UNIT (TREE_TYPE (TREE_TYPE (v)));
gcc_assert (TREE_CODE (unit) == INTEGER_CST);
if (tree_int_cst_equal (unit, step))
*cond_code = LT_EXPR;
else
{
gcc_assert (wi::neg (wi::to_widest (unit))
== wi::to_widest (step));
*cond_code = GT_EXPR;
}
}
break;
case LE_EXPR:
if (POINTER_TYPE_P (TREE_TYPE (*n2)))
*n2 = fold_build_pointer_plus_hwi_loc (loc, *n2, 1);
else
*n2 = fold_build2_loc (loc, PLUS_EXPR, TREE_TYPE (*n2), *n2,
build_int_cst (TREE_TYPE (*n2), 1));
*cond_code = LT_EXPR;
break;
case GE_EXPR:
if (POINTER_TYPE_P (TREE_TYPE (*n2)))
*n2 = fold_build_pointer_plus_hwi_loc (loc, *n2, -1);
else
*n2 = fold_build2_loc (loc, MINUS_EXPR, TREE_TYPE (*n2), *n2,
build_int_cst (TREE_TYPE (*n2), 1));
*cond_code = GT_EXPR;
break;
default:
gcc_unreachable ();
}
}
/* Return the looping step from INCR, extracted from the step of a gimple omp
for statement. */
tree
omp_get_for_step_from_incr (location_t loc, tree incr)
{
tree step;
switch (TREE_CODE (incr))
{
case PLUS_EXPR:
step = TREE_OPERAND (incr, 1);
break;
case POINTER_PLUS_EXPR:
step = fold_convert (ssizetype, TREE_OPERAND (incr, 1));
break;
case MINUS_EXPR:
step = TREE_OPERAND (incr, 1);
step = fold_build1_loc (loc, NEGATE_EXPR, TREE_TYPE (step), step);
break;
default:
gcc_unreachable ();
}
return step;
}
/* Extract the header elements of parallel loop FOR_STMT and store
them into *FD. */
void
omp_extract_for_data (gomp_for *for_stmt, struct omp_for_data *fd,
struct omp_for_data_loop *loops)
{
tree t, var, *collapse_iter, *collapse_count;
tree count = NULL_TREE, iter_type = long_integer_type_node;
struct omp_for_data_loop *loop;
int i;
struct omp_for_data_loop dummy_loop;
location_t loc = gimple_location (for_stmt);
bool simd = gimple_omp_for_kind (for_stmt) == GF_OMP_FOR_KIND_SIMD;
bool distribute = gimple_omp_for_kind (for_stmt)
== GF_OMP_FOR_KIND_DISTRIBUTE;
bool taskloop = gimple_omp_for_kind (for_stmt)
== GF_OMP_FOR_KIND_TASKLOOP;
bool order_reproducible = false;
tree iterv, countv;
fd->for_stmt = for_stmt;
fd->pre = NULL;
fd->have_nowait = distribute || simd;
fd->have_ordered = false;
fd->have_reductemp = false;
fd->have_pointer_condtemp = false;
fd->have_scantemp = false;
fd->have_nonctrl_scantemp = false;
fd->non_rect = false;
fd->lastprivate_conditional = 0;
fd->tiling = NULL_TREE;
fd->collapse = 1;
fd->ordered = 0;
fd->first_nonrect = -1;
fd->last_nonrect = -1;
fd->sched_kind = OMP_CLAUSE_SCHEDULE_STATIC;
fd->sched_modifiers = 0;
fd->chunk_size = NULL_TREE;
fd->simd_schedule = false;
fd->first_inner_iterations = NULL_TREE;
fd->factor = NULL_TREE;
fd->adjn1 = NULL_TREE;
collapse_iter = NULL;
collapse_count = NULL;
for (t = gimple_omp_for_clauses (for_stmt); t ; t = OMP_CLAUSE_CHAIN (t))
switch (OMP_CLAUSE_CODE (t))
{
case OMP_CLAUSE_NOWAIT:
fd->have_nowait = true;
break;
case OMP_CLAUSE_ORDERED:
fd->have_ordered = true;
if (OMP_CLAUSE_ORDERED_DOACROSS (t))
{
if (OMP_CLAUSE_ORDERED_EXPR (t))
fd->ordered = tree_to_shwi (OMP_CLAUSE_ORDERED_EXPR (t));
else
fd->ordered = -1;
}
break;
case OMP_CLAUSE_SCHEDULE:
gcc_assert (!distribute && !taskloop);
fd->sched_kind
= (enum omp_clause_schedule_kind)
(OMP_CLAUSE_SCHEDULE_KIND (t) & OMP_CLAUSE_SCHEDULE_MASK);
fd->sched_modifiers = (OMP_CLAUSE_SCHEDULE_KIND (t)
& ~OMP_CLAUSE_SCHEDULE_MASK);
fd->chunk_size = OMP_CLAUSE_SCHEDULE_CHUNK_EXPR (t);
fd->simd_schedule = OMP_CLAUSE_SCHEDULE_SIMD (t);
break;
case OMP_CLAUSE_DIST_SCHEDULE:
gcc_assert (distribute);
fd->chunk_size = OMP_CLAUSE_DIST_SCHEDULE_CHUNK_EXPR (t);
break;
case OMP_CLAUSE_COLLAPSE:
fd->collapse = tree_to_shwi (OMP_CLAUSE_COLLAPSE_EXPR (t));
if (fd->collapse > 1)
{
collapse_iter = &OMP_CLAUSE_COLLAPSE_ITERVAR (t);
collapse_count = &OMP_CLAUSE_COLLAPSE_COUNT (t);
}
break;
case OMP_CLAUSE_TILE:
fd->tiling = OMP_CLAUSE_TILE_LIST (t);
fd->collapse = list_length (fd->tiling);
gcc_assert (fd->collapse);
collapse_iter = &OMP_CLAUSE_TILE_ITERVAR (t);
collapse_count = &OMP_CLAUSE_TILE_COUNT (t);
break;
case OMP_CLAUSE__REDUCTEMP_:
fd->have_reductemp = true;
break;
case OMP_CLAUSE_LASTPRIVATE:
if (OMP_CLAUSE_LASTPRIVATE_CONDITIONAL (t))
fd->lastprivate_conditional++;
break;
case OMP_CLAUSE__CONDTEMP_:
if (POINTER_TYPE_P (TREE_TYPE (OMP_CLAUSE_DECL (t))))
fd->have_pointer_condtemp = true;
break;
case OMP_CLAUSE__SCANTEMP_:
fd->have_scantemp = true;
if (!OMP_CLAUSE__SCANTEMP__ALLOC (t)
&& !OMP_CLAUSE__SCANTEMP__CONTROL (t))
fd->have_nonctrl_scantemp = true;
break;
case OMP_CLAUSE_ORDER:
/* FIXME: For OpenMP 5.2 this should change to
if (OMP_CLAUSE_ORDER_REPRODUCIBLE (t))
(with the exception of loop construct but that lowers to
no schedule/dist_schedule clauses currently). */
if (!OMP_CLAUSE_ORDER_UNCONSTRAINED (t))
order_reproducible = true;
default:
break;
}
if (fd->ordered == -1)
fd->ordered = fd->collapse;
/* For order(reproducible:concurrent) schedule ({dynamic,guided,runtime})
we have either the option to expensively remember at runtime how we've
distributed work from first loop and reuse that in following loops with
the same number of iterations and schedule, or just force static schedule.
OpenMP API calls etc. aren't allowed in order(concurrent) bodies so
users can't observe it easily anyway. */
if (order_reproducible)
fd->sched_kind = OMP_CLAUSE_SCHEDULE_STATIC;
if (fd->collapse > 1 || fd->tiling)
fd->loops = loops;
else
fd->loops = &fd->loop;
if (fd->ordered && fd->collapse == 1 && loops != NULL)
{
fd->loops = loops;
iterv = NULL_TREE;
countv = NULL_TREE;
collapse_iter = &iterv;
collapse_count = &countv;
}
/* FIXME: for now map schedule(auto) to schedule(static).
There should be analysis to determine whether all iterations
are approximately the same amount of work (then schedule(static)
is best) or if it varies (then schedule(dynamic,N) is better). */
if (fd->sched_kind == OMP_CLAUSE_SCHEDULE_AUTO)
{
fd->sched_kind = OMP_CLAUSE_SCHEDULE_STATIC;
gcc_assert (fd->chunk_size == NULL);
}
gcc_assert ((fd->collapse == 1 && !fd->tiling) || collapse_iter != NULL);
if (taskloop)
fd->sched_kind = OMP_CLAUSE_SCHEDULE_RUNTIME;
if (fd->sched_kind == OMP_CLAUSE_SCHEDULE_RUNTIME)
gcc_assert (fd->chunk_size == NULL);
else if (fd->chunk_size == NULL)
{
/* We only need to compute a default chunk size for ordered
static loops and dynamic loops. */
if (fd->sched_kind != OMP_CLAUSE_SCHEDULE_STATIC
|| fd->have_ordered)
fd->chunk_size = (fd->sched_kind == OMP_CLAUSE_SCHEDULE_STATIC)
? integer_zero_node : integer_one_node;
}
int cnt = fd->ordered ? fd->ordered : fd->collapse;
int single_nonrect = -1;
tree single_nonrect_count = NULL_TREE;
enum tree_code single_nonrect_cond_code = ERROR_MARK;
for (i = 1; i < cnt; i++)
{
tree n1 = gimple_omp_for_initial (for_stmt, i);
tree n2 = gimple_omp_for_final (for_stmt, i);
if (TREE_CODE (n1) == TREE_VEC)
{
if (fd->non_rect)
{
single_nonrect = -1;
break;
}
for (int j = i - 1; j >= 0; j--)
if (TREE_VEC_ELT (n1, 0) == gimple_omp_for_index (for_stmt, j))
{
single_nonrect = j;
break;
}
fd->non_rect = true;
}
else if (TREE_CODE (n2) == TREE_VEC)
{
if (fd->non_rect)
{
single_nonrect = -1;
break;
}
for (int j = i - 1; j >= 0; j--)
if (TREE_VEC_ELT (n2, 0) == gimple_omp_for_index (for_stmt, j))
{
single_nonrect = j;
break;
}
fd->non_rect = true;
}
}
for (i = 0; i < cnt; i++)
{
if (i == 0
&& fd->collapse == 1
&& !fd->tiling
&& (fd->ordered == 0 || loops == NULL))
loop = &fd->loop;
else if (loops != NULL)
loop = loops + i;
else
loop = &dummy_loop;
loop->v = gimple_omp_for_index (for_stmt, i);
gcc_assert (SSA_VAR_P (loop->v));
gcc_assert (TREE_CODE (TREE_TYPE (loop->v)) == INTEGER_TYPE
|| TREE_CODE (TREE_TYPE (loop->v)) == BITINT_TYPE
|| TREE_CODE (TREE_TYPE (loop->v)) == POINTER_TYPE);
var = TREE_CODE (loop->v) == SSA_NAME ? SSA_NAME_VAR (loop->v) : loop->v;
loop->n1 = gimple_omp_for_initial (for_stmt, i);
loop->m1 = NULL_TREE;
loop->m2 = NULL_TREE;
loop->outer = 0;
loop->non_rect_referenced = false;
if (TREE_CODE (loop->n1) == TREE_VEC)
{
for (int j = i - 1; j >= 0; j--)
if (TREE_VEC_ELT (loop->n1, 0) == gimple_omp_for_index (for_stmt, j))
{
loop->outer = i - j;
if (loops != NULL)
loops[j].non_rect_referenced = true;
if (fd->first_nonrect == -1 || fd->first_nonrect > j)
fd->first_nonrect = j;
break;
}
gcc_assert (loop->outer);
loop->m1 = TREE_VEC_ELT (loop->n1, 1);
loop->n1 = TREE_VEC_ELT (loop->n1, 2);
fd->non_rect = true;
fd->last_nonrect = i;
}
loop->cond_code = gimple_omp_for_cond (for_stmt, i);
loop->n2 = gimple_omp_for_final (for_stmt, i);
gcc_assert (loop->cond_code != NE_EXPR
|| (gimple_omp_for_kind (for_stmt)
!= GF_OMP_FOR_KIND_OACC_LOOP));
if (TREE_CODE (loop->n2) == TREE_VEC)
{
if (loop->outer)
gcc_assert (TREE_VEC_ELT (loop->n2, 0)
== gimple_omp_for_index (for_stmt, i - loop->outer));
else
for (int j = i - 1; j >= 0; j--)
if (TREE_VEC_ELT (loop->n2, 0) == gimple_omp_for_index (for_stmt, j))
{
loop->outer = i - j;
if (loops != NULL)
loops[j].non_rect_referenced = true;
if (fd->first_nonrect == -1 || fd->first_nonrect > j)
fd->first_nonrect = j;
break;
}
gcc_assert (loop->outer);
loop->m2 = TREE_VEC_ELT (loop->n2, 1);
loop->n2 = TREE_VEC_ELT (loop->n2, 2);
fd->non_rect = true;
fd->last_nonrect = i;
}
t = gimple_omp_for_incr (for_stmt, i);
gcc_assert (TREE_OPERAND (t, 0) == var);
loop->step = omp_get_for_step_from_incr (loc, t);
omp_adjust_for_condition (loc, &loop->cond_code, &loop->n2, loop->v,
loop->step);
if (simd
|| (fd->sched_kind == OMP_CLAUSE_SCHEDULE_STATIC
&& !fd->have_ordered))
{
if (fd->collapse == 1 && !fd->tiling)
iter_type = TREE_TYPE (loop->v);
else if (i == 0
|| TYPE_PRECISION (iter_type)
< TYPE_PRECISION (TREE_TYPE (loop->v)))
{
if (TREE_CODE (iter_type) == BITINT_TYPE
|| TREE_CODE (TREE_TYPE (loop->v)) == BITINT_TYPE)
iter_type
= build_bitint_type (TYPE_PRECISION (TREE_TYPE (loop->v)),
1);
else
iter_type
= build_nonstandard_integer_type
(TYPE_PRECISION (TREE_TYPE (loop->v)), 1);
}
}
else if (iter_type != long_long_unsigned_type_node)
{
if (POINTER_TYPE_P (TREE_TYPE (loop->v)))
iter_type = long_long_unsigned_type_node;
else if (TYPE_UNSIGNED (TREE_TYPE (loop->v))
&& TYPE_PRECISION (TREE_TYPE (loop->v))
>= TYPE_PRECISION (iter_type))
{
tree n;
if (loop->cond_code == LT_EXPR)
n = fold_build2_loc (loc, PLUS_EXPR, TREE_TYPE (loop->v),
loop->n2, loop->step);
else
n = loop->n1;
if (loop->m1
|| loop->m2
|| TREE_CODE (n) != INTEGER_CST
|| tree_int_cst_lt (TYPE_MAX_VALUE (iter_type), n))
iter_type = long_long_unsigned_type_node;
}
else if (TYPE_PRECISION (TREE_TYPE (loop->v))
> TYPE_PRECISION (iter_type))
{
tree n1, n2;
if (loop->cond_code == LT_EXPR)
{
n1 = loop->n1;
n2 = fold_build2_loc (loc, PLUS_EXPR, TREE_TYPE (loop->v),
loop->n2, loop->step);
}
else
{
n1 = fold_build2_loc (loc, MINUS_EXPR, TREE_TYPE (loop->v),
loop->n2, loop->step);
n2 = loop->n1;
}
if (loop->m1
|| loop->m2
|| TREE_CODE (n1) != INTEGER_CST
|| TREE_CODE (n2) != INTEGER_CST
|| !tree_int_cst_lt (TYPE_MIN_VALUE (iter_type), n1)
|| !tree_int_cst_lt (n2, TYPE_MAX_VALUE (iter_type)))
iter_type = long_long_unsigned_type_node;
}
}
if (i >= fd->collapse)
continue;
if (collapse_count && *collapse_count == NULL)
{
if (count && integer_zerop (count))
continue;
tree n1first = NULL_TREE, n2first = NULL_TREE;
tree n1last = NULL_TREE, n2last = NULL_TREE;
tree ostep = NULL_TREE;
if (loop->m1 || loop->m2)
{
if (count == NULL_TREE)
continue;
if (single_nonrect == -1
|| (loop->m1 && TREE_CODE (loop->m1) != INTEGER_CST)
|| (loop->m2 && TREE_CODE (loop->m2) != INTEGER_CST)
|| TREE_CODE (loop->n1) != INTEGER_CST
|| TREE_CODE (loop->n2) != INTEGER_CST
|| TREE_CODE (loop->step) != INTEGER_CST)
{
count = NULL_TREE;
continue;
}
tree var = gimple_omp_for_initial (for_stmt, single_nonrect);
tree itype = TREE_TYPE (var);
tree first = gimple_omp_for_initial (for_stmt, single_nonrect);
t = gimple_omp_for_incr (for_stmt, single_nonrect);
ostep = omp_get_for_step_from_incr (loc, t);
t = fold_binary (MINUS_EXPR, long_long_unsigned_type_node,
single_nonrect_count,
build_one_cst (long_long_unsigned_type_node));
t = fold_convert (itype, t);
first = fold_convert (itype, first);
ostep = fold_convert (itype, ostep);
tree last = fold_binary (PLUS_EXPR, itype, first,
fold_binary (MULT_EXPR, itype, t,
ostep));
if (TREE_CODE (first) != INTEGER_CST
|| TREE_CODE (last) != INTEGER_CST)
{
count = NULL_TREE;
continue;
}
if (loop->m1)
{
tree m1 = fold_convert (itype, loop->m1);
tree n1 = fold_convert (itype, loop->n1);
n1first = fold_binary (PLUS_EXPR, itype,
fold_binary (MULT_EXPR, itype,
first, m1), n1);
n1last = fold_binary (PLUS_EXPR, itype,
fold_binary (MULT_EXPR, itype,
last, m1), n1);
}
else
n1first = n1last = loop->n1;
if (loop->m2)
{
tree n2 = fold_convert (itype, loop->n2);
tree m2 = fold_convert (itype, loop->m2);
n2first = fold_binary (PLUS_EXPR, itype,
fold_binary (MULT_EXPR, itype,
first, m2), n2);
n2last = fold_binary (PLUS_EXPR, itype,
fold_binary (MULT_EXPR, itype,
last, m2), n2);
}
else
n2first = n2last = loop->n2;
n1first = fold_convert (TREE_TYPE (loop->v), n1first);
n2first = fold_convert (TREE_TYPE (loop->v), n2first);
n1last = fold_convert (TREE_TYPE (loop->v), n1last);
n2last = fold_convert (TREE_TYPE (loop->v), n2last);
t = fold_binary (loop->cond_code, boolean_type_node,
n1first, n2first);
tree t2 = fold_binary (loop->cond_code, boolean_type_node,
n1last, n2last);
if (t && t2 && integer_nonzerop (t) && integer_nonzerop (t2))
/* All outer loop iterators have at least one inner loop
iteration. Try to compute the count at compile time. */
t = NULL_TREE;
else if (t && t2 && integer_zerop (t) && integer_zerop (t2))
/* No iterations of the inner loop. count will be set to
zero cst below. */;
else if (TYPE_UNSIGNED (itype)
|| t == NULL_TREE
|| t2 == NULL_TREE
|| TREE_CODE (t) != INTEGER_CST
|| TREE_CODE (t2) != INTEGER_CST)
{
/* Punt (for now). */
count = NULL_TREE;
continue;
}
else
{
/* Some iterations of the outer loop have zero iterations
of the inner loop, while others have at least one.
In this case, we need to adjust one of those outer
loop bounds. If ADJ_FIRST, we need to adjust outer n1
(first), otherwise outer n2 (last). */
bool adj_first = integer_zerop (t);
tree n1 = fold_convert (itype, loop->n1);
tree n2 = fold_convert (itype, loop->n2);
tree m1 = loop->m1 ? fold_convert (itype, loop->m1)
: build_zero_cst (itype);
tree m2 = loop->m2 ? fold_convert (itype, loop->m2)
: build_zero_cst (itype);
t = fold_binary (MINUS_EXPR, itype, n1, n2);
t2 = fold_binary (MINUS_EXPR, itype, m2, m1);
t = fold_binary (TRUNC_DIV_EXPR, itype, t, t2);
t2 = fold_binary (MINUS_EXPR, itype, t, first);
t2 = fold_binary (TRUNC_MOD_EXPR, itype, t2, ostep);
t = fold_binary (MINUS_EXPR, itype, t, t2);
tree n1cur
= fold_binary (PLUS_EXPR, itype, n1,
fold_binary (MULT_EXPR, itype, m1, t));
tree n2cur
= fold_binary (PLUS_EXPR, itype, n2,
fold_binary (MULT_EXPR, itype, m2, t));
t2 = fold_binary (loop->cond_code, boolean_type_node,
n1cur, n2cur);
tree t3 = fold_binary (MULT_EXPR, itype, m1, ostep);
tree t4 = fold_binary (MULT_EXPR, itype, m2, ostep);
tree diff;
if (adj_first)
{
tree new_first;
if (integer_nonzerop (t2))
{
new_first = t;
n1first = n1cur;
n2first = n2cur;
if (flag_checking)
{
t3 = fold_binary (MINUS_EXPR, itype, n1cur, t3);
t4 = fold_binary (MINUS_EXPR, itype, n2cur, t4);
t3 = fold_binary (loop->cond_code,
boolean_type_node, t3, t4);
gcc_assert (integer_zerop (t3));
}
}
else
{
t3 = fold_binary (PLUS_EXPR, itype, n1cur, t3);
t4 = fold_binary (PLUS_EXPR, itype, n2cur, t4);
new_first = fold_binary (PLUS_EXPR, itype, t, ostep);
n1first = t3;
n2first = t4;
if (flag_checking)
{
t3 = fold_binary (loop->cond_code,
boolean_type_node, t3, t4);
gcc_assert (integer_nonzerop (t3));
}
}
diff = fold_binary (MINUS_EXPR, itype, new_first, first);
first = new_first;
fd->adjn1 = first;
}
else
{
tree new_last;
if (integer_zerop (t2))
{
t3 = fold_binary (MINUS_EXPR, itype, n1cur, t3);
t4 = fold_binary (MINUS_EXPR, itype, n2cur, t4);
new_last = fold_binary (MINUS_EXPR, itype, t, ostep);
n1last = t3;
n2last = t4;
if (flag_checking)
{
t3 = fold_binary (loop->cond_code,
boolean_type_node, t3, t4);
gcc_assert (integer_nonzerop (t3));
}
}
else
{
new_last = t;
n1last = n1cur;
n2last = n2cur;
if (flag_checking)
{
t3 = fold_binary (PLUS_EXPR, itype, n1cur, t3);
t4 = fold_binary (PLUS_EXPR, itype, n2cur, t4);
t3 = fold_binary (loop->cond_code,
boolean_type_node, t3, t4);
gcc_assert (integer_zerop (t3));
}
}
diff = fold_binary (MINUS_EXPR, itype, last, new_last);
}
if (TYPE_UNSIGNED (itype)
&& single_nonrect_cond_code == GT_EXPR)
diff = fold_binary (TRUNC_DIV_EXPR, itype,
fold_unary (NEGATE_EXPR, itype, diff),
fold_unary (NEGATE_EXPR, itype,
ostep));
else
diff = fold_binary (TRUNC_DIV_EXPR, itype, diff, ostep);
diff = fold_convert (long_long_unsigned_type_node, diff);
single_nonrect_count
= fold_binary (MINUS_EXPR, long_long_unsigned_type_node,
single_nonrect_count, diff);
t = NULL_TREE;
}
}
else
t = fold_binary (loop->cond_code, boolean_type_node,
fold_convert (TREE_TYPE (loop->v), loop->n1),
fold_convert (TREE_TYPE (loop->v), loop->n2));
if (t && integer_zerop (t))
count = build_zero_cst (long_long_unsigned_type_node);
else if ((i == 0 || count != NULL_TREE)
&& (TREE_CODE (TREE_TYPE (loop->v)) == INTEGER_TYPE
|| TREE_CODE (TREE_TYPE (loop->v)) == BITINT_TYPE)
&& TREE_CONSTANT (loop->n1)
&& TREE_CONSTANT (loop->n2)
&& TREE_CODE (loop->step) == INTEGER_CST)
{
tree itype = TREE_TYPE (loop->v);
if (POINTER_TYPE_P (itype))
itype = signed_type_for (itype);
t = build_int_cst (itype, (loop->cond_code == LT_EXPR ? -1 : 1));
t = fold_build2 (PLUS_EXPR, itype,
fold_convert (itype, loop->step), t);
tree n1 = loop->n1;
tree n2 = loop->n2;
if (loop->m1 || loop->m2)
{
gcc_assert (single_nonrect != -1);
n1 = n1first;
n2 = n2first;
}
t = fold_build2 (PLUS_EXPR, itype, t, fold_convert (itype, n2));
t = fold_build2 (MINUS_EXPR, itype, t, fold_convert (itype, n1));
tree step = fold_convert_loc (loc, itype, loop->step);
if (TYPE_UNSIGNED (itype) && loop->cond_code == GT_EXPR)
t = fold_build2 (TRUNC_DIV_EXPR, itype,
fold_build1 (NEGATE_EXPR, itype, t),
fold_build1 (NEGATE_EXPR, itype, step));
else
t = fold_build2 (TRUNC_DIV_EXPR, itype, t, step);
tree llutype = long_long_unsigned_type_node;
t = fold_convert (llutype, t);
if (loop->m1 || loop->m2)
{
/* t is number of iterations of inner loop at either first
or last value of the outer iterator (the one with fewer
iterations).
Compute t2 = ((m2 - m1) * ostep) / step
and niters = outer_count * t
+ t2 * ((outer_count - 1) * outer_count / 2)
*/
tree m1 = loop->m1 ? loop->m1 : integer_zero_node;
tree m2 = loop->m2 ? loop->m2 : integer_zero_node;
m1 = fold_convert (itype, m1);
m2 = fold_convert (itype, m2);
tree t2 = fold_build2 (MINUS_EXPR, itype, m2, m1);
t2 = fold_build2 (MULT_EXPR, itype, t2, ostep);
if (TYPE_UNSIGNED (itype) && loop->cond_code == GT_EXPR)
t2 = fold_build2 (TRUNC_DIV_EXPR, itype,
fold_build1 (NEGATE_EXPR, itype, t2),
fold_build1 (NEGATE_EXPR, itype, step));
else
t2 = fold_build2 (TRUNC_DIV_EXPR, itype, t2, step);
t2 = fold_convert (llutype, t2);
fd->first_inner_iterations = t;
fd->factor = t2;
t = fold_build2 (MULT_EXPR, llutype, t,
single_nonrect_count);
tree t3 = fold_build2 (MINUS_EXPR, llutype,
single_nonrect_count,
build_one_cst (llutype));
t3 = fold_build2 (MULT_EXPR, llutype, t3,
single_nonrect_count);
t3 = fold_build2 (TRUNC_DIV_EXPR, llutype, t3,
build_int_cst (llutype, 2));
t2 = fold_build2 (MULT_EXPR, llutype, t2, t3);
t = fold_build2 (PLUS_EXPR, llutype, t, t2);
}
if (i == single_nonrect)
{
if (integer_zerop (t) || TREE_CODE (t) != INTEGER_CST)
count = t;
else
{
single_nonrect_count = t;
single_nonrect_cond_code = loop->cond_code;
if (count == NULL_TREE)
count = build_one_cst (llutype);
}
}
else if (count != NULL_TREE)
count = fold_build2 (MULT_EXPR, llutype, count, t);
else
count = t;
if (TREE_CODE (count) != INTEGER_CST)
count = NULL_TREE;
}
else if (count && !integer_zerop (count))
count = NULL_TREE;
}
}
if (count
&& !simd
&& (fd->sched_kind != OMP_CLAUSE_SCHEDULE_STATIC
|| fd->have_ordered))
{
if (!tree_int_cst_lt (count, TYPE_MAX_VALUE (long_integer_type_node)))
iter_type = long_long_unsigned_type_node;
else
iter_type = long_integer_type_node;
}
else if (collapse_iter && *collapse_iter != NULL)
iter_type = TREE_TYPE (*collapse_iter);
fd->iter_type = iter_type;
if (collapse_iter && *collapse_iter == NULL)
*collapse_iter = create_tmp_var (iter_type, ".iter");
if (collapse_count && *collapse_count == NULL)
{
if (count)
{
*collapse_count = fold_convert_loc (loc, iter_type, count);
if (fd->first_inner_iterations && fd->factor)
{
t = make_tree_vec (4);
TREE_VEC_ELT (t, 0) = *collapse_count;
TREE_VEC_ELT (t, 1) = fd->first_inner_iterations;
TREE_VEC_ELT (t, 2) = fd->factor;
TREE_VEC_ELT (t, 3) = fd->adjn1;
*collapse_count = t;
}
}
else
*collapse_count = create_tmp_var (iter_type, ".count");
}
if (fd->collapse > 1 || fd->tiling || (fd->ordered && loops))
{
fd->loop.v = *collapse_iter;
fd->loop.n1 = build_int_cst (TREE_TYPE (fd->loop.v), 0);
fd->loop.n2 = *collapse_count;
if (TREE_CODE (fd->loop.n2) == TREE_VEC)
{
gcc_assert (fd->non_rect);
fd->first_inner_iterations = TREE_VEC_ELT (fd->loop.n2, 1);
fd->factor = TREE_VEC_ELT (fd->loop.n2, 2);
fd->adjn1 = TREE_VEC_ELT (fd->loop.n2, 3);
fd->loop.n2 = TREE_VEC_ELT (fd->loop.n2, 0);
}
fd->loop.step = build_int_cst (TREE_TYPE (fd->loop.v), 1);
fd->loop.m1 = NULL_TREE;
fd->loop.m2 = NULL_TREE;
fd->loop.outer = 0;
fd->loop.cond_code = LT_EXPR;
}
else if (loops)
loops[0] = fd->loop;
}
/* Build a call to GOMP_barrier. */
gimple *
omp_build_barrier (tree lhs)
{
tree fndecl = builtin_decl_explicit (lhs ? BUILT_IN_GOMP_BARRIER_CANCEL
: BUILT_IN_GOMP_BARRIER);
gcall *g = gimple_build_call (fndecl, 0);
if (lhs)
gimple_call_set_lhs (g, lhs);
return g;
}
/* Find OMP_FOR resp. OMP_SIMD with non-NULL OMP_FOR_INIT. Also, fill in pdata
array, pdata[0] non-NULL if there is anything non-trivial in between,
pdata[1] is address of OMP_PARALLEL in between if any, pdata[2] is address
of OMP_FOR in between if any and pdata[3] is address of the inner
OMP_FOR/OMP_SIMD. */
tree
find_combined_omp_for (tree *tp, int *walk_subtrees, void *data)
{
tree **pdata = (tree **) data;
*walk_subtrees = 0;
switch (TREE_CODE (*tp))
{
case OMP_FOR:
if (OMP_FOR_INIT (*tp) != NULL_TREE)
{
pdata[3] = tp;
return *tp;
}
pdata[2] = tp;
*walk_subtrees = 1;
break;
case OMP_SIMD:
if (OMP_FOR_INIT (*tp) != NULL_TREE)
{
pdata[3] = tp;
return *tp;
}
break;
case BIND_EXPR:
if (BIND_EXPR_VARS (*tp)
|| (BIND_EXPR_BLOCK (*tp)
&& BLOCK_VARS (BIND_EXPR_BLOCK (*tp))))
pdata[0] = tp;
*walk_subtrees = 1;
break;
case STATEMENT_LIST:
if (!tsi_one_before_end_p (tsi_start (*tp)))
pdata[0] = tp;
*walk_subtrees = 1;
break;
case TRY_FINALLY_EXPR:
pdata[0] = tp;
*walk_subtrees = 1;
break;
case OMP_PARALLEL:
pdata[1] = tp;
*walk_subtrees = 1;
break;
default:
break;
}
return NULL_TREE;
}
/* Return maximum possible vectorization factor for the target. */
poly_uint64
omp_max_vf (void)
{
if (!optimize
|| optimize_debug
|| !flag_tree_loop_optimize
|| (!flag_tree_loop_vectorize
&& OPTION_SET_P (flag_tree_loop_vectorize)))
return 1;
auto_vector_modes modes;
targetm.vectorize.autovectorize_vector_modes (&modes, true);
if (!modes.is_empty ())
{
poly_uint64 vf = 0;
for (unsigned int i = 0; i < modes.length (); ++i)
/* The returned modes use the smallest element size (and thus
the largest nunits) for the vectorization approach that they
represent. */
vf = ordered_max (vf, GET_MODE_NUNITS (modes[i]));
return vf;
}
machine_mode vqimode = targetm.vectorize.preferred_simd_mode (QImode);
if (GET_MODE_CLASS (vqimode) == MODE_VECTOR_INT)
return GET_MODE_NUNITS (vqimode);
return 1;
}
/* Return maximum SIMT width if offloading may target SIMT hardware. */
int
omp_max_simt_vf (void)
{
if (!optimize)
return 0;
if (ENABLE_OFFLOADING)
for (const char *c = getenv ("OFFLOAD_TARGET_NAMES"); c;)
{
if (startswith (c, "nvptx"))
return 32;
else if ((c = strchr (c, ':')))
c++;
}
return 0;
}
/* Store the construct selectors as tree codes from last to first.
CTX is a list of trait selectors, nconstructs must be equal to its
length, and the array CONSTRUCTS holds the output. */
void
omp_construct_traits_to_codes (tree ctx, int nconstructs,
enum tree_code *constructs)
{
int i = nconstructs - 1;
/* Order must match the OMP_TRAIT_CONSTRUCT_* enumerators in
enum omp_ts_code. */
static enum tree_code code_map[]
= { OMP_TARGET, OMP_TEAMS, OMP_PARALLEL, OMP_FOR, OMP_SIMD };
for (tree ts = ctx; ts; ts = TREE_CHAIN (ts), i--)
{
enum omp_ts_code sel = OMP_TS_CODE (ts);
int j = (int)sel - (int)OMP_TRAIT_CONSTRUCT_TARGET;
gcc_assert (j >= 0 && (unsigned int) j < ARRAY_SIZE (code_map));
constructs[i] = code_map[j];
}
gcc_assert (i == -1);
}
/* Return true if PROP is possibly present in one of the offloading target's
OpenMP contexts. The format of PROPS string is always offloading target's
name terminated by '\0', followed by properties for that offloading
target separated by '\0' and terminated by another '\0'. The strings
are created from omp-device-properties installed files of all configured
offloading targets. */
static bool
omp_offload_device_kind_arch_isa (const char *props, const char *prop)
{
const char *names = getenv ("OFFLOAD_TARGET_NAMES");
if (names == NULL || *names == '\0')
return false;
while (*props != '\0')
{
size_t name_len = strlen (props);
bool matches = false;
for (const char *c = names; c; )
{
if (strncmp (props, c, name_len) == 0
&& (c[name_len] == '\0'
|| c[name_len] == ':'
|| c[name_len] == '='))
{
matches = true;
break;
}
else if ((c = strchr (c, ':')))
c++;
}
props = props + name_len + 1;
while (*props != '\0')
{
if (matches && strcmp (props, prop) == 0)
return true;
props = strchr (props, '\0') + 1;
}
props++;
}
return false;
}
/* Return true if the current code location is or might be offloaded.
Return true in declare target functions, or when nested in a target
region or when unsure, return false otherwise. */
static bool
omp_maybe_offloaded (void)
{
if (!ENABLE_OFFLOADING)
return false;
const char *names = getenv ("OFFLOAD_TARGET_NAMES");
if (names == NULL || *names == '\0')
return false;
if (symtab->state == PARSING)
/* Maybe. */
return true;
if (cfun && cfun->after_inlining)
return false;
if (current_function_decl
&& lookup_attribute ("omp declare target",
DECL_ATTRIBUTES (current_function_decl)))
return true;
if (cfun && (cfun->curr_properties & PROP_gimple_any) == 0)
{
enum tree_code construct = OMP_TARGET;
if (omp_construct_selector_matches (&construct, 1, NULL))
return true;
}
return false;
}
/* Lookup tables for context selectors. */
const char *omp_tss_map[] =
{
"construct",
"device",
"target_device",
"implementation",
"user",
NULL
};
/* Arrays of property candidates must be null-terminated. */
static const char *const kind_properties[] =
{ "host", "nohost", "cpu", "gpu", "fpga", "any", NULL };
static const char *const vendor_properties[] =
{ "amd", "arm", "bsc", "cray", "fujitsu", "gnu", "hpe", "ibm", "intel",
"llvm", "nvidia", "pgi", "ti", "unknown", NULL };
static const char *const extension_properties[] =
{ NULL };
static const char *const atomic_default_mem_order_properties[] =
{ "seq_cst", "relaxed", "acq_rel", "acquire", "release", NULL };
struct omp_ts_info omp_ts_map[] =
{
{ "kind",
(1 << OMP_TRAIT_SET_DEVICE) | (1 << OMP_TRAIT_SET_TARGET_DEVICE),
OMP_TRAIT_PROPERTY_NAME_LIST, false,
kind_properties
},
{ "isa",
(1 << OMP_TRAIT_SET_DEVICE) | (1 << OMP_TRAIT_SET_TARGET_DEVICE),
OMP_TRAIT_PROPERTY_NAME_LIST, false,
NULL
},
{ "arch",
(1 << OMP_TRAIT_SET_DEVICE) | (1 << OMP_TRAIT_SET_TARGET_DEVICE),
OMP_TRAIT_PROPERTY_NAME_LIST, false,
NULL
},
{ "device_num",
(1 << OMP_TRAIT_SET_TARGET_DEVICE),
OMP_TRAIT_PROPERTY_DEV_NUM_EXPR, false,
NULL
},
{ "vendor",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_NAME_LIST, true,
vendor_properties,
},
{ "extension",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_NAME_LIST, true,
extension_properties,
},
{ "atomic_default_mem_order",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_ID, true,
atomic_default_mem_order_properties,
},
{ "requires",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_CLAUSE_LIST, true,
NULL
},
{ "unified_address",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_NONE, true,
NULL
},
{ "unified_shared_memory",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_NONE, true,
NULL
},
{ "dynamic_allocators",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_NONE, true,
NULL
},
{ "reverse_offload",
(1 << OMP_TRAIT_SET_IMPLEMENTATION),
OMP_TRAIT_PROPERTY_NONE, true,
NULL
},
{ "condition",
(1 << OMP_TRAIT_SET_USER),
OMP_TRAIT_PROPERTY_BOOL_EXPR, true,
NULL
},
{ "target",
(1 << OMP_TRAIT_SET_CONSTRUCT),
OMP_TRAIT_PROPERTY_NONE, false,
NULL
},
{ "teams",
(1 << OMP_TRAIT_SET_CONSTRUCT),
OMP_TRAIT_PROPERTY_NONE, false,
NULL
},
{ "parallel",
(1 << OMP_TRAIT_SET_CONSTRUCT),
OMP_TRAIT_PROPERTY_NONE, false,
NULL
},
{ "for",
(1 << OMP_TRAIT_SET_CONSTRUCT),
OMP_TRAIT_PROPERTY_NONE, false,
NULL
},
{ "simd",
(1 << OMP_TRAIT_SET_CONSTRUCT),
OMP_TRAIT_PROPERTY_CLAUSE_LIST, false,
NULL
},
{ NULL, 0, OMP_TRAIT_PROPERTY_NONE, false, NULL } /* OMP_TRAIT_LAST */
};
/* Return a name from PROP, a property in selectors accepting
name lists. */
const char *
omp_context_name_list_prop (tree prop)
{
gcc_assert (OMP_TP_NAME (prop) == OMP_TP_NAMELIST_NODE);
tree val = OMP_TP_VALUE (prop);
switch (TREE_CODE (val))
{
case IDENTIFIER_NODE:
return IDENTIFIER_POINTER (val);
case STRING_CST:
{
const char *ret = TREE_STRING_POINTER (val);
if ((size_t) TREE_STRING_LENGTH (val)
== strlen (ret) + (lang_GNU_Fortran () ? 0 : 1))
return ret;
return NULL;
}
default:
return NULL;
}
}
/* Diagnose errors in an OpenMP context selector, return CTX if
it is correct or error_mark_node otherwise. */
tree
omp_check_context_selector (location_t loc, tree ctx)
{
bool tss_seen[OMP_TRAIT_SET_LAST], ts_seen[OMP_TRAIT_LAST];
memset (tss_seen, 0, sizeof (tss_seen));
for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
{
enum omp_tss_code tss_code = OMP_TSS_CODE (tss);
/* We can parse this, but not handle it yet. */
if (tss_code == OMP_TRAIT_SET_TARGET_DEVICE)
sorry_at (loc, "%<target_device%> selector set is not supported yet");
/* Each trait-set-selector-name can only be specified once. */
if (tss_seen[tss_code])
{
error_at (loc, "selector set %qs specified more than once",
OMP_TSS_NAME (tss));
return error_mark_node;
}
else
tss_seen[tss_code] = true;
memset (ts_seen, 0, sizeof (ts_seen));
for (tree ts = OMP_TSS_TRAIT_SELECTORS (tss); ts; ts = TREE_CHAIN (ts))
{
enum omp_ts_code ts_code = OMP_TS_CODE (ts);
/* Ignore unknown traits. */
if (ts_code == OMP_TRAIT_INVALID)
continue;
/* Each trait-selector-name can only be specified once. */
if (ts_seen[ts_code])
{
error_at (loc,
"selector %qs specified more than once in set %qs",
OMP_TS_NAME (ts),
OMP_TSS_NAME (tss));
return error_mark_node;
}
else
ts_seen[ts_code] = true;
if (omp_ts_map[ts_code].valid_properties == NULL)
continue;
for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
for (unsigned j = 0; ; j++)
{
const char *candidate
= omp_ts_map[ts_code].valid_properties[j];
if (candidate == NULL)
{
/* We've reached the end of the candidate array. */
if (ts_code == OMP_TRAIT_IMPLEMENTATION_ADMO)
/* FIXME: not sure why this is an error vs warnings
for the others, + incorrect/unknown wording? */
{
error_at (loc,
"incorrect property %qs of %qs selector",
IDENTIFIER_POINTER (OMP_TP_NAME (p)),
"atomic_default_mem_order");
return error_mark_node;
}
if (OMP_TP_NAME (p) == OMP_TP_NAMELIST_NODE
&& (TREE_CODE (OMP_TP_VALUE (p)) == STRING_CST))
warning_at (loc, OPT_Wopenmp,
"unknown property %qE of %qs selector",
OMP_TP_VALUE (p),
OMP_TS_NAME (ts));
else if (OMP_TP_NAME (p) == OMP_TP_NAMELIST_NODE)
warning_at (loc, OPT_Wopenmp,
"unknown property %qs of %qs selector",
omp_context_name_list_prop (p),
OMP_TS_NAME (ts));
else if (OMP_TP_NAME (p))
warning_at (loc, OPT_Wopenmp,
"unknown property %qs of %qs selector",
IDENTIFIER_POINTER (OMP_TP_NAME (p)),
OMP_TS_NAME (ts));
break;
}
else if (OMP_TP_NAME (p) == OMP_TP_NAMELIST_NODE)
/* Property-list traits. */
{
const char *str = omp_context_name_list_prop (p);
if (str && !strcmp (str, candidate))
break;
}
else if (!strcmp (IDENTIFIER_POINTER (OMP_TP_NAME (p)),
candidate))
/* Identifier traits. */
break;
}
}
}
return ctx;
}
/* Register VARIANT as variant of some base function marked with
#pragma omp declare variant. CONSTRUCT is corresponding list of
trait-selectors for the construct selector set. This is stashed as the
value of the "omp declare variant variant" attribute on VARIANT. */
void
omp_mark_declare_variant (location_t loc, tree variant, tree construct)
{
/* Ignore this variant if it contains unknown construct selectors.
It will never match, and the front ends have already issued a warning
about it. */
for (tree c = construct; c; c = TREE_CHAIN (c))
if (OMP_TS_CODE (c) == OMP_TRAIT_INVALID)
return;
tree attr = lookup_attribute ("omp declare variant variant",
DECL_ATTRIBUTES (variant));
if (attr == NULL_TREE)
{
attr = tree_cons (get_identifier ("omp declare variant variant"),
unshare_expr (construct),
DECL_ATTRIBUTES (variant));
DECL_ATTRIBUTES (variant) = attr;
return;
}
if ((TREE_VALUE (attr) != NULL_TREE) != (construct != NULL_TREE)
|| (construct != NULL_TREE
&& omp_context_selector_set_compare (OMP_TRAIT_SET_CONSTRUCT,
TREE_VALUE (attr),
construct)))
error_at (loc, "%qD used as a variant with incompatible %<construct%> "
"selector sets", variant);
}
/* Constructors for context selectors. */
tree
make_trait_set_selector (enum omp_tss_code code, tree selectors, tree chain)
{
return tree_cons (build_int_cst (integer_type_node, code),
selectors, chain);
}
tree
make_trait_selector (enum omp_ts_code code, tree score, tree properties,
tree chain)
{
if (score == NULL_TREE)
return tree_cons (build_int_cst (integer_type_node, code),
properties, chain);
else
return tree_cons (build_int_cst (integer_type_node, code),
tree_cons (OMP_TS_SCORE_NODE, score, properties),
chain);
}
tree
make_trait_property (tree name, tree value, tree chain)
{
return tree_cons (name, value, chain);
}
/* Return 1 if context selector matches the current OpenMP context, 0
if it does not and -1 if it is unknown and need to be determined later.
Some properties can be checked right away during parsing (this routine),
others need to wait until the whole TU is parsed, others need to wait until
IPA, others until vectorization. */
int
omp_context_selector_matches (tree ctx)
{
int ret = 1;
for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
{
enum omp_tss_code set = OMP_TSS_CODE (tss);
tree selectors = OMP_TSS_TRAIT_SELECTORS (tss);
/* Immediately reject the match if there are any ignored
selectors present. */
for (tree ts = selectors; ts; ts = TREE_CHAIN (ts))
if (OMP_TS_CODE (ts) == OMP_TRAIT_INVALID)
return 0;
if (set == OMP_TRAIT_SET_CONSTRUCT)
{
/* For now, ignore the construct set. While something can be
determined already during parsing, we don't know until end of TU
whether additional constructs aren't added through declare variant
unless "omp declare variant variant" attribute exists already
(so in most of the cases), and we'd need to maintain set of
surrounding OpenMP constructs, which is better handled during
gimplification. */
if (symtab->state == PARSING)
{
ret = -1;
continue;
}
int nconstructs = list_length (selectors);
enum tree_code *constructs = NULL;
if (nconstructs)
{
/* Even though this alloca appears in a loop over selector
sets, it does not repeatedly grow the stack, because
there can be only one construct selector set specified.
This is enforced by omp_check_context_selector. */
constructs
= (enum tree_code *) alloca (nconstructs
* sizeof (enum tree_code));
omp_construct_traits_to_codes (selectors, nconstructs,
constructs);
}
if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
{
if (!cfun->after_inlining)
{
ret = -1;
continue;
}
int i;
for (i = 0; i < nconstructs; ++i)
if (constructs[i] == OMP_SIMD)
break;
if (i < nconstructs)
{
ret = -1;
continue;
}
/* If there is no simd, assume it is ok after IPA,
constructs should have been checked before. */
continue;
}
int r = omp_construct_selector_matches (constructs, nconstructs,
NULL);
if (r == 0)
return 0;
if (r == -1)
ret = -1;
continue;
}
for (tree ts = selectors; ts; ts = TREE_CHAIN (ts))
{
enum omp_ts_code sel = OMP_TS_CODE (ts);
switch (sel)
{
case OMP_TRAIT_IMPLEMENTATION_VENDOR:
if (set == OMP_TRAIT_SET_IMPLEMENTATION)
for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
{
const char *prop = omp_context_name_list_prop (p);
if (prop == NULL)
return 0;
if (!strcmp (prop, "gnu"))
continue;
return 0;
}
break;
case OMP_TRAIT_IMPLEMENTATION_EXTENSION:
if (set == OMP_TRAIT_SET_IMPLEMENTATION)
/* We don't support any extensions right now. */
return 0;
break;
case OMP_TRAIT_IMPLEMENTATION_ADMO:
if (set == OMP_TRAIT_SET_IMPLEMENTATION)
{
if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
break;
enum omp_memory_order omo
= ((enum omp_memory_order)
(omp_requires_mask
& OMP_REQUIRES_ATOMIC_DEFAULT_MEM_ORDER));
if (omo == OMP_MEMORY_ORDER_UNSPECIFIED)
{
/* We don't know yet, until end of TU. */
if (symtab->state == PARSING)
{
ret = -1;
break;
}
else
omo = OMP_MEMORY_ORDER_RELAXED;
}
tree p = OMP_TS_PROPERTIES (ts);
const char *prop = IDENTIFIER_POINTER (OMP_TP_NAME (p));
if (!strcmp (prop, "relaxed")
&& omo != OMP_MEMORY_ORDER_RELAXED)
return 0;
else if (!strcmp (prop, "seq_cst")
&& omo != OMP_MEMORY_ORDER_SEQ_CST)
return 0;
else if (!strcmp (prop, "acq_rel")
&& omo != OMP_MEMORY_ORDER_ACQ_REL)
return 0;
else if (!strcmp (prop, "acquire")
&& omo != OMP_MEMORY_ORDER_ACQUIRE)
return 0;
else if (!strcmp (prop, "release")
&& omo != OMP_MEMORY_ORDER_RELEASE)
return 0;
}
break;
case OMP_TRAIT_DEVICE_ARCH:
if (set == OMP_TRAIT_SET_DEVICE)
for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
{
const char *arch = omp_context_name_list_prop (p);
if (arch == NULL)
return 0;
int r = 0;
if (targetm.omp.device_kind_arch_isa != NULL)
r = targetm.omp.device_kind_arch_isa (omp_device_arch,
arch);
if (r == 0 || (r == -1 && symtab->state != PARSING))
{
/* If we are or might be in a target region or
declare target function, need to take into account
also offloading values. */
if (!omp_maybe_offloaded ())
return 0;
if (ENABLE_OFFLOADING)
{
const char *arches = omp_offload_device_arch;
if (omp_offload_device_kind_arch_isa (arches,
arch))
{
ret = -1;
continue;
}
}
return 0;
}
else if (r == -1)
ret = -1;
/* If arch matches on the host, it still might not match
in the offloading region. */
else if (omp_maybe_offloaded ())
ret = -1;
}
break;
case OMP_TRAIT_IMPLEMENTATION_UNIFIED_ADDRESS:
if (set == OMP_TRAIT_SET_IMPLEMENTATION)
{
if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
break;
if ((omp_requires_mask & OMP_REQUIRES_UNIFIED_ADDRESS) == 0)
{
if (symtab->state == PARSING)
ret = -1;
else
return 0;
}
}
break;
case OMP_TRAIT_IMPLEMENTATION_UNIFIED_SHARED_MEMORY:
if (set == OMP_TRAIT_SET_IMPLEMENTATION)
{
if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
break;
if ((omp_requires_mask
& OMP_REQUIRES_UNIFIED_SHARED_MEMORY) == 0)
{
if (symtab->state == PARSING)
ret = -1;
else
return 0;
}
}
break;
case OMP_TRAIT_IMPLEMENTATION_DYNAMIC_ALLOCATORS:
if (set == OMP_TRAIT_SET_IMPLEMENTATION)
{
if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
break;
if ((omp_requires_mask
& OMP_REQUIRES_DYNAMIC_ALLOCATORS) == 0)
{
if (symtab->state == PARSING)
ret = -1;
else
return 0;
}
}
break;
case OMP_TRAIT_IMPLEMENTATION_REVERSE_OFFLOAD:
if (set == OMP_TRAIT_SET_IMPLEMENTATION)
{
if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
break;
if ((omp_requires_mask & OMP_REQUIRES_REVERSE_OFFLOAD) == 0)
{
if (symtab->state == PARSING)
ret = -1;
else
return 0;
}
}
break;
case OMP_TRAIT_DEVICE_KIND:
if (set == OMP_TRAIT_SET_DEVICE)
for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
{
const char *prop = omp_context_name_list_prop (p);
if (prop == NULL)
return 0;
if (!strcmp (prop, "any"))
continue;
if (!strcmp (prop, "host"))
{
#ifdef ACCEL_COMPILER
return 0;
#else
if (omp_maybe_offloaded ())
ret = -1;
continue;
#endif
}
if (!strcmp (prop, "nohost"))
{
#ifndef ACCEL_COMPILER
if (omp_maybe_offloaded ())
ret = -1;
else
return 0;
#endif
continue;
}
int r = 0;
if (targetm.omp.device_kind_arch_isa != NULL)
r = targetm.omp.device_kind_arch_isa (omp_device_kind,
prop);
else
r = strcmp (prop, "cpu") == 0;
if (r == 0 || (r == -1 && symtab->state != PARSING))
{
/* If we are or might be in a target region or
declare target function, need to take into account
also offloading values. */
if (!omp_maybe_offloaded ())
return 0;
if (ENABLE_OFFLOADING)
{
const char *kinds = omp_offload_device_kind;
if (omp_offload_device_kind_arch_isa (kinds, prop))
{
ret = -1;
continue;
}
}
return 0;
}
else if (r == -1)
ret = -1;
/* If kind matches on the host, it still might not match
in the offloading region. */
else if (omp_maybe_offloaded ())
ret = -1;
}
break;
case OMP_TRAIT_DEVICE_ISA:
if (set == OMP_TRAIT_SET_DEVICE)
for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
{
const char *isa = omp_context_name_list_prop (p);
if (isa == NULL)
return 0;
int r = 0;
if (targetm.omp.device_kind_arch_isa != NULL)
r = targetm.omp.device_kind_arch_isa (omp_device_isa,
isa);
if (r == 0 || (r == -1 && symtab->state != PARSING))
{
/* If isa is valid on the target, but not in the
current function and current function has
#pragma omp declare simd on it, some simd clones
might have the isa added later on. */
if (r == -1
&& targetm.simd_clone.compute_vecsize_and_simdlen
&& (cfun == NULL || !cfun->after_inlining))
{
tree attrs
= DECL_ATTRIBUTES (current_function_decl);
if (lookup_attribute ("omp declare simd", attrs))
{
ret = -1;
continue;
}
}
/* If we are or might be in a target region or
declare target function, need to take into account
also offloading values. */
if (!omp_maybe_offloaded ())
return 0;
if (ENABLE_OFFLOADING)
{
const char *isas = omp_offload_device_isa;
if (omp_offload_device_kind_arch_isa (isas, isa))
{
ret = -1;
continue;
}
}
return 0;
}
else if (r == -1)
ret = -1;
/* If isa matches on the host, it still might not match
in the offloading region. */
else if (omp_maybe_offloaded ())
ret = -1;
}
break;
case OMP_TRAIT_USER_CONDITION:
if (set == OMP_TRAIT_SET_USER)
for (tree p = OMP_TS_PROPERTIES (ts); p; p = TREE_CHAIN (p))
if (OMP_TP_NAME (p) == NULL_TREE)
{
if (integer_zerop (OMP_TP_VALUE (p)))
return 0;
if (integer_nonzerop (OMP_TP_VALUE (p)))
break;
ret = -1;
}
break;
default:
break;
}
}
}
return ret;
}
/* Compare construct={simd} CLAUSES1 with CLAUSES2, return 0/-1/1/2 as
in omp_context_selector_set_compare. */
static int
omp_construct_simd_compare (tree clauses1, tree clauses2)
{
if (clauses1 == NULL_TREE)
return clauses2 == NULL_TREE ? 0 : -1;
if (clauses2 == NULL_TREE)
return 1;
int r = 0;
struct declare_variant_simd_data {
bool inbranch, notinbranch;
tree simdlen;
auto_vec<tree,16> data_sharing;
auto_vec<tree,16> aligned;
declare_variant_simd_data ()
: inbranch(false), notinbranch(false), simdlen(NULL_TREE) {}
} data[2];
unsigned int i;
for (i = 0; i < 2; i++)
for (tree c = i ? clauses2 : clauses1; c; c = OMP_CLAUSE_CHAIN (c))
{
vec<tree> *v;
switch (OMP_CLAUSE_CODE (c))
{
case OMP_CLAUSE_INBRANCH:
data[i].inbranch = true;
continue;
case OMP_CLAUSE_NOTINBRANCH:
data[i].notinbranch = true;
continue;
case OMP_CLAUSE_SIMDLEN:
data[i].simdlen = OMP_CLAUSE_SIMDLEN_EXPR (c);
continue;
case OMP_CLAUSE_UNIFORM:
case OMP_CLAUSE_LINEAR:
v = &data[i].data_sharing;
break;
case OMP_CLAUSE_ALIGNED:
v = &data[i].aligned;
break;
default:
gcc_unreachable ();
}
unsigned HOST_WIDE_INT argno = tree_to_uhwi (OMP_CLAUSE_DECL (c));
if (argno >= v->length ())
v->safe_grow_cleared (argno + 1, true);
(*v)[argno] = c;
}
/* Here, r is used as a bitmask, 2 is set if CLAUSES1 has something
CLAUSES2 doesn't, 1 is set if CLAUSES2 has something CLAUSES1
doesn't. Thus, r == 3 implies return value 2, r == 1 implies
-1, r == 2 implies 1 and r == 0 implies 0. */
if (data[0].inbranch != data[1].inbranch)
r |= data[0].inbranch ? 2 : 1;
if (data[0].notinbranch != data[1].notinbranch)
r |= data[0].notinbranch ? 2 : 1;
if (!simple_cst_equal (data[0].simdlen, data[1].simdlen))
{
if (data[0].simdlen && data[1].simdlen)
return 2;
r |= data[0].simdlen ? 2 : 1;
}
if (data[0].data_sharing.length () < data[1].data_sharing.length ()
|| data[0].aligned.length () < data[1].aligned.length ())
r |= 1;
tree c1, c2;
FOR_EACH_VEC_ELT (data[0].data_sharing, i, c1)
{
c2 = (i < data[1].data_sharing.length ()
? data[1].data_sharing[i] : NULL_TREE);
if ((c1 == NULL_TREE) != (c2 == NULL_TREE))
{
r |= c1 != NULL_TREE ? 2 : 1;
continue;
}
if (c1 == NULL_TREE)
continue;
if (OMP_CLAUSE_CODE (c1) != OMP_CLAUSE_CODE (c2))
return 2;
if (OMP_CLAUSE_CODE (c1) != OMP_CLAUSE_LINEAR)
continue;
if (OMP_CLAUSE_LINEAR_VARIABLE_STRIDE (c1)
!= OMP_CLAUSE_LINEAR_VARIABLE_STRIDE (c2))
return 2;
if (OMP_CLAUSE_LINEAR_KIND (c1) != OMP_CLAUSE_LINEAR_KIND (c2))
return 2;
if (!simple_cst_equal (OMP_CLAUSE_LINEAR_STEP (c1),
OMP_CLAUSE_LINEAR_STEP (c2)))
return 2;
}
FOR_EACH_VEC_ELT (data[0].aligned, i, c1)
{
c2 = i < data[1].aligned.length () ? data[1].aligned[i] : NULL_TREE;
if ((c1 == NULL_TREE) != (c2 == NULL_TREE))
{
r |= c1 != NULL_TREE ? 2 : 1;
continue;
}
if (c1 == NULL_TREE)
continue;
if (!simple_cst_equal (OMP_CLAUSE_ALIGNED_ALIGNMENT (c1),
OMP_CLAUSE_ALIGNED_ALIGNMENT (c2)))
return 2;
}
switch (r)
{
case 0: return 0;
case 1: return -1;
case 2: return 1;
case 3: return 2;
default: gcc_unreachable ();
}
}
/* Compare properties of selectors SEL from SET other than construct.
CTX1 and CTX2 are the lists of properties to compare.
Return 0/-1/1/2 as in omp_context_selector_set_compare.
Unlike set names or selector names, properties can have duplicates. */
static int
omp_context_selector_props_compare (enum omp_tss_code set,
enum omp_ts_code sel,
tree ctx1, tree ctx2)
{
int ret = 0;
for (int pass = 0; pass < 2; pass++)
for (tree p1 = pass ? ctx2 : ctx1; p1; p1 = TREE_CHAIN (p1))
{
tree p2;
for (p2 = pass ? ctx1 : ctx2; p2; p2 = TREE_CHAIN (p2))
if (OMP_TP_NAME (p1) == OMP_TP_NAME (p2))
{
if (OMP_TP_NAME (p1) == NULL_TREE)
{
if (set == OMP_TRAIT_SET_USER
&& sel == OMP_TRAIT_USER_CONDITION)
{
if (integer_zerop (OMP_TP_VALUE (p1))
!= integer_zerop (OMP_TP_VALUE (p2)))
return 2;
break;
}
if (simple_cst_equal (OMP_TP_VALUE (p1), OMP_TP_VALUE (p2)))
break;
}
else if (OMP_TP_NAME (p1) == OMP_TP_NAMELIST_NODE)
{
/* Handle string constant vs identifier comparison for
name-list properties. */
const char *n1 = omp_context_name_list_prop (p1);
const char *n2 = omp_context_name_list_prop (p2);
if (n1 && n2 && !strcmp (n1, n2))
break;
}
else
break;
}
if (p2 == NULL_TREE)
{
int r = pass ? -1 : 1;
if (ret && ret != r)
return 2;
else if (pass)
return r;
else
{
ret = r;
break;
}
}
}
return ret;
}
/* Compare single context selector sets CTX1 and CTX2 with SET name.
CTX1 and CTX2 are lists of trait-selectors.
Return 0 if CTX1 is equal to CTX2,
-1 if CTX1 is a strict subset of CTX2,
1 if CTX2 is a strict subset of CTX1, or
2 if neither context is a subset of another one. */
int
omp_context_selector_set_compare (enum omp_tss_code set, tree ctx1, tree ctx2)
{
/* If either list includes an ignored selector trait, neither can
be a subset of the other. */
for (tree ts = ctx1; ts; ts = TREE_CHAIN (ts))
if (OMP_TS_CODE (ts) == OMP_TRAIT_INVALID)
return 2;
for (tree ts = ctx2; ts; ts = TREE_CHAIN (ts))
if (OMP_TS_CODE (ts) == OMP_TRAIT_INVALID)
return 2;
bool swapped = false;
int ret = 0;
int len1 = list_length (ctx1);
int len2 = list_length (ctx2);
int cnt = 0;
if (len1 < len2)
{
swapped = true;
std::swap (ctx1, ctx2);
std::swap (len1, len2);
}
if (set == OMP_TRAIT_SET_CONSTRUCT)
{
tree ts1;
tree ts2 = ctx2;
/* Handle construct set specially. In this case the order
of the selector matters too. */
for (ts1 = ctx1; ts1; ts1 = TREE_CHAIN (ts1))
if (OMP_TS_CODE (ts1) == OMP_TS_CODE (ts2))
{
int r = 0;
if (OMP_TS_CODE (ts1) == OMP_TRAIT_CONSTRUCT_SIMD)
r = omp_construct_simd_compare (OMP_TS_PROPERTIES (ts1),
OMP_TS_PROPERTIES (ts2));
if (r == 2 || (ret && r && (ret < 0) != (r < 0)))
return 2;
if (ret == 0)
ret = r;
ts2 = TREE_CHAIN (ts2);
if (ts2 == NULL_TREE)
{
ts1 = TREE_CHAIN (ts1);
break;
}
}
else if (ret < 0)
return 2;
else
ret = 1;
if (ts2 != NULL_TREE)
return 2;
if (ts1 != NULL_TREE)
{
if (ret < 0)
return 2;
ret = 1;
}
if (ret == 0)
return 0;
return swapped ? -ret : ret;
}
for (tree ts1 = ctx1; ts1; ts1 = TREE_CHAIN (ts1))
{
enum omp_ts_code sel = OMP_TS_CODE (ts1);
tree ts2;
for (ts2 = ctx2; ts2; ts2 = TREE_CHAIN (ts2))
if (sel == OMP_TS_CODE (ts2))
{
tree score1 = OMP_TS_SCORE (ts1);
tree score2 = OMP_TS_SCORE (ts2);
if (score1 && score2 && !simple_cst_equal (score1, score2))
return 2;
int r = omp_context_selector_props_compare (set, OMP_TS_CODE (ts1),
OMP_TS_PROPERTIES (ts1),
OMP_TS_PROPERTIES (ts2));
if (r == 2 || (ret && r && (ret < 0) != (r < 0)))
return 2;
if (ret == 0)
ret = r;
cnt++;
break;
}
if (ts2 == NULL_TREE)
{
if (ret == -1)
return 2;
ret = 1;
}
}
if (cnt < len2)
return 2;
if (ret == 0)
return 0;
return swapped ? -ret : ret;
}
/* Compare whole context selector specification CTX1 and CTX2.
Return 0 if CTX1 is equal to CTX2,
-1 if CTX1 is a strict subset of CTX2,
1 if CTX2 is a strict subset of CTX1, or
2 if neither context is a subset of another one. */
static int
omp_context_selector_compare (tree ctx1, tree ctx2)
{
bool swapped = false;
int ret = 0;
int len1 = list_length (ctx1);
int len2 = list_length (ctx2);
int cnt = 0;
if (len1 < len2)
{
swapped = true;
std::swap (ctx1, ctx2);
std::swap (len1, len2);
}
for (tree tss1 = ctx1; tss1; tss1 = TREE_CHAIN (tss1))
{
enum omp_tss_code set = OMP_TSS_CODE (tss1);
tree tss2;
for (tss2 = ctx2; tss2; tss2 = TREE_CHAIN (tss2))
if (set == OMP_TSS_CODE (tss2))
{
int r
= omp_context_selector_set_compare
(set, OMP_TSS_TRAIT_SELECTORS (tss1),
OMP_TSS_TRAIT_SELECTORS (tss2));
if (r == 2 || (ret && r && (ret < 0) != (r < 0)))
return 2;
if (ret == 0)
ret = r;
cnt++;
break;
}
if (tss2 == NULL_TREE)
{
if (ret == -1)
return 2;
ret = 1;
}
}
if (cnt < len2)
return 2;
if (ret == 0)
return 0;
return swapped ? -ret : ret;
}
/* From context selector CTX, return trait-selector with name SEL in
trait-selector-set with name SET if any, or NULL_TREE if not found. */
tree
omp_get_context_selector (tree ctx, enum omp_tss_code set,
enum omp_ts_code sel)
{
for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
if (OMP_TSS_CODE (tss) == set)
for (tree ts = OMP_TSS_TRAIT_SELECTORS (tss); ts; ts = TREE_CHAIN (ts))
if (OMP_TS_CODE (ts) == sel)
return ts;
return NULL_TREE;
}
/* Similar, but returns the whole trait-selector list for SET in CTX. */
tree
omp_get_context_selector_list (tree ctx, enum omp_tss_code set)
{
for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
if (OMP_TSS_CODE (tss) == set)
return OMP_TSS_TRAIT_SELECTORS (tss);
return NULL_TREE;
}
/* Map string S onto a trait selector set code. */
enum omp_tss_code
omp_lookup_tss_code (const char * s)
{
for (int i = 0; i < OMP_TRAIT_SET_LAST; i++)
if (strcmp (s, omp_tss_map[i]) == 0)
return (enum omp_tss_code) i;
return OMP_TRAIT_SET_INVALID;
}
/* Map string S onto a trait selector code for set SET. */
enum omp_ts_code
omp_lookup_ts_code (enum omp_tss_code set, const char *s)
{
unsigned int mask = 1 << set;
for (int i = 0; i < OMP_TRAIT_LAST; i++)
if ((mask & omp_ts_map[i].tss_mask) != 0
&& strcmp (s, omp_ts_map[i].name) == 0)
return (enum omp_ts_code) i;
return OMP_TRAIT_INVALID;
}
/* Needs to be a GC-friendly widest_int variant, but precision is
desirable to be the same on all targets. */
typedef generic_wide_int <fixed_wide_int_storage <1024> > score_wide_int;
/* Compute *SCORE for context selector CTX. Return true if the score
would be different depending on whether it is a declare simd clone or
not. DECLARE_SIMD should be true for the case when it would be
a declare simd clone. */
static bool
omp_context_compute_score (tree ctx, score_wide_int *score, bool declare_simd)
{
tree selectors
= omp_get_context_selector_list (ctx, OMP_TRAIT_SET_CONSTRUCT);
bool has_kind = omp_get_context_selector (ctx, OMP_TRAIT_SET_DEVICE,
OMP_TRAIT_DEVICE_KIND);
bool has_arch = omp_get_context_selector (ctx, OMP_TRAIT_SET_DEVICE,
OMP_TRAIT_DEVICE_ARCH);
bool has_isa = omp_get_context_selector (ctx, OMP_TRAIT_SET_DEVICE,
OMP_TRAIT_DEVICE_ISA);
bool ret = false;
*score = 1;
for (tree tss = ctx; tss; tss = TREE_CHAIN (tss))
if (OMP_TSS_TRAIT_SELECTORS (tss) != selectors)
for (tree ts = OMP_TSS_TRAIT_SELECTORS (tss); ts; ts = TREE_CHAIN (ts))
{
tree s = OMP_TS_SCORE (ts);
if (s && TREE_CODE (s) == INTEGER_CST)
*score += score_wide_int::from (wi::to_wide (s),
TYPE_SIGN (TREE_TYPE (s)));
}
if (selectors || has_kind || has_arch || has_isa)
{
int nconstructs = list_length (selectors);
enum tree_code *constructs = NULL;
if (nconstructs)
{
constructs
= (enum tree_code *) alloca (nconstructs
* sizeof (enum tree_code));
omp_construct_traits_to_codes (selectors, nconstructs, constructs);
}
int *scores
= (int *) alloca ((2 * nconstructs + 2) * sizeof (int));
if (omp_construct_selector_matches (constructs, nconstructs, scores)
== 2)
ret = true;
int b = declare_simd ? nconstructs + 1 : 0;
if (scores[b + nconstructs] + 4U < score->get_precision ())
{
for (int n = 0; n < nconstructs; ++n)
{
if (scores[b + n] < 0)
{
*score = -1;
return ret;
}
*score += wi::shifted_mask <score_wide_int> (scores[b + n], 1, false);
}
if (has_kind)
*score += wi::shifted_mask <score_wide_int> (scores[b + nconstructs],
1, false);
if (has_arch)
*score += wi::shifted_mask <score_wide_int> (scores[b + nconstructs] + 1,
1, false);
if (has_isa)
*score += wi::shifted_mask <score_wide_int> (scores[b + nconstructs] + 2,
1, false);
}
else /* FIXME: Implement this. */
gcc_unreachable ();
}
return ret;
}
/* Class describing a single variant. */
struct GTY(()) omp_declare_variant_entry {
/* NODE of the variant. */
cgraph_node *variant;
/* Score if not in declare simd clone. */
score_wide_int score;
/* Score if in declare simd clone. */
score_wide_int score_in_declare_simd_clone;
/* Context selector for the variant. */
tree ctx;
/* True if the context selector is known to match already. */
bool matches;
};
/* Class describing a function with variants. */
struct GTY((for_user)) omp_declare_variant_base_entry {
/* NODE of the base function. */
cgraph_node *base;
/* NODE of the artificial function created for the deferred variant
resolution. */
cgraph_node *node;
/* Vector of the variants. */
vec<omp_declare_variant_entry, va_gc> *variants;
};
struct omp_declare_variant_hasher
: ggc_ptr_hash<omp_declare_variant_base_entry> {
static hashval_t hash (omp_declare_variant_base_entry *);
static bool equal (omp_declare_variant_base_entry *,
omp_declare_variant_base_entry *);
};
hashval_t
omp_declare_variant_hasher::hash (omp_declare_variant_base_entry *x)
{
inchash::hash hstate;
hstate.add_int (DECL_UID (x->base->decl));
hstate.add_int (x->variants->length ());
omp_declare_variant_entry *variant;
unsigned int i;
FOR_EACH_VEC_SAFE_ELT (x->variants, i, variant)
{
hstate.add_int (DECL_UID (variant->variant->decl));
hstate.add_wide_int (variant->score);
hstate.add_wide_int (variant->score_in_declare_simd_clone);
hstate.add_ptr (variant->ctx);
hstate.add_int (variant->matches);
}
return hstate.end ();
}
bool
omp_declare_variant_hasher::equal (omp_declare_variant_base_entry *x,
omp_declare_variant_base_entry *y)
{
if (x->base != y->base
|| x->variants->length () != y->variants->length ())
return false;
omp_declare_variant_entry *variant;
unsigned int i;
FOR_EACH_VEC_SAFE_ELT (x->variants, i, variant)
if (variant->variant != (*y->variants)[i].variant
|| variant->score != (*y->variants)[i].score
|| (variant->score_in_declare_simd_clone
!= (*y->variants)[i].score_in_declare_simd_clone)
|| variant->ctx != (*y->variants)[i].ctx
|| variant->matches != (*y->variants)[i].matches)
return false;
return true;
}
static GTY(()) hash_table<omp_declare_variant_hasher> *omp_declare_variants;
struct omp_declare_variant_alt_hasher
: ggc_ptr_hash<omp_declare_variant_base_entry> {
static hashval_t hash (omp_declare_variant_base_entry *);
static bool equal (omp_declare_variant_base_entry *,
omp_declare_variant_base_entry *);
};
hashval_t
omp_declare_variant_alt_hasher::hash (omp_declare_variant_base_entry *x)
{
return DECL_UID (x->node->decl);
}
bool
omp_declare_variant_alt_hasher::equal (omp_declare_variant_base_entry *x,
omp_declare_variant_base_entry *y)
{
return x->node == y->node;
}
static GTY(()) hash_table<omp_declare_variant_alt_hasher>
*omp_declare_variant_alt;
/* Try to resolve declare variant after gimplification. */
static tree
omp_resolve_late_declare_variant (tree alt)
{
cgraph_node *node = cgraph_node::get (alt);
cgraph_node *cur_node = cgraph_node::get (cfun->decl);
if (node == NULL
|| !node->declare_variant_alt
|| !cfun->after_inlining)
return alt;
omp_declare_variant_base_entry entry;
entry.base = NULL;
entry.node = node;
entry.variants = NULL;
omp_declare_variant_base_entry *entryp
= omp_declare_variant_alt->find_with_hash (&entry, DECL_UID (alt));
unsigned int i, j;
omp_declare_variant_entry *varentry1, *varentry2;
auto_vec <bool, 16> matches;
unsigned int nmatches = 0;
FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry1)
{
if (varentry1->matches)
{
/* This has been checked to be ok already. */
matches.safe_push (true);
nmatches++;
continue;
}
switch (omp_context_selector_matches (varentry1->ctx))
{
case 0:
matches.safe_push (false);
break;
case -1:
return alt;
default:
matches.safe_push (true);
nmatches++;
break;
}
}
if (nmatches == 0)
return entryp->base->decl;
/* A context selector that is a strict subset of another context selector
has a score of zero. */
FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry1)
if (matches[i])
{
for (j = i + 1;
vec_safe_iterate (entryp->variants, j, &varentry2); ++j)
if (matches[j])
{
int r = omp_context_selector_compare (varentry1->ctx,
varentry2->ctx);
if (r == -1)
{
/* ctx1 is a strict subset of ctx2, ignore ctx1. */
matches[i] = false;
break;
}
else if (r == 1)
/* ctx2 is a strict subset of ctx1, remove ctx2. */
matches[j] = false;
}
}
score_wide_int max_score = -1;
varentry2 = NULL;
FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry1)
if (matches[i])
{
score_wide_int score
= (cur_node->simdclone ? varentry1->score_in_declare_simd_clone
: varentry1->score);
if (score > max_score)
{
max_score = score;
varentry2 = varentry1;
}
}
return varentry2->variant->decl;
}
/* Hook to adjust hash tables on cgraph_node removal. */
static void
omp_declare_variant_remove_hook (struct cgraph_node *node, void *)
{
if (!node->declare_variant_alt)
return;
/* Drop this hash table completely. */
omp_declare_variants = NULL;
/* And remove node from the other hash table. */
if (omp_declare_variant_alt)
{
omp_declare_variant_base_entry entry;
entry.base = NULL;
entry.node = node;
entry.variants = NULL;
omp_declare_variant_alt->remove_elt_with_hash (&entry,
DECL_UID (node->decl));
}
}
/* Try to resolve declare variant, return the variant decl if it should
be used instead of base, or base otherwise. */
tree
omp_resolve_declare_variant (tree base)
{
tree variant1 = NULL_TREE, variant2 = NULL_TREE;
if (cfun && (cfun->curr_properties & PROP_gimple_any) != 0)
return omp_resolve_late_declare_variant (base);
auto_vec <tree, 16> variants;
auto_vec <bool, 16> defer;
bool any_deferred = false;
for (tree attr = DECL_ATTRIBUTES (base); attr; attr = TREE_CHAIN (attr))
{
attr = lookup_attribute ("omp declare variant base", attr);
if (attr == NULL_TREE)
break;
if (TREE_CODE (TREE_PURPOSE (TREE_VALUE (attr))) != FUNCTION_DECL)
continue;
cgraph_node *node = cgraph_node::get (base);
/* If this is already a magic decl created by this function,
don't process it again. */
if (node && node->declare_variant_alt)
return base;
switch (omp_context_selector_matches (TREE_VALUE (TREE_VALUE (attr))))
{
case 0:
/* No match, ignore. */
break;
case -1:
/* Needs to be deferred. */
any_deferred = true;
variants.safe_push (attr);
defer.safe_push (true);
break;
default:
variants.safe_push (attr);
defer.safe_push (false);
break;
}
}
if (variants.length () == 0)
return base;
if (any_deferred)
{
score_wide_int max_score1 = 0;
score_wide_int max_score2 = 0;
bool first = true;
unsigned int i;
tree attr1, attr2;
omp_declare_variant_base_entry entry;
entry.base = cgraph_node::get_create (base);
entry.node = NULL;
vec_alloc (entry.variants, variants.length ());
FOR_EACH_VEC_ELT (variants, i, attr1)
{
score_wide_int score1;
score_wide_int score2;
bool need_two;
tree ctx = TREE_VALUE (TREE_VALUE (attr1));
need_two = omp_context_compute_score (ctx, &score1, false);
if (need_two)
omp_context_compute_score (ctx, &score2, true);
else
score2 = score1;
if (first)
{
first = false;
max_score1 = score1;
max_score2 = score2;
if (!defer[i])
{
variant1 = attr1;
variant2 = attr1;
}
}
else
{
if (max_score1 == score1)
variant1 = NULL_TREE;
else if (score1 > max_score1)
{
max_score1 = score1;
variant1 = defer[i] ? NULL_TREE : attr1;
}
if (max_score2 == score2)
variant2 = NULL_TREE;
else if (score2 > max_score2)
{
max_score2 = score2;
variant2 = defer[i] ? NULL_TREE : attr1;
}
}
omp_declare_variant_entry varentry;
varentry.variant
= cgraph_node::get_create (TREE_PURPOSE (TREE_VALUE (attr1)));
varentry.score = score1;
varentry.score_in_declare_simd_clone = score2;
varentry.ctx = ctx;
varentry.matches = !defer[i];
entry.variants->quick_push (varentry);
}
/* If there is a clear winner variant with the score which is not
deferred, verify it is not a strict subset of any other context
selector and if it is not, it is the best alternative no matter
whether the others do or don't match. */
if (variant1 && variant1 == variant2)
{
tree ctx1 = TREE_VALUE (TREE_VALUE (variant1));
FOR_EACH_VEC_ELT (variants, i, attr2)
{
if (attr2 == variant1)
continue;
tree ctx2 = TREE_VALUE (TREE_VALUE (attr2));
int r = omp_context_selector_compare (ctx1, ctx2);
if (r == -1)
{
/* The winner is a strict subset of ctx2, can't
decide now. */
variant1 = NULL_TREE;
break;
}
}
if (variant1)
{
vec_free (entry.variants);
return TREE_PURPOSE (TREE_VALUE (variant1));
}
}
static struct cgraph_node_hook_list *node_removal_hook_holder;
if (!node_removal_hook_holder)
node_removal_hook_holder
= symtab->add_cgraph_removal_hook (omp_declare_variant_remove_hook,
NULL);
if (omp_declare_variants == NULL)
omp_declare_variants
= hash_table<omp_declare_variant_hasher>::create_ggc (64);
omp_declare_variant_base_entry **slot
= omp_declare_variants->find_slot (&entry, INSERT);
if (*slot != NULL)
{
vec_free (entry.variants);
return (*slot)->node->decl;
}
*slot = ggc_cleared_alloc<omp_declare_variant_base_entry> ();
(*slot)->base = entry.base;
(*slot)->node = entry.base;
(*slot)->variants = entry.variants;
tree alt = build_decl (DECL_SOURCE_LOCATION (base), FUNCTION_DECL,
DECL_NAME (base), TREE_TYPE (base));
DECL_ARTIFICIAL (alt) = 1;
DECL_IGNORED_P (alt) = 1;
TREE_STATIC (alt) = 1;
tree attributes = DECL_ATTRIBUTES (base);
if (lookup_attribute ("noipa", attributes) == NULL)
{
attributes = tree_cons (get_identifier ("noipa"), NULL, attributes);
if (lookup_attribute ("noinline", attributes) == NULL)
attributes = tree_cons (get_identifier ("noinline"), NULL,
attributes);
if (lookup_attribute ("noclone", attributes) == NULL)
attributes = tree_cons (get_identifier ("noclone"), NULL,
attributes);
if (lookup_attribute ("no_icf", attributes) == NULL)
attributes = tree_cons (get_identifier ("no_icf"), NULL,
attributes);
}
DECL_ATTRIBUTES (alt) = attributes;
DECL_INITIAL (alt) = error_mark_node;
(*slot)->node = cgraph_node::create (alt);
(*slot)->node->declare_variant_alt = 1;
(*slot)->node->create_reference (entry.base, IPA_REF_ADDR);
omp_declare_variant_entry *varentry;
FOR_EACH_VEC_SAFE_ELT (entry.variants, i, varentry)
(*slot)->node->create_reference (varentry->variant, IPA_REF_ADDR);
if (omp_declare_variant_alt == NULL)
omp_declare_variant_alt
= hash_table<omp_declare_variant_alt_hasher>::create_ggc (64);
*omp_declare_variant_alt->find_slot_with_hash (*slot, DECL_UID (alt),
INSERT) = *slot;
return alt;
}
if (variants.length () == 1)
return TREE_PURPOSE (TREE_VALUE (variants[0]));
/* A context selector that is a strict subset of another context selector
has a score of zero. */
tree attr1, attr2;
unsigned int i, j;
FOR_EACH_VEC_ELT (variants, i, attr1)
if (attr1)
{
tree ctx1 = TREE_VALUE (TREE_VALUE (attr1));
FOR_EACH_VEC_ELT_FROM (variants, j, attr2, i + 1)
if (attr2)
{
tree ctx2 = TREE_VALUE (TREE_VALUE (attr2));
int r = omp_context_selector_compare (ctx1, ctx2);
if (r == -1)
{
/* ctx1 is a strict subset of ctx2, remove
attr1 from the vector. */
variants[i] = NULL_TREE;
break;
}
else if (r == 1)
/* ctx2 is a strict subset of ctx1, remove attr2
from the vector. */
variants[j] = NULL_TREE;
}
}
score_wide_int max_score1 = 0;
score_wide_int max_score2 = 0;
bool first = true;
FOR_EACH_VEC_ELT (variants, i, attr1)
if (attr1)
{
if (variant1)
{
score_wide_int score1;
score_wide_int score2;
bool need_two;
tree ctx;
if (first)
{
first = false;
ctx = TREE_VALUE (TREE_VALUE (variant1));
need_two = omp_context_compute_score (ctx, &max_score1, false);
if (need_two)
omp_context_compute_score (ctx, &max_score2, true);
else
max_score2 = max_score1;
}
ctx = TREE_VALUE (TREE_VALUE (attr1));
need_two = omp_context_compute_score (ctx, &score1, false);
if (need_two)
omp_context_compute_score (ctx, &score2, true);
else
score2 = score1;
if (score1 > max_score1)
{
max_score1 = score1;
variant1 = attr1;
}
if (score2 > max_score2)
{
max_score2 = score2;
variant2 = attr1;
}
}
else
{
variant1 = attr1;
variant2 = attr1;
}
}
/* If there is a disagreement on which variant has the highest score
depending on whether it will be in a declare simd clone or not,
punt for now and defer until after IPA where we will know that. */
return ((variant1 && variant1 == variant2)
? TREE_PURPOSE (TREE_VALUE (variant1)) : base);
}
void
omp_lto_output_declare_variant_alt (lto_simple_output_block *ob,
cgraph_node *node,
lto_symtab_encoder_t encoder)
{
gcc_assert (node->declare_variant_alt);
omp_declare_variant_base_entry entry;
entry.base = NULL;
entry.node = node;
entry.variants = NULL;
omp_declare_variant_base_entry *entryp
= omp_declare_variant_alt->find_with_hash (&entry, DECL_UID (node->decl));
gcc_assert (entryp);
int nbase = lto_symtab_encoder_lookup (encoder, entryp->base);
gcc_assert (nbase != LCC_NOT_FOUND);
streamer_write_hwi_stream (ob->main_stream, nbase);
streamer_write_hwi_stream (ob->main_stream, entryp->variants->length ());
unsigned int i;
omp_declare_variant_entry *varentry;
FOR_EACH_VEC_SAFE_ELT (entryp->variants, i, varentry)
{
int nvar = lto_symtab_encoder_lookup (encoder, varentry->variant);
gcc_assert (nvar != LCC_NOT_FOUND);
streamer_write_hwi_stream (ob->main_stream, nvar);
for (score_wide_int *w = &varentry->score; ;
w = &varentry->score_in_declare_simd_clone)
{
unsigned len = w->get_len ();
streamer_write_hwi_stream (ob->main_stream, len);
const HOST_WIDE_INT *val = w->get_val ();
for (unsigned j = 0; j < len; j++)
streamer_write_hwi_stream (ob->main_stream, val[j]);
if (w == &varentry->score_in_declare_simd_clone)
break;
}
HOST_WIDE_INT cnt = -1;
HOST_WIDE_INT i = varentry->matches ? 1 : 0;
for (tree attr = DECL_ATTRIBUTES (entryp->base->decl);
attr; attr = TREE_CHAIN (attr), i += 2)
{
attr = lookup_attribute ("omp declare variant base", attr);
if (attr == NULL_TREE)
break;
if (varentry->ctx == TREE_VALUE (TREE_VALUE (attr)))
{
cnt = i;
break;
}
}
gcc_assert (cnt != -1);
streamer_write_hwi_stream (ob->main_stream, cnt);
}
}
void
omp_lto_input_declare_variant_alt (lto_input_block *ib, cgraph_node *node,
vec<symtab_node *> nodes)
{
gcc_assert (node->declare_variant_alt);
omp_declare_variant_base_entry *entryp
= ggc_cleared_alloc<omp_declare_variant_base_entry> ();
entryp->base = dyn_cast<cgraph_node *> (nodes[streamer_read_hwi (ib)]);
entryp->node = node;
unsigned int len = streamer_read_hwi (ib);
vec_alloc (entryp->variants, len);
for (unsigned int i = 0; i < len; i++)
{
omp_declare_variant_entry varentry;
varentry.variant
= dyn_cast<cgraph_node *> (nodes[streamer_read_hwi (ib)]);
for (score_wide_int *w = &varentry.score; ;
w = &varentry.score_in_declare_simd_clone)
{
unsigned len2 = streamer_read_hwi (ib);
HOST_WIDE_INT arr[WIDE_INT_MAX_HWIS (1024)];
gcc_assert (len2 <= WIDE_INT_MAX_HWIS (1024));
for (unsigned int j = 0; j < len2; j++)
arr[j] = streamer_read_hwi (ib);
*w = score_wide_int::from_array (arr, len2, true);
if (w == &varentry.score_in_declare_simd_clone)
break;
}
HOST_WIDE_INT cnt = streamer_read_hwi (ib);
HOST_WIDE_INT j = 0;
varentry.ctx = NULL_TREE;
varentry.matches = (cnt & 1) ? true : false;
cnt &= ~HOST_WIDE_INT_1;
for (tree attr = DECL_ATTRIBUTES (entryp->base->decl);
attr; attr = TREE_CHAIN (attr), j += 2)
{
attr = lookup_attribute ("omp declare variant base", attr);
if (attr == NULL_TREE)
break;
if (cnt == j)
{
varentry.ctx = TREE_VALUE (TREE_VALUE (attr));
break;
}
}
gcc_assert (varentry.ctx != NULL_TREE);
entryp->variants->quick_push (varentry);
}
if (omp_declare_variant_alt == NULL)
omp_declare_variant_alt
= hash_table<omp_declare_variant_alt_hasher>::create_ggc (64);
*omp_declare_variant_alt->find_slot_with_hash (entryp, DECL_UID (node->decl),
INSERT) = entryp;
}
/* Encode an oacc launch argument. This matches the GOMP_LAUNCH_PACK
macro on gomp-constants.h. We do not check for overflow. */
tree
oacc_launch_pack (unsigned code, tree device, unsigned op)
{
tree res;
res = build_int_cst (unsigned_type_node, GOMP_LAUNCH_PACK (code, 0, op));
if (device)
{
device = fold_build2 (LSHIFT_EXPR, unsigned_type_node,
device, build_int_cst (unsigned_type_node,
GOMP_LAUNCH_DEVICE_SHIFT));
res = fold_build2 (BIT_IOR_EXPR, unsigned_type_node, res, device);
}
return res;
}
/* Openacc compute grid dimension clauses are converted to an attribute
attached to the function. This permits the target-side code to (a) massage
the dimensions, (b) emit that data and (c) optimize. Non-constant
dimensions are pushed onto ARGS.
The attribute value is a TREE_LIST. A set of dimensions is
represented as a list of INTEGER_CST. Those that are runtime
exprs are represented as an INTEGER_CST of zero.
TODO: Normally the attribute will just contain a single such list. If
however it contains a list of lists, this will represent the use of
device_type. Each member of the outer list is an assoc list of
dimensions, keyed by the device type. The first entry will be the
default. Well, that's the plan. */
/* Replace any existing oacc fn attribute in ATTRIBS with updated
dimensions. */
tree
oacc_replace_fn_attrib_attr (tree attribs, tree dims)
{
tree ident = get_identifier (OACC_FN_ATTRIB);
/* If we happen to be present as the first attrib, drop it. */
if (attribs && TREE_PURPOSE (attribs) == ident)
attribs = TREE_CHAIN (attribs);
return tree_cons (ident, dims, attribs);
}
/* Replace any existing oacc fn attribute on FN with updated
dimensions. */
void
oacc_replace_fn_attrib (tree fn, tree dims)
{
DECL_ATTRIBUTES (fn)
= oacc_replace_fn_attrib_attr (DECL_ATTRIBUTES (fn), dims);
}
/* Scan CLAUSES for launch dimensions and attach them to the oacc
function attribute. Push any that are non-constant onto the ARGS
list, along with an appropriate GOMP_LAUNCH_DIM tag. */
void
oacc_set_fn_attrib (tree fn, tree clauses, vec<tree> *args)
{
/* Must match GOMP_DIM ordering. */
static const omp_clause_code ids[]
= { OMP_CLAUSE_NUM_GANGS, OMP_CLAUSE_NUM_WORKERS,
OMP_CLAUSE_VECTOR_LENGTH };
unsigned ix;
tree dims[GOMP_DIM_MAX];
tree attr = NULL_TREE;
unsigned non_const = 0;
for (ix = GOMP_DIM_MAX; ix--;)
{
tree clause = omp_find_clause (clauses, ids[ix]);
tree dim = NULL_TREE;
if (clause)
dim = OMP_CLAUSE_EXPR (clause, ids[ix]);
dims[ix] = dim;
if (dim && TREE_CODE (dim) != INTEGER_CST)
{
dim = integer_zero_node;
non_const |= GOMP_DIM_MASK (ix);
}
attr = tree_cons (NULL_TREE, dim, attr);
}
oacc_replace_fn_attrib (fn, attr);
if (non_const)
{
/* Push a dynamic argument set. */
args->safe_push (oacc_launch_pack (GOMP_LAUNCH_DIM,
NULL_TREE, non_const));
for (unsigned ix = 0; ix != GOMP_DIM_MAX; ix++)
if (non_const & GOMP_DIM_MASK (ix))
args->safe_push (dims[ix]);
}
}
/* Verify OpenACC routine clauses.
Returns 0 if FNDECL should be marked with an OpenACC 'routine' directive, 1
if it has already been marked in compatible way, and -1 if incompatible.
Upon returning, the chain of clauses will contain exactly one clause
specifying the level of parallelism. */
int
oacc_verify_routine_clauses (tree fndecl, tree *clauses, location_t loc,
const char *routine_str)
{
tree c_level = NULL_TREE;
tree c_nohost = NULL_TREE;
tree c_p = NULL_TREE;
for (tree c = *clauses; c; c_p = c, c = OMP_CLAUSE_CHAIN (c))
switch (OMP_CLAUSE_CODE (c))
{
case OMP_CLAUSE_GANG:
case OMP_CLAUSE_WORKER:
case OMP_CLAUSE_VECTOR:
case OMP_CLAUSE_SEQ:
if (c_level == NULL_TREE)
c_level = c;
else if (OMP_CLAUSE_CODE (c) == OMP_CLAUSE_CODE (c_level))
{
/* This has already been diagnosed in the front ends. */
/* Drop the duplicate clause. */
gcc_checking_assert (c_p != NULL_TREE);
OMP_CLAUSE_CHAIN (c_p) = OMP_CLAUSE_CHAIN (c);
c = c_p;
}
else
{
error_at (OMP_CLAUSE_LOCATION (c),
"%qs specifies a conflicting level of parallelism",
omp_clause_code_name[OMP_CLAUSE_CODE (c)]);
inform (OMP_CLAUSE_LOCATION (c_level),
"... to the previous %qs clause here",
omp_clause_code_name[OMP_CLAUSE_CODE (c_level)]);
/* Drop the conflicting clause. */
gcc_checking_assert (c_p != NULL_TREE);
OMP_CLAUSE_CHAIN (c_p) = OMP_CLAUSE_CHAIN (c);
c = c_p;
}
break;
case OMP_CLAUSE_NOHOST:
/* Don't worry about duplicate clauses here. */
c_nohost = c;
break;
default:
gcc_unreachable ();
}
if (c_level == NULL_TREE)
{
/* Default to an implicit 'seq' clause. */
c_level = build_omp_clause (loc, OMP_CLAUSE_SEQ);
OMP_CLAUSE_CHAIN (c_level) = *clauses;
*clauses = c_level;
}
/* In *clauses, we now have exactly one clause specifying the level of
parallelism. */
tree attr
= lookup_attribute ("omp declare target", DECL_ATTRIBUTES (fndecl));
if (attr != NULL_TREE)
{
/* Diagnose if "#pragma omp declare target" has also been applied. */
if (TREE_VALUE (attr) == NULL_TREE)
{
/* See <https://gcc.gnu.org/PR93465>; the semantics of combining
OpenACC and OpenMP 'target' are not clear. */
error_at (loc,
"cannot apply %<%s%> to %qD, which has also been"
" marked with an OpenMP 'declare target' directive",
routine_str, fndecl);
/* Incompatible. */
return -1;
}
/* If a "#pragma acc routine" has already been applied, just verify
this one for compatibility. */
/* Collect previous directive's clauses. */
tree c_level_p = NULL_TREE;
tree c_nohost_p = NULL_TREE;
for (tree c = TREE_VALUE (attr); c; c = OMP_CLAUSE_CHAIN (c))
switch (OMP_CLAUSE_CODE (c))
{
case OMP_CLAUSE_GANG:
case OMP_CLAUSE_WORKER:
case OMP_CLAUSE_VECTOR:
case OMP_CLAUSE_SEQ:
gcc_checking_assert (c_level_p == NULL_TREE);
c_level_p = c;
break;
case OMP_CLAUSE_NOHOST:
gcc_checking_assert (c_nohost_p == NULL_TREE);
c_nohost_p = c;
break;
default:
gcc_unreachable ();
}
gcc_checking_assert (c_level_p != NULL_TREE);
/* ..., and compare to current directive's, which we've already collected
above. */
tree c_diag;
tree c_diag_p;
/* Matching level of parallelism? */
if (OMP_CLAUSE_CODE (c_level) != OMP_CLAUSE_CODE (c_level_p))
{
c_diag = c_level;
c_diag_p = c_level_p;
goto incompatible;
}
/* Matching 'nohost' clauses? */
if ((c_nohost == NULL_TREE) != (c_nohost_p == NULL_TREE))
{
c_diag = c_nohost;
c_diag_p = c_nohost_p;
goto incompatible;
}
/* Compatible. */
return 1;
incompatible:
if (c_diag != NULL_TREE)
error_at (OMP_CLAUSE_LOCATION (c_diag),
"incompatible %qs clause when applying"
" %<%s%> to %qD, which has already been"
" marked with an OpenACC 'routine' directive",
omp_clause_code_name[OMP_CLAUSE_CODE (c_diag)],
routine_str, fndecl);
else if (c_diag_p != NULL_TREE)
error_at (loc,
"missing %qs clause when applying"
" %<%s%> to %qD, which has already been"
" marked with an OpenACC 'routine' directive",
omp_clause_code_name[OMP_CLAUSE_CODE (c_diag_p)],
routine_str, fndecl);
else
gcc_unreachable ();
if (c_diag_p != NULL_TREE)
inform (OMP_CLAUSE_LOCATION (c_diag_p),
"... with %qs clause here",
omp_clause_code_name[OMP_CLAUSE_CODE (c_diag_p)]);
else
{
/* In the front ends, we don't preserve location information for the
OpenACC routine directive itself. However, that of c_level_p
should be close. */
location_t loc_routine = OMP_CLAUSE_LOCATION (c_level_p);
inform (loc_routine, "... without %qs clause near to here",
omp_clause_code_name[OMP_CLAUSE_CODE (c_diag)]);
}
/* Incompatible. */
return -1;
}
return 0;
}
/* Process the OpenACC 'routine' directive clauses to generate an attribute
for the level of parallelism. All dimensions have a size of zero
(dynamic). TREE_PURPOSE is set to indicate whether that dimension
can have a loop partitioned on it. non-zero indicates
yes, zero indicates no. By construction once a non-zero has been
reached, further inner dimensions must also be non-zero. We set
TREE_VALUE to zero for the dimensions that may be partitioned and
1 for the other ones -- if a loop is (erroneously) spawned at
an outer level, we don't want to try and partition it. */
tree
oacc_build_routine_dims (tree clauses)
{
/* Must match GOMP_DIM ordering. */
static const omp_clause_code ids[]
= {OMP_CLAUSE_GANG, OMP_CLAUSE_WORKER, OMP_CLAUSE_VECTOR, OMP_CLAUSE_SEQ};
int ix;
int level = -1;
for (; clauses; clauses = OMP_CLAUSE_CHAIN (clauses))
for (ix = GOMP_DIM_MAX + 1; ix--;)
if (OMP_CLAUSE_CODE (clauses) == ids[ix])
{
level = ix;
break;
}
gcc_checking_assert (level >= 0);
tree dims = NULL_TREE;
for (ix = GOMP_DIM_MAX; ix--;)
dims = tree_cons (build_int_cst (boolean_type_node, ix >= level),
build_int_cst (integer_type_node, ix < level), dims);
return dims;
}
/* Retrieve the oacc function attrib and return it. Non-oacc
functions will return NULL. */
tree
oacc_get_fn_attrib (tree fn)
{
return lookup_attribute (OACC_FN_ATTRIB, DECL_ATTRIBUTES (fn));
}
/* Return true if FN is an OpenMP or OpenACC offloading function. */
bool
offloading_function_p (tree fn)
{
tree attrs = DECL_ATTRIBUTES (fn);
return (lookup_attribute ("omp declare target", attrs)
|| lookup_attribute ("omp target entrypoint", attrs));
}
/* Extract an oacc execution dimension from FN. FN must be an
offloaded function or routine that has already had its execution
dimensions lowered to the target-specific values. */
int
oacc_get_fn_dim_size (tree fn, int axis)
{
tree attrs = oacc_get_fn_attrib (fn);
gcc_assert (axis < GOMP_DIM_MAX);
tree dims = TREE_VALUE (attrs);
while (axis--)
dims = TREE_CHAIN (dims);
int size = TREE_INT_CST_LOW (TREE_VALUE (dims));
return size;
}
/* Extract the dimension axis from an IFN_GOACC_DIM_POS or
IFN_GOACC_DIM_SIZE call. */
int
oacc_get_ifn_dim_arg (const gimple *stmt)
{
gcc_checking_assert (gimple_call_internal_fn (stmt) == IFN_GOACC_DIM_SIZE
|| gimple_call_internal_fn (stmt) == IFN_GOACC_DIM_POS);
tree arg = gimple_call_arg (stmt, 0);
HOST_WIDE_INT axis = TREE_INT_CST_LOW (arg);
gcc_checking_assert (axis >= 0 && axis < GOMP_DIM_MAX);
return (int) axis;
}
/* Build COMPONENT_REF and set TREE_THIS_VOLATILE and TREE_READONLY on it
as appropriate. */
tree
omp_build_component_ref (tree obj, tree field)
{
tree ret = build3 (COMPONENT_REF, TREE_TYPE (field), obj, field, NULL);
if (TREE_THIS_VOLATILE (field))
TREE_THIS_VOLATILE (ret) |= 1;
if (TREE_READONLY (field))
TREE_READONLY (ret) |= 1;
return ret;
}
/* Return true if NAME is the name of an omp_* runtime API call. */
bool
omp_runtime_api_procname (const char *name)
{
if (!startswith (name, "omp_"))
return false;
static const char *omp_runtime_apis[] =
{
/* This array has 3 sections. First omp_* calls that don't
have any suffixes. */
"aligned_alloc",
"aligned_calloc",
"alloc",
"calloc",
"free",
"get_mapped_ptr",
"realloc",
"target_alloc",
"target_associate_ptr",
"target_disassociate_ptr",
"target_free",
"target_is_accessible",
"target_is_present",
"target_memcpy",
"target_memcpy_async",
"target_memcpy_rect",
"target_memcpy_rect_async",
NULL,
/* Now omp_* calls that are available as omp_* and omp_*_; however, the
DECL_NAME is always omp_* without tailing underscore. */
"capture_affinity",
"destroy_allocator",
"destroy_lock",
"destroy_nest_lock",
"display_affinity",
"fulfill_event",
"get_active_level",
"get_affinity_format",
"get_cancellation",
"get_default_allocator",
"get_default_device",
"get_device_num",
"get_dynamic",
"get_initial_device",
"get_level",
"get_max_active_levels",
"get_max_task_priority",
"get_max_teams",
"get_max_threads",
"get_nested",
"get_num_devices",
"get_num_places",
"get_num_procs",
"get_num_teams",
"get_num_threads",
"get_partition_num_places",
"get_place_num",
"get_proc_bind",
"get_supported_active_levels",
"get_team_num",
"get_teams_thread_limit",
"get_thread_limit",
"get_thread_num",
"get_wtick",
"get_wtime",
"in_explicit_task",
"in_final",
"in_parallel",
"init_lock",
"init_nest_lock",
"is_initial_device",
"pause_resource",
"pause_resource_all",
"set_affinity_format",
"set_default_allocator",
"set_lock",
"set_nest_lock",
"test_lock",
"test_nest_lock",
"unset_lock",
"unset_nest_lock",
NULL,
/* And finally calls available as omp_*, omp_*_ and omp_*_8_; however,
as DECL_NAME only omp_* and omp_*_8 appear. */
"display_env",
"get_ancestor_thread_num",
"init_allocator",
"get_partition_place_nums",
"get_place_num_procs",
"get_place_proc_ids",
"get_schedule",
"get_team_size",
"set_default_device",
"set_dynamic",
"set_max_active_levels",
"set_nested",
"set_num_teams",
"set_num_threads",
"set_schedule",
"set_teams_thread_limit"
};
int mode = 0;
for (unsigned i = 0; i < ARRAY_SIZE (omp_runtime_apis); i++)
{
if (omp_runtime_apis[i] == NULL)
{
mode++;
continue;
}
size_t len = strlen (omp_runtime_apis[i]);
if (strncmp (name + 4, omp_runtime_apis[i], len) == 0
&& (name[4 + len] == '\0'
|| (mode > 1 && strcmp (name + 4 + len, "_8") == 0)))
return true;
}
return false;
}
/* Return true if FNDECL is an omp_* runtime API call. */
bool
omp_runtime_api_call (const_tree fndecl)
{
tree declname = DECL_NAME (fndecl);
if (!declname
|| (DECL_CONTEXT (fndecl) != NULL_TREE
&& TREE_CODE (DECL_CONTEXT (fndecl)) != TRANSLATION_UNIT_DECL)
|| !TREE_PUBLIC (fndecl))
return false;
return omp_runtime_api_procname (IDENTIFIER_POINTER (declname));
}
namespace omp_addr_tokenizer {
/* We scan an expression by recursive descent, and build a vector of
"omp_addr_token *" pointers representing a "parsed" version of the
expression. The grammar we use is something like this:
expr0::
expr [section-access]
expr::
structured-expr access-method
| array-base access-method
structured-expr::
structure-base component-selector
arbitrary-expr::
(anything else)
structure-base::
DECL access-method
| structured-expr access-method
| arbitrary-expr access-method
array-base::
DECL
| arbitrary-expr
access-method::
DIRECT
| REF
| POINTER
| REF_TO_POINTER
| POINTER_OFFSET
| REF_TO_POINTER_OFFSET
| INDEXED_ARRAY
| INDEXED_REF_TO_ARRAY
| index-expr
index-expr::
INDEX_EXPR access-method
component-selector::
component-selector COMPONENT_REF
| component-selector ARRAY_REF
| COMPONENT_REF
This tokenized form is then used both in parsing, for OpenMP clause
expansion (for C and C++) and in gimplify.cc for sibling-list handling
(for C, C++ and Fortran). */
omp_addr_token::omp_addr_token (token_type t, tree e)
: type(t), expr(e)
{
}
omp_addr_token::omp_addr_token (access_method_kinds k, tree e)
: type(ACCESS_METHOD), expr(e)
{
u.access_kind = k;
}
omp_addr_token::omp_addr_token (token_type t, structure_base_kinds k, tree e)
: type(t), expr(e)
{
u.structure_base_kind = k;
}
static bool
omp_parse_component_selector (tree *expr0)
{
tree expr = *expr0;
tree last_component = NULL_TREE;
while (TREE_CODE (expr) == COMPONENT_REF
|| TREE_CODE (expr) == ARRAY_REF)
{
if (TREE_CODE (expr) == COMPONENT_REF)
last_component = expr;
expr = TREE_OPERAND (expr, 0);
if (TREE_CODE (TREE_TYPE (expr)) == REFERENCE_TYPE)
break;
}
if (!last_component)
return false;
*expr0 = last_component;
return true;
}
/* This handles references that have had convert_from_reference called on
them, and also those that haven't. */
static bool
omp_parse_ref (tree *expr0)
{
tree expr = *expr0;
if (TREE_CODE (TREE_TYPE (expr)) == REFERENCE_TYPE)
return true;
else if ((TREE_CODE (expr) == INDIRECT_REF
|| (TREE_CODE (expr) == MEM_REF
&& integer_zerop (TREE_OPERAND (expr, 1))))
&& TREE_CODE (TREE_TYPE (TREE_OPERAND (expr, 0))) == REFERENCE_TYPE)
{
*expr0 = TREE_OPERAND (expr, 0);
return true;
}
return false;
}
static bool
omp_parse_pointer (tree *expr0, bool *has_offset)
{
tree expr = *expr0;
*has_offset = false;
if ((TREE_CODE (expr) == INDIRECT_REF
|| (TREE_CODE (expr) == MEM_REF
&& integer_zerop (TREE_OPERAND (expr, 1))))
&& TREE_CODE (TREE_TYPE (TREE_OPERAND (expr, 0))) == POINTER_TYPE)
{
expr = TREE_OPERAND (expr, 0);
/* The Fortran FE sometimes emits a no-op cast here. */
STRIP_NOPS (expr);
while (1)
{
if (TREE_CODE (expr) == COMPOUND_EXPR)
{
expr = TREE_OPERAND (expr, 1);
STRIP_NOPS (expr);
}
else if (TREE_CODE (expr) == SAVE_EXPR)
expr = TREE_OPERAND (expr, 0);
else if (TREE_CODE (expr) == POINTER_PLUS_EXPR)
{
*has_offset = true;
expr = TREE_OPERAND (expr, 0);
}
else
break;
}
STRIP_NOPS (expr);
*expr0 = expr;
return true;
}
return false;
}
static bool
omp_parse_access_method (tree *expr0, enum access_method_kinds *kind)
{
tree expr = *expr0;
bool has_offset;
if (omp_parse_ref (&expr))
*kind = ACCESS_REF;
else if (omp_parse_pointer (&expr, &has_offset))
{
if (omp_parse_ref (&expr))
*kind = has_offset ? ACCESS_REF_TO_POINTER_OFFSET
: ACCESS_REF_TO_POINTER;
else
*kind = has_offset ? ACCESS_POINTER_OFFSET : ACCESS_POINTER;
}
else if (TREE_CODE (expr) == ARRAY_REF)
{
while (TREE_CODE (expr) == ARRAY_REF)
expr = TREE_OPERAND (expr, 0);
if (omp_parse_ref (&expr))
*kind = ACCESS_INDEXED_REF_TO_ARRAY;
else
*kind = ACCESS_INDEXED_ARRAY;
}
else
*kind = ACCESS_DIRECT;
STRIP_NOPS (expr);
*expr0 = expr;
return true;
}
static bool
omp_parse_access_methods (vec<omp_addr_token *> &addr_tokens, tree *expr0)
{
tree expr = *expr0;
enum access_method_kinds kind;
tree am_expr;
if (omp_parse_access_method (&expr, &kind))
am_expr = expr;
if (TREE_CODE (expr) == INDIRECT_REF
|| TREE_CODE (expr) == MEM_REF
|| TREE_CODE (expr) == ARRAY_REF)
omp_parse_access_methods (addr_tokens, &expr);
addr_tokens.safe_push (new omp_addr_token (kind, am_expr));
*expr0 = expr;
return true;
}
static bool omp_parse_structured_expr (vec<omp_addr_token *> &, tree *);
static bool
omp_parse_structure_base (vec<omp_addr_token *> &addr_tokens,
tree *expr0, structure_base_kinds *kind,
vec<omp_addr_token *> &base_access_tokens,
bool allow_structured = true)
{
tree expr = *expr0;
if (allow_structured)
omp_parse_access_methods (base_access_tokens, &expr);
if (DECL_P (expr))
{
*kind = BASE_DECL;
return true;
}
if (allow_structured && omp_parse_structured_expr (addr_tokens, &expr))
{
*kind = BASE_COMPONENT_EXPR;
*expr0 = expr;
return true;
}
*kind = BASE_ARBITRARY_EXPR;
*expr0 = expr;
return true;
}
static bool
omp_parse_structured_expr (vec<omp_addr_token *> &addr_tokens, tree *expr0)
{
tree expr = *expr0;
tree base_component = NULL_TREE;
structure_base_kinds struct_base_kind;
auto_vec<omp_addr_token *> base_access_tokens;
if (omp_parse_component_selector (&expr))
base_component = expr;
else
return false;
gcc_assert (TREE_CODE (expr) == COMPONENT_REF);
expr = TREE_OPERAND (expr, 0);
tree structure_base = expr;
if (!omp_parse_structure_base (addr_tokens, &expr, &struct_base_kind,
base_access_tokens))
return false;
addr_tokens.safe_push (new omp_addr_token (STRUCTURE_BASE, struct_base_kind,
structure_base));
addr_tokens.safe_splice (base_access_tokens);
addr_tokens.safe_push (new omp_addr_token (COMPONENT_SELECTOR,
base_component));
*expr0 = expr;
return true;
}
static bool
omp_parse_array_expr (vec<omp_addr_token *> &addr_tokens, tree *expr0)
{
tree expr = *expr0;
structure_base_kinds s_kind;
auto_vec<omp_addr_token *> base_access_tokens;
if (!omp_parse_structure_base (addr_tokens, &expr, &s_kind,
base_access_tokens, false))
return false;
addr_tokens.safe_push (new omp_addr_token (ARRAY_BASE, s_kind, expr));
addr_tokens.safe_splice (base_access_tokens);
*expr0 = expr;
return true;
}
/* Return TRUE if the ACCESS_METHOD token at index 'i' has a further
ACCESS_METHOD chained after it (e.g., if we're processing an expression
containing multiple pointer indirections). */
bool
omp_access_chain_p (vec<omp_addr_token *> &addr_tokens, unsigned i)
{
gcc_assert (addr_tokens[i]->type == ACCESS_METHOD);
return (i + 1 < addr_tokens.length ()
&& addr_tokens[i + 1]->type == ACCESS_METHOD);
}
/* Return the address of the object accessed by the ACCESS_METHOD token
at 'i': either of the next access method's expr, or of EXPR if we're at
the end of the list of tokens. */
tree
omp_accessed_addr (vec<omp_addr_token *> &addr_tokens, unsigned i, tree expr)
{
if (i + 1 < addr_tokens.length ())
return build_fold_addr_expr (addr_tokens[i + 1]->expr);
else
return build_fold_addr_expr (expr);
}
} /* namespace omp_addr_tokenizer. */
bool
omp_parse_expr (vec<omp_addr_token *> &addr_tokens, tree expr)
{
using namespace omp_addr_tokenizer;
auto_vec<omp_addr_token *> expr_access_tokens;
if (!omp_parse_access_methods (expr_access_tokens, &expr))
return false;
if (omp_parse_structured_expr (addr_tokens, &expr))
;
else if (omp_parse_array_expr (addr_tokens, &expr))
;
else
return false;
addr_tokens.safe_splice (expr_access_tokens);
return true;
}
DEBUG_FUNCTION void
debug_omp_tokenized_addr (vec<omp_addr_token *> &addr_tokens,
bool with_exprs)
{
using namespace omp_addr_tokenizer;
const char *sep = with_exprs ? " " : "";
for (auto e : addr_tokens)
{
const char *pfx = "";
fputs (sep, stderr);
switch (e->type)
{
case COMPONENT_SELECTOR:
fputs ("component_selector", stderr);
break;
case ACCESS_METHOD:
switch (e->u.access_kind)
{
case ACCESS_DIRECT:
fputs ("access_direct", stderr);
break;
case ACCESS_REF:
fputs ("access_ref", stderr);
break;
case ACCESS_POINTER:
fputs ("access_pointer", stderr);
break;
case ACCESS_POINTER_OFFSET:
fputs ("access_pointer_offset", stderr);
break;
case ACCESS_REF_TO_POINTER:
fputs ("access_ref_to_pointer", stderr);
break;
case ACCESS_REF_TO_POINTER_OFFSET:
fputs ("access_ref_to_pointer_offset", stderr);
break;
case ACCESS_INDEXED_ARRAY:
fputs ("access_indexed_array", stderr);
break;
case ACCESS_INDEXED_REF_TO_ARRAY:
fputs ("access_indexed_ref_to_array", stderr);
break;
}
break;
case ARRAY_BASE:
case STRUCTURE_BASE:
pfx = e->type == ARRAY_BASE ? "array_" : "struct_";
switch (e->u.structure_base_kind)
{
case BASE_DECL:
fprintf (stderr, "%sbase_decl", pfx);
break;
case BASE_COMPONENT_EXPR:
fputs ("base_component_expr", stderr);
break;
case BASE_ARBITRARY_EXPR:
fprintf (stderr, "%sbase_arbitrary_expr", pfx);
break;
}
break;
}
if (with_exprs)
{
fputs (" [", stderr);
print_generic_expr (stderr, e->expr);
fputc (']', stderr);
sep = ",\n ";
}
else
sep = " ";
}
fputs ("\n", stderr);
}
#include "gt-omp-general.h"