Google Git
Sign in
gnu / gcc / a8404c07e7fca388c02c39077865f7d5fa928430 / . / gcc / tree-vect-patterns.cc
blob: f6c34bb32631cdb7885dbcae18b1b79836cc17fe [file] [log] [blame]
/* Analysis Utilities for Loop Vectorization.
Copyright (C) 2006-2022 Free Software Foundation, Inc.
Contributed by Dorit Nuzman <dorit@il.ibm.com>
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 "rtl.h"
#include "tree.h"
#include "gimple.h"
#include "ssa.h"
#include "expmed.h"
#include "optabs-tree.h"
#include "insn-config.h"
#include "recog.h" /* FIXME: for insn_data */
#include "fold-const.h"
#include "stor-layout.h"
#include "tree-eh.h"
#include "gimplify.h"
#include "gimple-iterator.h"
#include "gimple-fold.h"
#include "gimplify-me.h"
#include "cfgloop.h"
#include "tree-vectorizer.h"
#include "dumpfile.h"
#include "builtins.h"
#include "internal-fn.h"
#include "case-cfn-macros.h"
#include "fold-const-call.h"
#include "attribs.h"
#include "cgraph.h"
#include "omp-simd-clone.h"
#include "predict.h"
#include "tree-vector-builder.h"
#include "vec-perm-indices.h"
#include "gimple-range.h"
/* TODO: Note the vectorizer still builds COND_EXPRs with GENERIC compares
in the first operand. Disentangling this is future work, the
IL is properly transfered to VEC_COND_EXPRs with separate compares. */
/* Return true if we have a useful VR_RANGE range for VAR, storing it
in *MIN_VALUE and *MAX_VALUE if so. Note the range in the dump files. */
static bool
vect_get_range_info (tree var, wide_int *min_value, wide_int *max_value)
{
value_range vr;
get_range_query (cfun)->range_of_expr (vr, var);
if (vr.undefined_p ())
vr.set_varying (TREE_TYPE (var));
*min_value = wi::to_wide (vr.min ());
*max_value = wi::to_wide (vr.max ());
value_range_kind vr_type = vr.kind ();
wide_int nonzero = get_nonzero_bits (var);
signop sgn = TYPE_SIGN (TREE_TYPE (var));
if (intersect_range_with_nonzero_bits (vr_type, min_value, max_value,
nonzero, sgn) == VR_RANGE)
{
if (dump_enabled_p ())
{
dump_generic_expr_loc (MSG_NOTE, vect_location, TDF_SLIM, var);
dump_printf (MSG_NOTE, " has range [");
dump_hex (MSG_NOTE, *min_value);
dump_printf (MSG_NOTE, ", ");
dump_hex (MSG_NOTE, *max_value);
dump_printf (MSG_NOTE, "]\n");
}
return true;
}
else
{
if (dump_enabled_p ())
{
dump_generic_expr_loc (MSG_NOTE, vect_location, TDF_SLIM, var);
dump_printf (MSG_NOTE, " has no range info\n");
}
return false;
}
}
/* Report that we've found an instance of pattern PATTERN in
statement STMT. */
static void
vect_pattern_detected (const char *name, gimple *stmt)
{
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location, "%s: detected: %G", name, stmt);
}
/* Associate pattern statement PATTERN_STMT with ORIG_STMT_INFO and
return the pattern statement's stmt_vec_info. Set its vector type to
VECTYPE if it doesn't have one already. */
static stmt_vec_info
vect_init_pattern_stmt (vec_info *vinfo, gimple *pattern_stmt,
stmt_vec_info orig_stmt_info, tree vectype)
{
stmt_vec_info pattern_stmt_info = vinfo->lookup_stmt (pattern_stmt);
if (pattern_stmt_info == NULL)
pattern_stmt_info = vinfo->add_stmt (pattern_stmt);
gimple_set_bb (pattern_stmt, gimple_bb (orig_stmt_info->stmt));
pattern_stmt_info->pattern_stmt_p = true;
STMT_VINFO_RELATED_STMT (pattern_stmt_info) = orig_stmt_info;
STMT_VINFO_DEF_TYPE (pattern_stmt_info)
= STMT_VINFO_DEF_TYPE (orig_stmt_info);
if (!STMT_VINFO_VECTYPE (pattern_stmt_info))
{
gcc_assert (!vectype
|| (VECTOR_BOOLEAN_TYPE_P (vectype)
== vect_use_mask_type_p (orig_stmt_info)));
STMT_VINFO_VECTYPE (pattern_stmt_info) = vectype;
pattern_stmt_info->mask_precision = orig_stmt_info->mask_precision;
}
return pattern_stmt_info;
}
/* Set the pattern statement of ORIG_STMT_INFO to PATTERN_STMT.
Also set the vector type of PATTERN_STMT to VECTYPE, if it doesn't
have one already. */
static void
vect_set_pattern_stmt (vec_info *vinfo, gimple *pattern_stmt,
stmt_vec_info orig_stmt_info, tree vectype)
{
STMT_VINFO_IN_PATTERN_P (orig_stmt_info) = true;
STMT_VINFO_RELATED_STMT (orig_stmt_info)
= vect_init_pattern_stmt (vinfo, pattern_stmt, orig_stmt_info, vectype);
}
/* Add NEW_STMT to STMT_INFO's pattern definition statements. If VECTYPE
is nonnull, record that NEW_STMT's vector type is VECTYPE, which might
be different from the vector type of the final pattern statement.
If VECTYPE is a mask type, SCALAR_TYPE_FOR_MASK is the scalar type
from which it was derived. */
static inline void
append_pattern_def_seq (vec_info *vinfo,
stmt_vec_info stmt_info, gimple *new_stmt,
tree vectype = NULL_TREE,
tree scalar_type_for_mask = NULL_TREE)
{
gcc_assert (!scalar_type_for_mask
== (!vectype || !VECTOR_BOOLEAN_TYPE_P (vectype)));
if (vectype)
{
stmt_vec_info new_stmt_info = vinfo->add_stmt (new_stmt);
STMT_VINFO_VECTYPE (new_stmt_info) = vectype;
if (scalar_type_for_mask)
new_stmt_info->mask_precision
= GET_MODE_BITSIZE (SCALAR_TYPE_MODE (scalar_type_for_mask));
}
gimple_seq_add_stmt_without_update (&STMT_VINFO_PATTERN_DEF_SEQ (stmt_info),
new_stmt);
}
/* The caller wants to perform new operations on vect_external variable
VAR, so that the result of the operations would also be vect_external.
Return the edge on which the operations can be performed, if one exists.
Return null if the operations should instead be treated as part of
the pattern that needs them. */
static edge
vect_get_external_def_edge (vec_info *vinfo, tree var)
{
edge e = NULL;
if (loop_vec_info loop_vinfo = dyn_cast <loop_vec_info> (vinfo))
{
e = loop_preheader_edge (loop_vinfo->loop);
if (!SSA_NAME_IS_DEFAULT_DEF (var))
{
basic_block bb = gimple_bb (SSA_NAME_DEF_STMT (var));
if (bb == NULL
|| !dominated_by_p (CDI_DOMINATORS, e->dest, bb))
e = NULL;
}
}
return e;
}
/* Return true if the target supports a vector version of CODE,
where CODE is known to map to a direct optab with the given SUBTYPE.
ITYPE specifies the type of (some of) the scalar inputs and OTYPE
specifies the type of the scalar result.
If CODE allows the inputs and outputs to have different type
(such as for WIDEN_SUM_EXPR), it is the input mode rather
than the output mode that determines the appropriate target pattern.
Operand 0 of the target pattern then specifies the mode that the output
must have.
When returning true, set *VECOTYPE_OUT to the vector version of OTYPE.
Also set *VECITYPE_OUT to the vector version of ITYPE if VECITYPE_OUT
is nonnull. */
static bool
vect_supportable_direct_optab_p (vec_info *vinfo, tree otype, tree_code code,
tree itype, tree *vecotype_out,
tree *vecitype_out = NULL,
enum optab_subtype subtype = optab_default)
{
tree vecitype = get_vectype_for_scalar_type (vinfo, itype);
if (!vecitype)
return false;
tree vecotype = get_vectype_for_scalar_type (vinfo, otype);
if (!vecotype)
return false;
optab optab = optab_for_tree_code (code, vecitype, subtype);
if (!optab)
return false;
insn_code icode = optab_handler (optab, TYPE_MODE (vecitype));
if (icode == CODE_FOR_nothing
|| insn_data[icode].operand[0].mode != TYPE_MODE (vecotype))
return false;
*vecotype_out = vecotype;
if (vecitype_out)
*vecitype_out = vecitype;
return true;
}
/* Round bit precision PRECISION up to a full element. */
static unsigned int
vect_element_precision (unsigned int precision)
{
precision = 1 << ceil_log2 (precision);
return MAX (precision, BITS_PER_UNIT);
}
/* If OP is defined by a statement that's being considered for vectorization,
return information about that statement, otherwise return NULL. */
static stmt_vec_info
vect_get_internal_def (vec_info *vinfo, tree op)
{
stmt_vec_info def_stmt_info = vinfo->lookup_def (op);
if (def_stmt_info
&& STMT_VINFO_DEF_TYPE (def_stmt_info) == vect_internal_def)
return def_stmt_info;
return NULL;
}
/* Check whether NAME, an ssa-name used in STMT_VINFO,
is a result of a type promotion, such that:
DEF_STMT: NAME = NOP (name0)
If CHECK_SIGN is TRUE, check that either both types are signed or both are
unsigned. */
static bool
type_conversion_p (vec_info *vinfo, tree name, bool check_sign,
tree *orig_type, gimple **def_stmt, bool *promotion)
{
tree type = TREE_TYPE (name);
tree oprnd0;
enum vect_def_type dt;
stmt_vec_info def_stmt_info;
if (!vect_is_simple_use (name, vinfo, &dt, &def_stmt_info, def_stmt))
return false;
if (dt != vect_internal_def
&& dt != vect_external_def && dt != vect_constant_def)
return false;
if (!*def_stmt)
return false;
if (!is_gimple_assign (*def_stmt))
return false;
if (!CONVERT_EXPR_CODE_P (gimple_assign_rhs_code (*def_stmt)))
return false;
oprnd0 = gimple_assign_rhs1 (*def_stmt);
*orig_type = TREE_TYPE (oprnd0);
if (!INTEGRAL_TYPE_P (type) || !INTEGRAL_TYPE_P (*orig_type)
|| ((TYPE_UNSIGNED (type) != TYPE_UNSIGNED (*orig_type)) && check_sign))
return false;
if (TYPE_PRECISION (type) >= (TYPE_PRECISION (*orig_type) * 2))
*promotion = true;
else
*promotion = false;
if (!vect_is_simple_use (oprnd0, vinfo, &dt))
return false;
return true;
}
/* Holds information about an input operand after some sign changes
and type promotions have been peeled away. */
class vect_unpromoted_value {
public:
vect_unpromoted_value ();
void set_op (tree, vect_def_type, stmt_vec_info = NULL);
/* The value obtained after peeling away zero or more casts. */
tree op;
/* The type of OP. */
tree type;
/* The definition type of OP. */
vect_def_type dt;
/* If OP is the result of peeling at least one cast, and if the cast
of OP itself is a vectorizable statement, CASTER identifies that
statement, otherwise it is null. */
stmt_vec_info caster;
};
inline vect_unpromoted_value::vect_unpromoted_value ()
: op (NULL_TREE),
type (NULL_TREE),
dt (vect_uninitialized_def),
caster (NULL)
{
}
/* Set the operand to OP_IN, its definition type to DT_IN, and the
statement that casts it to CASTER_IN. */
inline void
vect_unpromoted_value::set_op (tree op_in, vect_def_type dt_in,
stmt_vec_info caster_in)
{
op = op_in;
type = TREE_TYPE (op);
dt = dt_in;
caster = caster_in;
}
/* If OP is a vectorizable SSA name, strip a sequence of integer conversions
to reach some vectorizable inner operand OP', continuing as long as it
is possible to convert OP' back to OP using a possible sign change
followed by a possible promotion P. Return this OP', or null if OP is
not a vectorizable SSA name. If there is a promotion P, describe its
input in UNPROM, otherwise describe OP' in UNPROM. If SINGLE_USE_P
is nonnull, set *SINGLE_USE_P to false if any of the SSA names involved
have more than one user.
A successful return means that it is possible to go from OP' to OP
via UNPROM. The cast from OP' to UNPROM is at most a sign change,
whereas the cast from UNPROM to OP might be a promotion, a sign
change, or a nop.
E.g. say we have:
signed short *ptr = ...;
signed short C = *ptr;
unsigned short B = (unsigned short) C; // sign change
signed int A = (signed int) B; // unsigned promotion
...possible other uses of A...
unsigned int OP = (unsigned int) A; // sign change
In this case it's possible to go directly from C to OP using:
OP = (unsigned int) (unsigned short) C;
+------------+ +--------------+
promotion sign change
so OP' would be C. The input to the promotion is B, so UNPROM
would describe B. */
static tree
vect_look_through_possible_promotion (vec_info *vinfo, tree op,
vect_unpromoted_value *unprom,
bool *single_use_p = NULL)
{
tree res = NULL_TREE;
tree op_type = TREE_TYPE (op);
unsigned int orig_precision = TYPE_PRECISION (op_type);
unsigned int min_precision = orig_precision;
stmt_vec_info caster = NULL;
while (TREE_CODE (op) == SSA_NAME && INTEGRAL_TYPE_P (op_type))
{
/* See whether OP is simple enough to vectorize. */
stmt_vec_info def_stmt_info;
gimple *def_stmt;
vect_def_type dt;
if (!vect_is_simple_use (op, vinfo, &dt, &def_stmt_info, &def_stmt))
break;
/* If OP is the input of a demotion, skip over it to see whether
OP is itself the result of a promotion. If so, the combined
effect of the promotion and the demotion might fit the required
pattern, otherwise neither operation fits.
This copes with cases such as the result of an arithmetic
operation being truncated before being stored, and where that
arithmetic operation has been recognized as an over-widened one. */
if (TYPE_PRECISION (op_type) <= min_precision)
{
/* Use OP as the UNPROM described above if we haven't yet
found a promotion, or if using the new input preserves the
sign of the previous promotion. */
if (!res
|| TYPE_PRECISION (unprom->type) == orig_precision
|| TYPE_SIGN (unprom->type) == TYPE_SIGN (op_type))
{
unprom->set_op (op, dt, caster);
min_precision = TYPE_PRECISION (op_type);
}
/* Stop if we've already seen a promotion and if this
conversion does more than change the sign. */
else if (TYPE_PRECISION (op_type)
!= TYPE_PRECISION (unprom->type))
break;
/* The sequence now extends to OP. */
res = op;
}
/* See whether OP is defined by a cast. Record it as CASTER if
the cast is potentially vectorizable. */
if (!def_stmt)
break;
caster = def_stmt_info;
/* Ignore pattern statements, since we don't link uses for them. */
if (caster
&& single_use_p
&& !STMT_VINFO_RELATED_STMT (caster)
&& !has_single_use (res))
*single_use_p = false;
gassign *assign = dyn_cast <gassign *> (def_stmt);
if (!assign || !CONVERT_EXPR_CODE_P (gimple_assign_rhs_code (def_stmt)))
break;
/* Continue with the input to the cast. */
op = gimple_assign_rhs1 (def_stmt);
op_type = TREE_TYPE (op);
}
return res;
}
/* OP is an integer operand to an operation that returns TYPE, and we
want to treat the operation as a widening one. So far we can treat
it as widening from *COMMON_TYPE.
Return true if OP is suitable for such a widening operation,
either widening from *COMMON_TYPE or from some supertype of it.
Update *COMMON_TYPE to the supertype in the latter case.
SHIFT_P is true if OP is a shift amount. */
static bool
vect_joust_widened_integer (tree type, bool shift_p, tree op,
tree *common_type)
{
/* Calculate the minimum precision required by OP, without changing
the sign of either operand. */
unsigned int precision;
if (shift_p)
{
if (!wi::leu_p (wi::to_widest (op), TYPE_PRECISION (type) / 2))
return false;
precision = TREE_INT_CST_LOW (op);
}
else
{
precision = wi::min_precision (wi::to_widest (op),
TYPE_SIGN (*common_type));
if (precision * 2 > TYPE_PRECISION (type))
return false;
}
/* If OP requires a wider type, switch to that type. The checks
above ensure that this is still narrower than the result. */
precision = vect_element_precision (precision);
if (TYPE_PRECISION (*common_type) < precision)
*common_type = build_nonstandard_integer_type
(precision, TYPE_UNSIGNED (*common_type));
return true;
}
/* Return true if the common supertype of NEW_TYPE and *COMMON_TYPE
is narrower than type, storing the supertype in *COMMON_TYPE if so. */
static bool
vect_joust_widened_type (tree type, tree new_type, tree *common_type)
{
if (types_compatible_p (*common_type, new_type))
return true;
/* See if *COMMON_TYPE can hold all values of NEW_TYPE. */
if ((TYPE_PRECISION (new_type) < TYPE_PRECISION (*common_type))
&& (TYPE_UNSIGNED (new_type) || !TYPE_UNSIGNED (*common_type)))
return true;
/* See if NEW_TYPE can hold all values of *COMMON_TYPE. */
if (TYPE_PRECISION (*common_type) < TYPE_PRECISION (new_type)
&& (TYPE_UNSIGNED (*common_type) || !TYPE_UNSIGNED (new_type)))
{
*common_type = new_type;
return true;
}
/* We have mismatched signs, with the signed type being
no wider than the unsigned type. In this case we need
a wider signed type. */
unsigned int precision = MAX (TYPE_PRECISION (*common_type),
TYPE_PRECISION (new_type));
precision *= 2;
if (precision * 2 > TYPE_PRECISION (type))
return false;
*common_type = build_nonstandard_integer_type (precision, false);
return true;
}
/* Check whether STMT_INFO can be viewed as a tree of integer operations
in which each node either performs CODE or WIDENED_CODE, and where
each leaf operand is narrower than the result of STMT_INFO. MAX_NOPS
specifies the maximum number of leaf operands. SHIFT_P says whether
CODE and WIDENED_CODE are some sort of shift.
If STMT_INFO is such a tree, return the number of leaf operands
and describe them in UNPROM[0] onwards. Also set *COMMON_TYPE
to a type that (a) is narrower than the result of STMT_INFO and
(b) can hold all leaf operand values.
If SUBTYPE then allow that the signs of the operands
may differ in signs but not in precision. SUBTYPE is updated to reflect
this.
Return 0 if STMT_INFO isn't such a tree, or if no such COMMON_TYPE
exists. */
static unsigned int
vect_widened_op_tree (vec_info *vinfo, stmt_vec_info stmt_info, tree_code code,
tree_code widened_code, bool shift_p,
unsigned int max_nops,
vect_unpromoted_value *unprom, tree *common_type,
enum optab_subtype *subtype = NULL)
{
/* Check for an integer operation with the right code. */
gassign *assign = dyn_cast <gassign *> (stmt_info->stmt);
if (!assign)
return 0;
tree_code rhs_code = gimple_assign_rhs_code (assign);
if (rhs_code != code && rhs_code != widened_code)
return 0;
tree type = TREE_TYPE (gimple_assign_lhs (assign));
if (!INTEGRAL_TYPE_P (type))
return 0;
/* Assume that both operands will be leaf operands. */
max_nops -= 2;
/* Check the operands. */
unsigned int next_op = 0;
for (unsigned int i = 0; i < 2; ++i)
{
vect_unpromoted_value *this_unprom = &unprom[next_op];
unsigned int nops = 1;
tree op = gimple_op (assign, i + 1);
if (i == 1 && TREE_CODE (op) == INTEGER_CST)
{
/* We already have a common type from earlier operands.
Update it to account for OP. */
this_unprom->set_op (op, vect_constant_def);
if (!vect_joust_widened_integer (type, shift_p, op, common_type))
return 0;
}
else
{
/* Only allow shifts by constants. */
if (shift_p && i == 1)
return 0;
if (!vect_look_through_possible_promotion (vinfo, op, this_unprom))
return 0;
if (TYPE_PRECISION (this_unprom->type) == TYPE_PRECISION (type))
{
/* The operand isn't widened. If STMT_INFO has the code
for an unwidened operation, recursively check whether
this operand is a node of the tree. */
if (rhs_code != code
|| max_nops == 0
|| this_unprom->dt != vect_internal_def)
return 0;
/* Give back the leaf slot allocated above now that we're
not treating this as a leaf operand. */
max_nops += 1;
/* Recursively process the definition of the operand. */
stmt_vec_info def_stmt_info
= vinfo->lookup_def (this_unprom->op);
nops = vect_widened_op_tree (vinfo, def_stmt_info, code,
widened_code, shift_p, max_nops,
this_unprom, common_type,
subtype);
if (nops == 0)
return 0;
max_nops -= nops;
}
else
{
/* Make sure that the operand is narrower than the result. */
if (TYPE_PRECISION (this_unprom->type) * 2
> TYPE_PRECISION (type))
return 0;
/* Update COMMON_TYPE for the new operand. */
if (i == 0)
*common_type = this_unprom->type;
else if (!vect_joust_widened_type (type, this_unprom->type,
common_type))
{
if (subtype)
{
/* See if we can sign extend the smaller type. */
if (TYPE_PRECISION (this_unprom->type)
> TYPE_PRECISION (*common_type))
*common_type = this_unprom->type;
*subtype = optab_vector_mixed_sign;
}
else
return 0;
}
}
}
next_op += nops;
}
return next_op;
}
/* Helper to return a new temporary for pattern of TYPE for STMT. If STMT
is NULL, the caller must set SSA_NAME_DEF_STMT for the returned SSA var. */
static tree
vect_recog_temp_ssa_var (tree type, gimple *stmt = NULL)
{
return make_temp_ssa_name (type, stmt, "patt");
}
/* STMT2_INFO describes a type conversion that could be split into STMT1
followed by a version of STMT2_INFO that takes NEW_RHS as its first
input. Try to do this using pattern statements, returning true on
success. */
static bool
vect_split_statement (vec_info *vinfo, stmt_vec_info stmt2_info, tree new_rhs,
gimple *stmt1, tree vectype)
{
if (is_pattern_stmt_p (stmt2_info))
{
/* STMT2_INFO is part of a pattern. Get the statement to which
the pattern is attached. */
stmt_vec_info orig_stmt2_info = STMT_VINFO_RELATED_STMT (stmt2_info);
vect_init_pattern_stmt (vinfo, stmt1, orig_stmt2_info, vectype);
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location,
"Splitting pattern statement: %G", stmt2_info->stmt);
/* Since STMT2_INFO is a pattern statement, we can change it
in-situ without worrying about changing the code for the
containing block. */
gimple_assign_set_rhs1 (stmt2_info->stmt, new_rhs);
if (dump_enabled_p ())
{
dump_printf_loc (MSG_NOTE, vect_location, "into: %G", stmt1);
dump_printf_loc (MSG_NOTE, vect_location, "and: %G",
stmt2_info->stmt);
}
gimple_seq *def_seq = &STMT_VINFO_PATTERN_DEF_SEQ (orig_stmt2_info);
if (STMT_VINFO_RELATED_STMT (orig_stmt2_info) == stmt2_info)
/* STMT2_INFO is the actual pattern statement. Add STMT1
to the end of the definition sequence. */
gimple_seq_add_stmt_without_update (def_seq, stmt1);
else
{
/* STMT2_INFO belongs to the definition sequence. Insert STMT1
before it. */
gimple_stmt_iterator gsi = gsi_for_stmt (stmt2_info->stmt, def_seq);
gsi_insert_before_without_update (&gsi, stmt1, GSI_SAME_STMT);
}
return true;
}
else
{
/* STMT2_INFO doesn't yet have a pattern. Try to create a
two-statement pattern now. */
gcc_assert (!STMT_VINFO_RELATED_STMT (stmt2_info));
tree lhs_type = TREE_TYPE (gimple_get_lhs (stmt2_info->stmt));
tree lhs_vectype = get_vectype_for_scalar_type (vinfo, lhs_type);
if (!lhs_vectype)
return false;
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location,
"Splitting statement: %G", stmt2_info->stmt);
/* Add STMT1 as a singleton pattern definition sequence. */
gimple_seq *def_seq = &STMT_VINFO_PATTERN_DEF_SEQ (stmt2_info);
vect_init_pattern_stmt (vinfo, stmt1, stmt2_info, vectype);
gimple_seq_add_stmt_without_update (def_seq, stmt1);
/* Build the second of the two pattern statements. */
tree new_lhs = vect_recog_temp_ssa_var (lhs_type, NULL);
gassign *new_stmt2 = gimple_build_assign (new_lhs, NOP_EXPR, new_rhs);
vect_set_pattern_stmt (vinfo, new_stmt2, stmt2_info, lhs_vectype);
if (dump_enabled_p ())
{
dump_printf_loc (MSG_NOTE, vect_location,
"into pattern statements: %G", stmt1);
dump_printf_loc (MSG_NOTE, vect_location, "and: %G",
(gimple *) new_stmt2);
}
return true;
}
}
/* Convert UNPROM to TYPE and return the result, adding new statements
to STMT_INFO's pattern definition statements if no better way is
available. VECTYPE is the vector form of TYPE.
If SUBTYPE then convert the type based on the subtype. */
static tree
vect_convert_input (vec_info *vinfo, stmt_vec_info stmt_info, tree type,
vect_unpromoted_value *unprom, tree vectype,
enum optab_subtype subtype = optab_default)
{
/* Update the type if the signs differ. */
if (subtype == optab_vector_mixed_sign)
{
gcc_assert (!TYPE_UNSIGNED (type));
if (TYPE_UNSIGNED (TREE_TYPE (unprom->op)))
{
type = unsigned_type_for (type);
vectype = unsigned_type_for (vectype);
}
}
/* Check for a no-op conversion. */
if (types_compatible_p (type, TREE_TYPE (unprom->op)))
return unprom->op;
/* Allow the caller to create constant vect_unpromoted_values. */
if (TREE_CODE (unprom->op) == INTEGER_CST)
return wide_int_to_tree (type, wi::to_widest (unprom->op));
tree input = unprom->op;
if (unprom->caster)
{
tree lhs = gimple_get_lhs (unprom->caster->stmt);
tree lhs_type = TREE_TYPE (lhs);
/* If the result of the existing cast is the right width, use it
instead of the source of the cast. */
if (TYPE_PRECISION (lhs_type) == TYPE_PRECISION (type))
input = lhs;
/* If the precision we want is between the source and result
precisions of the existing cast, try splitting the cast into
two and tapping into a mid-way point. */
else if (TYPE_PRECISION (lhs_type) > TYPE_PRECISION (type)
&& TYPE_PRECISION (type) > TYPE_PRECISION (unprom->type))
{
/* In order to preserve the semantics of the original cast,
give the mid-way point the same signedness as the input value.
It would be possible to use a signed type here instead if
TYPE is signed and UNPROM->TYPE is unsigned, but that would
make the sign of the midtype sensitive to the order in
which we process the statements, since the signedness of
TYPE is the signedness required by just one of possibly
many users. Also, unsigned promotions are usually as cheap
as or cheaper than signed ones, so it's better to keep an
unsigned promotion. */
tree midtype = build_nonstandard_integer_type
(TYPE_PRECISION (type), TYPE_UNSIGNED (unprom->type));
tree vec_midtype = get_vectype_for_scalar_type (vinfo, midtype);
if (vec_midtype)
{
input = vect_recog_temp_ssa_var (midtype, NULL);
gassign *new_stmt = gimple_build_assign (input, NOP_EXPR,
unprom->op);
if (!vect_split_statement (vinfo, unprom->caster, input, new_stmt,
vec_midtype))
append_pattern_def_seq (vinfo, stmt_info,
new_stmt, vec_midtype);
}
}
/* See if we can reuse an existing result. */
if (types_compatible_p (type, TREE_TYPE (input)))
return input;
}
/* We need a new conversion statement. */
tree new_op = vect_recog_temp_ssa_var (type, NULL);
gassign *new_stmt = gimple_build_assign (new_op, NOP_EXPR, input);
/* If OP is an external value, see if we can insert the new statement
on an incoming edge. */
if (input == unprom->op && unprom->dt == vect_external_def)
if (edge e = vect_get_external_def_edge (vinfo, input))
{
basic_block new_bb = gsi_insert_on_edge_immediate (e, new_stmt);
gcc_assert (!new_bb);
return new_op;
}
/* As a (common) last resort, add the statement to the pattern itself. */
append_pattern_def_seq (vinfo, stmt_info, new_stmt, vectype);
return new_op;
}
/* Invoke vect_convert_input for N elements of UNPROM and store the
result in the corresponding elements of RESULT.
If SUBTYPE then convert the type based on the subtype. */
static void
vect_convert_inputs (vec_info *vinfo, stmt_vec_info stmt_info, unsigned int n,
tree *result, tree type, vect_unpromoted_value *unprom,
tree vectype, enum optab_subtype subtype = optab_default)
{
for (unsigned int i = 0; i < n; ++i)
{
unsigned int j;
for (j = 0; j < i; ++j)
if (unprom[j].op == unprom[i].op)
break;
if (j < i)
result[i] = result[j];
else
result[i] = vect_convert_input (vinfo, stmt_info,
type, &unprom[i], vectype, subtype);
}
}
/* The caller has created a (possibly empty) sequence of pattern definition
statements followed by a single statement PATTERN_STMT. Cast the result
of this final statement to TYPE. If a new statement is needed, add
PATTERN_STMT to the end of STMT_INFO's pattern definition statements
and return the new statement, otherwise return PATTERN_STMT as-is.
VECITYPE is the vector form of PATTERN_STMT's result type. */
static gimple *
vect_convert_output (vec_info *vinfo, stmt_vec_info stmt_info, tree type,
gimple *pattern_stmt, tree vecitype)
{
tree lhs = gimple_get_lhs (pattern_stmt);
if (!types_compatible_p (type, TREE_TYPE (lhs)))
{
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt, vecitype);
tree cast_var = vect_recog_temp_ssa_var (type, NULL);
pattern_stmt = gimple_build_assign (cast_var, NOP_EXPR, lhs);
}
return pattern_stmt;
}
/* Return true if STMT_VINFO describes a reduction for which reassociation
is allowed. If STMT_INFO is part of a group, assume that it's part of
a reduction chain and optimistically assume that all statements
except the last allow reassociation.
Also require it to have code CODE and to be a reduction
in the outermost loop. When returning true, store the operands in
*OP0_OUT and *OP1_OUT. */
static bool
vect_reassociating_reduction_p (vec_info *vinfo,
stmt_vec_info stmt_info, tree_code code,
tree *op0_out, tree *op1_out)
{
loop_vec_info loop_info = dyn_cast <loop_vec_info> (vinfo);
if (!loop_info)
return false;
gassign *assign = dyn_cast <gassign *> (stmt_info->stmt);
if (!assign || gimple_assign_rhs_code (assign) != code)
return false;
/* We don't allow changing the order of the computation in the inner-loop
when doing outer-loop vectorization. */
class loop *loop = LOOP_VINFO_LOOP (loop_info);
if (loop && nested_in_vect_loop_p (loop, stmt_info))
return false;
if (STMT_VINFO_DEF_TYPE (stmt_info) == vect_reduction_def)
{
if (needs_fold_left_reduction_p (TREE_TYPE (gimple_assign_lhs (assign)),
code))
return false;
}
else if (REDUC_GROUP_FIRST_ELEMENT (stmt_info) == NULL)
return false;
*op0_out = gimple_assign_rhs1 (assign);
*op1_out = gimple_assign_rhs2 (assign);
if (commutative_tree_code (code) && STMT_VINFO_REDUC_IDX (stmt_info) == 0)
std::swap (*op0_out, *op1_out);
return true;
}
/* match.pd function to match
(cond (cmp@3 a b) (convert@1 c) (convert@2 d))
with conditions:
1) @1, @2, c, d, a, b are all integral type.
2) There's single_use for both @1 and @2.
3) a, c have same precision.
4) c and @1 have different precision.
5) c, d are the same type or they can differ in sign when convert is
truncation.
record a and c and d and @3. */
extern bool gimple_cond_expr_convert_p (tree, tree*, tree (*)(tree));
/* Function vect_recog_cond_expr_convert
Try to find the following pattern:
TYPE_AB A,B;
TYPE_CD C,D;
TYPE_E E;
TYPE_E op_true = (TYPE_E) A;
TYPE_E op_false = (TYPE_E) B;
E = C cmp D ? op_true : op_false;
where
TYPE_PRECISION (TYPE_E) != TYPE_PRECISION (TYPE_CD);
TYPE_PRECISION (TYPE_AB) == TYPE_PRECISION (TYPE_CD);
single_use of op_true and op_false.
TYPE_AB could differ in sign when (TYPE_E) A is a truncation.
Input:
* STMT_VINFO: The stmt from which the pattern search begins.
here it starts with E = c cmp D ? op_true : op_false;
Output:
TYPE1 E' = C cmp D ? A : B;
TYPE3 E = (TYPE3) E';
There may extra nop_convert for A or B to handle different signness.
* TYPE_OUT: The vector type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. In this case it will be:
E = (TYPE3)E';
E' = C cmp D ? A : B; is recorded in pattern definition statements; */
static gimple *
vect_recog_cond_expr_convert_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo, tree *type_out)
{
gassign *last_stmt = dyn_cast <gassign *> (stmt_vinfo->stmt);
tree lhs, match[4], temp, type, new_lhs, op2;
gimple *cond_stmt;
gimple *pattern_stmt;
if (!last_stmt)
return NULL;
lhs = gimple_assign_lhs (last_stmt);
/* Find E = C cmp D ? (TYPE3) A ? (TYPE3) B;
TYPE_PRECISION (A) == TYPE_PRECISION (C). */
if (!gimple_cond_expr_convert_p (lhs, &match[0], NULL))
return NULL;
vect_pattern_detected ("vect_recog_cond_expr_convert_pattern", last_stmt);
op2 = match[2];
type = TREE_TYPE (match[1]);
if (TYPE_SIGN (type) != TYPE_SIGN (TREE_TYPE (match[2])))
{
op2 = vect_recog_temp_ssa_var (type, NULL);
gimple* nop_stmt = gimple_build_assign (op2, NOP_EXPR, match[2]);
append_pattern_def_seq (vinfo, stmt_vinfo, nop_stmt,
get_vectype_for_scalar_type (vinfo, type));
}
temp = vect_recog_temp_ssa_var (type, NULL);
cond_stmt = gimple_build_assign (temp, build3 (COND_EXPR, type, match[3],
match[1], op2));
append_pattern_def_seq (vinfo, stmt_vinfo, cond_stmt,
get_vectype_for_scalar_type (vinfo, type));
new_lhs = vect_recog_temp_ssa_var (TREE_TYPE (lhs), NULL);
pattern_stmt = gimple_build_assign (new_lhs, NOP_EXPR, temp);
*type_out = STMT_VINFO_VECTYPE (stmt_vinfo);
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location,
"created pattern stmt: %G", pattern_stmt);
return pattern_stmt;
}
/* Function vect_recog_dot_prod_pattern
Try to find the following pattern:
type1a x_t
type1b y_t;
TYPE1 prod;
TYPE2 sum = init;
loop:
sum_0 = phi <init, sum_1>
S1 x_t = ...
S2 y_t = ...
S3 x_T = (TYPE1) x_t;
S4 y_T = (TYPE1) y_t;
S5 prod = x_T * y_T;
[S6 prod = (TYPE2) prod; #optional]
S7 sum_1 = prod + sum_0;
where 'TYPE1' is exactly double the size of type 'type1a' and 'type1b',
the sign of 'TYPE1' must be one of 'type1a' or 'type1b' but the sign of
'type1a' and 'type1b' can differ.
Input:
* STMT_VINFO: The stmt from which the pattern search begins. In the
example, when this function is called with S7, the pattern {S3,S4,S5,S6,S7}
will be detected.
Output:
* TYPE_OUT: The type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. In this case it will be:
WIDEN_DOT_PRODUCT <x_t, y_t, sum_0>
Note: The dot-prod idiom is a widening reduction pattern that is
vectorized without preserving all the intermediate results. It
produces only N/2 (widened) results (by summing up pairs of
intermediate results) rather than all N results. Therefore, we
cannot allow this pattern when we want to get all the results and in
the correct order (as is the case when this computation is in an
inner-loop nested in an outer-loop that us being vectorized). */
static gimple *
vect_recog_dot_prod_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo, tree *type_out)
{
tree oprnd0, oprnd1;
gimple *last_stmt = stmt_vinfo->stmt;
tree type, half_type;
gimple *pattern_stmt;
tree var;
/* Look for the following pattern
DX = (TYPE1) X;
DY = (TYPE1) Y;
DPROD = DX * DY;
DDPROD = (TYPE2) DPROD;
sum_1 = DDPROD + sum_0;
In which
- DX is double the size of X
- DY is double the size of Y
- DX, DY, DPROD all have the same type but the sign
between X, Y and DPROD can differ.
- sum is the same size of DPROD or bigger
- sum has been recognized as a reduction variable.
This is equivalent to:
DPROD = X w* Y; #widen mult
sum_1 = DPROD w+ sum_0; #widen summation
or
DPROD = X w* Y; #widen mult
sum_1 = DPROD + sum_0; #summation
*/
/* Starting from LAST_STMT, follow the defs of its uses in search
of the above pattern. */
if (!vect_reassociating_reduction_p (vinfo, stmt_vinfo, PLUS_EXPR,
&oprnd0, &oprnd1))
return NULL;
type = TREE_TYPE (gimple_get_lhs (last_stmt));
vect_unpromoted_value unprom_mult;
oprnd0 = vect_look_through_possible_promotion (vinfo, oprnd0, &unprom_mult);
/* So far so good. Since last_stmt was detected as a (summation) reduction,
we know that oprnd1 is the reduction variable (defined by a loop-header
phi), and oprnd0 is an ssa-name defined by a stmt in the loop body.
Left to check that oprnd0 is defined by a (widen_)mult_expr */
if (!oprnd0)
return NULL;
stmt_vec_info mult_vinfo = vect_get_internal_def (vinfo, oprnd0);
if (!mult_vinfo)
return NULL;
/* FORNOW. Can continue analyzing the def-use chain when this stmt in a phi
inside the loop (in case we are analyzing an outer-loop). */
vect_unpromoted_value unprom0[2];
enum optab_subtype subtype = optab_vector;
if (!vect_widened_op_tree (vinfo, mult_vinfo, MULT_EXPR, WIDEN_MULT_EXPR,
false, 2, unprom0, &half_type, &subtype))
return NULL;
/* If there are two widening operations, make sure they agree on the sign
of the extension. The result of an optab_vector_mixed_sign operation
is signed; otherwise, the result has the same sign as the operands. */
if (TYPE_PRECISION (unprom_mult.type) != TYPE_PRECISION (type)
&& (subtype == optab_vector_mixed_sign
? TYPE_UNSIGNED (unprom_mult.type)
: TYPE_SIGN (unprom_mult.type) != TYPE_SIGN (half_type)))
return NULL;
vect_pattern_detected ("vect_recog_dot_prod_pattern", last_stmt);
/* If the inputs have mixed signs, canonicalize on using the signed
input type for analysis. This also helps when emulating mixed-sign
operations using signed operations. */
if (subtype == optab_vector_mixed_sign)
half_type = signed_type_for (half_type);
tree half_vectype;
if (!vect_supportable_direct_optab_p (vinfo, type, DOT_PROD_EXPR, half_type,
type_out, &half_vectype, subtype))
{
/* We can emulate a mixed-sign dot-product using a sequence of
signed dot-products; see vect_emulate_mixed_dot_prod for details. */
if (subtype != optab_vector_mixed_sign
|| !vect_supportable_direct_optab_p (vinfo, signed_type_for (type),
DOT_PROD_EXPR, half_type,
type_out, &half_vectype,
optab_vector))
return NULL;
*type_out = signed_or_unsigned_type_for (TYPE_UNSIGNED (type),
*type_out);
}
/* Get the inputs in the appropriate types. */
tree mult_oprnd[2];
vect_convert_inputs (vinfo, stmt_vinfo, 2, mult_oprnd, half_type,
unprom0, half_vectype, subtype);
var = vect_recog_temp_ssa_var (type, NULL);
pattern_stmt = gimple_build_assign (var, DOT_PROD_EXPR,
mult_oprnd[0], mult_oprnd[1], oprnd1);
return pattern_stmt;
}
/* Function vect_recog_sad_pattern
Try to find the following Sum of Absolute Difference (SAD) pattern:
type x_t, y_t;
signed TYPE1 diff, abs_diff;
TYPE2 sum = init;
loop:
sum_0 = phi <init, sum_1>
S1 x_t = ...
S2 y_t = ...
S3 x_T = (TYPE1) x_t;
S4 y_T = (TYPE1) y_t;
S5 diff = x_T - y_T;
S6 abs_diff = ABS_EXPR <diff>;
[S7 abs_diff = (TYPE2) abs_diff; #optional]
S8 sum_1 = abs_diff + sum_0;
where 'TYPE1' is at least double the size of type 'type', and 'TYPE2' is the
same size of 'TYPE1' or bigger. This is a special case of a reduction
computation.
Input:
* STMT_VINFO: The stmt from which the pattern search begins. In the
example, when this function is called with S8, the pattern
{S3,S4,S5,S6,S7,S8} will be detected.
Output:
* TYPE_OUT: The type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. In this case it will be:
SAD_EXPR <x_t, y_t, sum_0>
*/
static gimple *
vect_recog_sad_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo, tree *type_out)
{
gimple *last_stmt = stmt_vinfo->stmt;
tree half_type;
/* Look for the following pattern
DX = (TYPE1) X;
DY = (TYPE1) Y;
DDIFF = DX - DY;
DAD = ABS_EXPR <DDIFF>;
DDPROD = (TYPE2) DPROD;
sum_1 = DAD + sum_0;
In which
- DX is at least double the size of X
- DY is at least double the size of Y
- DX, DY, DDIFF, DAD all have the same type
- sum is the same size of DAD or bigger
- sum has been recognized as a reduction variable.
This is equivalent to:
DDIFF = X w- Y; #widen sub
DAD = ABS_EXPR <DDIFF>;
sum_1 = DAD w+ sum_0; #widen summation
or
DDIFF = X w- Y; #widen sub
DAD = ABS_EXPR <DDIFF>;
sum_1 = DAD + sum_0; #summation
*/
/* Starting from LAST_STMT, follow the defs of its uses in search
of the above pattern. */
tree plus_oprnd0, plus_oprnd1;
if (!vect_reassociating_reduction_p (vinfo, stmt_vinfo, PLUS_EXPR,
&plus_oprnd0, &plus_oprnd1))
return NULL;
tree sum_type = TREE_TYPE (gimple_get_lhs (last_stmt));
/* Any non-truncating sequence of conversions is OK here, since
with a successful match, the result of the ABS(U) is known to fit
within the nonnegative range of the result type. (It cannot be the
negative of the minimum signed value due to the range of the widening
MINUS_EXPR.) */
vect_unpromoted_value unprom_abs;
plus_oprnd0 = vect_look_through_possible_promotion (vinfo, plus_oprnd0,
&unprom_abs);
/* So far so good. Since last_stmt was detected as a (summation) reduction,
we know that plus_oprnd1 is the reduction variable (defined by a loop-header
phi), and plus_oprnd0 is an ssa-name defined by a stmt in the loop body.
Then check that plus_oprnd0 is defined by an abs_expr. */
if (!plus_oprnd0)
return NULL;
stmt_vec_info abs_stmt_vinfo = vect_get_internal_def (vinfo, plus_oprnd0);
if (!abs_stmt_vinfo)
return NULL;
/* FORNOW. Can continue analyzing the def-use chain when this stmt in a phi
inside the loop (in case we are analyzing an outer-loop). */
gassign *abs_stmt = dyn_cast <gassign *> (abs_stmt_vinfo->stmt);
if (!abs_stmt
|| (gimple_assign_rhs_code (abs_stmt) != ABS_EXPR
&& gimple_assign_rhs_code (abs_stmt) != ABSU_EXPR))
return NULL;
tree abs_oprnd = gimple_assign_rhs1 (abs_stmt);
tree abs_type = TREE_TYPE (abs_oprnd);
if (TYPE_UNSIGNED (abs_type))
return NULL;
/* Peel off conversions from the ABS input. This can involve sign
changes (e.g. from an unsigned subtraction to a signed ABS input)
or signed promotion, but it can't include unsigned promotion.
(Note that ABS of an unsigned promotion should have been folded
away before now anyway.) */
vect_unpromoted_value unprom_diff;
abs_oprnd = vect_look_through_possible_promotion (vinfo, abs_oprnd,
&unprom_diff);
if (!abs_oprnd)
return NULL;
if (TYPE_PRECISION (unprom_diff.type) != TYPE_PRECISION (abs_type)
&& TYPE_UNSIGNED (unprom_diff.type))
return NULL;
/* We then detect if the operand of abs_expr is defined by a minus_expr. */
stmt_vec_info diff_stmt_vinfo = vect_get_internal_def (vinfo, abs_oprnd);
if (!diff_stmt_vinfo)
return NULL;
/* FORNOW. Can continue analyzing the def-use chain when this stmt in a phi
inside the loop (in case we are analyzing an outer-loop). */
vect_unpromoted_value unprom[2];
if (!vect_widened_op_tree (vinfo, diff_stmt_vinfo, MINUS_EXPR, WIDEN_MINUS_EXPR,
false, 2, unprom, &half_type))
return NULL;
vect_pattern_detected ("vect_recog_sad_pattern", last_stmt);
tree half_vectype;
if (!vect_supportable_direct_optab_p (vinfo, sum_type, SAD_EXPR, half_type,
type_out, &half_vectype))
return NULL;
/* Get the inputs to the SAD_EXPR in the appropriate types. */
tree sad_oprnd[2];
vect_convert_inputs (vinfo, stmt_vinfo, 2, sad_oprnd, half_type,
unprom, half_vectype);
tree var = vect_recog_temp_ssa_var (sum_type, NULL);
gimple *pattern_stmt = gimple_build_assign (var, SAD_EXPR, sad_oprnd[0],
sad_oprnd[1], plus_oprnd1);
return pattern_stmt;
}
/* Recognize an operation that performs ORIG_CODE on widened inputs,
so that it can be treated as though it had the form:
A_TYPE a;
B_TYPE b;
HALF_TYPE a_cast = (HALF_TYPE) a; // possible no-op
HALF_TYPE b_cast = (HALF_TYPE) b; // possible no-op
| RES_TYPE a_extend = (RES_TYPE) a_cast; // promotion from HALF_TYPE
| RES_TYPE b_extend = (RES_TYPE) b_cast; // promotion from HALF_TYPE
| RES_TYPE res = a_extend ORIG_CODE b_extend;
Try to replace the pattern with:
A_TYPE a;
B_TYPE b;
HALF_TYPE a_cast = (HALF_TYPE) a; // possible no-op
HALF_TYPE b_cast = (HALF_TYPE) b; // possible no-op
| EXT_TYPE ext = a_cast WIDE_CODE b_cast;
| RES_TYPE res = (EXT_TYPE) ext; // possible no-op
where EXT_TYPE is wider than HALF_TYPE but has the same signedness.
SHIFT_P is true if ORIG_CODE and WIDE_CODE are shifts. NAME is the
name of the pattern being matched, for dump purposes. */
static gimple *
vect_recog_widen_op_pattern (vec_info *vinfo,
stmt_vec_info last_stmt_info, tree *type_out,
tree_code orig_code, tree_code wide_code,
bool shift_p, const char *name)
{
gimple *last_stmt = last_stmt_info->stmt;
vect_unpromoted_value unprom[2];
tree half_type;
if (!vect_widened_op_tree (vinfo, last_stmt_info, orig_code, orig_code,
shift_p, 2, unprom, &half_type))
return NULL;
/* Pattern detected. */
vect_pattern_detected (name, last_stmt);
tree type = TREE_TYPE (gimple_get_lhs (last_stmt));
tree itype = type;
if (TYPE_PRECISION (type) != TYPE_PRECISION (half_type) * 2
|| TYPE_UNSIGNED (type) != TYPE_UNSIGNED (half_type))
itype = build_nonstandard_integer_type (TYPE_PRECISION (half_type) * 2,
TYPE_UNSIGNED (half_type));
/* Check target support */
tree vectype = get_vectype_for_scalar_type (vinfo, half_type);
tree vecitype = get_vectype_for_scalar_type (vinfo, itype);
tree ctype = itype;
tree vecctype = vecitype;
if (orig_code == MINUS_EXPR
&& TYPE_UNSIGNED (itype)
&& TYPE_PRECISION (type) > TYPE_PRECISION (itype))
{
/* Subtraction is special, even if half_type is unsigned and no matter
whether type is signed or unsigned, if type is wider than itype,
we need to sign-extend from the widening operation result to the
result type.
Consider half_type unsigned char, operand 1 0xfe, operand 2 0xff,
itype unsigned short and type either int or unsigned int.
Widened (unsigned short) 0xfe - (unsigned short) 0xff is
(unsigned short) 0xffff, but for type int we want the result -1
and for type unsigned int 0xffffffff rather than 0xffff. */
ctype = build_nonstandard_integer_type (TYPE_PRECISION (itype), 0);
vecctype = get_vectype_for_scalar_type (vinfo, ctype);
}
enum tree_code dummy_code;
int dummy_int;
auto_vec<tree> dummy_vec;
if (!vectype
|| !vecitype
|| !vecctype
|| !supportable_widening_operation (vinfo, wide_code, last_stmt_info,
vecitype, vectype,
&dummy_code, &dummy_code,
&dummy_int, &dummy_vec))
return NULL;
*type_out = get_vectype_for_scalar_type (vinfo, type);
if (!*type_out)
return NULL;
tree oprnd[2];
vect_convert_inputs (vinfo, last_stmt_info,
2, oprnd, half_type, unprom, vectype);
tree var = vect_recog_temp_ssa_var (itype, NULL);
gimple *pattern_stmt = gimple_build_assign (var, wide_code,
oprnd[0], oprnd[1]);
if (vecctype != vecitype)
pattern_stmt = vect_convert_output (vinfo, last_stmt_info, ctype,
pattern_stmt, vecitype);
return vect_convert_output (vinfo, last_stmt_info,
type, pattern_stmt, vecctype);
}
/* Try to detect multiplication on widened inputs, converting MULT_EXPR
to WIDEN_MULT_EXPR. See vect_recog_widen_op_pattern for details. */
static gimple *
vect_recog_widen_mult_pattern (vec_info *vinfo, stmt_vec_info last_stmt_info,
tree *type_out)
{
return vect_recog_widen_op_pattern (vinfo, last_stmt_info, type_out,
MULT_EXPR, WIDEN_MULT_EXPR, false,
"vect_recog_widen_mult_pattern");
}
/* Try to detect addition on widened inputs, converting PLUS_EXPR
to WIDEN_PLUS_EXPR. See vect_recog_widen_op_pattern for details. */
static gimple *
vect_recog_widen_plus_pattern (vec_info *vinfo, stmt_vec_info last_stmt_info,
tree *type_out)
{
return vect_recog_widen_op_pattern (vinfo, last_stmt_info, type_out,
PLUS_EXPR, WIDEN_PLUS_EXPR, false,
"vect_recog_widen_plus_pattern");
}
/* Try to detect subtraction on widened inputs, converting MINUS_EXPR
to WIDEN_MINUS_EXPR. See vect_recog_widen_op_pattern for details. */
static gimple *
vect_recog_widen_minus_pattern (vec_info *vinfo, stmt_vec_info last_stmt_info,
tree *type_out)
{
return vect_recog_widen_op_pattern (vinfo, last_stmt_info, type_out,
MINUS_EXPR, WIDEN_MINUS_EXPR, false,
"vect_recog_widen_minus_pattern");
}
/* Function vect_recog_popcount_pattern
Try to find the following pattern:
UTYPE1 A;
TYPE1 B;
UTYPE2 temp_in;
TYPE3 temp_out;
temp_in = (UTYPE2)A;
temp_out = __builtin_popcount{,l,ll} (temp_in);
B = (TYPE1) temp_out;
TYPE2 may or may not be equal to TYPE3.
i.e. TYPE2 is equal to TYPE3 for __builtin_popcount
i.e. TYPE2 is not equal to TYPE3 for __builtin_popcountll
Input:
* STMT_VINFO: The stmt from which the pattern search begins.
here it starts with B = (TYPE1) temp_out;
Output:
* TYPE_OUT: The vector type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. In this case it will be:
B = .POPCOUNT (A);
*/
static gimple *
vect_recog_popcount_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo, tree *type_out)
{
gassign *last_stmt = dyn_cast <gassign *> (stmt_vinfo->stmt);
gimple *popcount_stmt, *pattern_stmt;
tree rhs_oprnd, rhs_origin, lhs_oprnd, lhs_type, vec_type, new_var;
auto_vec<tree> vargs;
/* Find B = (TYPE1) temp_out. */
if (!last_stmt)
return NULL;
tree_code code = gimple_assign_rhs_code (last_stmt);
if (!CONVERT_EXPR_CODE_P (code))
return NULL;
lhs_oprnd = gimple_assign_lhs (last_stmt);
lhs_type = TREE_TYPE (lhs_oprnd);
if (!INTEGRAL_TYPE_P (lhs_type))
return NULL;
rhs_oprnd = gimple_assign_rhs1 (last_stmt);
if (TREE_CODE (rhs_oprnd) != SSA_NAME
|| !has_single_use (rhs_oprnd))
return NULL;
popcount_stmt = SSA_NAME_DEF_STMT (rhs_oprnd);
/* Find temp_out = __builtin_popcount{,l,ll} (temp_in); */
if (!is_gimple_call (popcount_stmt))
return NULL;
switch (gimple_call_combined_fn (popcount_stmt))
{
CASE_CFN_POPCOUNT:
break;
default:
return NULL;
}
if (gimple_call_num_args (popcount_stmt) != 1)
return NULL;
rhs_oprnd = gimple_call_arg (popcount_stmt, 0);
vect_unpromoted_value unprom_diff;
rhs_origin = vect_look_through_possible_promotion (vinfo, rhs_oprnd,
&unprom_diff);
if (!rhs_origin)
return NULL;
/* Input and output of .POPCOUNT should be same-precision integer.
Also A should be unsigned or same precision as temp_in,
otherwise there would be sign_extend from A to temp_in. */
if (TYPE_PRECISION (unprom_diff.type) != TYPE_PRECISION (lhs_type)
|| (!TYPE_UNSIGNED (unprom_diff.type)
&& (TYPE_PRECISION (unprom_diff.type)
!= TYPE_PRECISION (TREE_TYPE (rhs_oprnd)))))
return NULL;
vargs.safe_push (unprom_diff.op);
vect_pattern_detected ("vec_regcog_popcount_pattern", popcount_stmt);
vec_type = get_vectype_for_scalar_type (vinfo, lhs_type);
/* Do it only if the backend has popcount<vector_mode>2 pattern. */
if (!vec_type
|| !direct_internal_fn_supported_p (IFN_POPCOUNT, vec_type,
OPTIMIZE_FOR_SPEED))
return NULL;
/* Create B = .POPCOUNT (A). */
new_var = vect_recog_temp_ssa_var (lhs_type, NULL);
pattern_stmt = gimple_build_call_internal_vec (IFN_POPCOUNT, vargs);
gimple_call_set_lhs (pattern_stmt, new_var);
gimple_set_location (pattern_stmt, gimple_location (last_stmt));
*type_out = vec_type;
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location,
"created pattern stmt: %G", pattern_stmt);
return pattern_stmt;
}
/* Function vect_recog_pow_pattern
Try to find the following pattern:
x = POW (y, N);
with POW being one of pow, powf, powi, powif and N being
either 2 or 0.5.
Input:
* STMT_VINFO: The stmt from which the pattern search begins.
Output:
* TYPE_OUT: The type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. In this case it will be:
x = x * x
or
x = sqrt (x)
*/
static gimple *
vect_recog_pow_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo, tree *type_out)
{
gimple *last_stmt = stmt_vinfo->stmt;
tree base, exp;
gimple *stmt;
tree var;
if (!is_gimple_call (last_stmt) || gimple_call_lhs (last_stmt) == NULL)
return NULL;
switch (gimple_call_combined_fn (last_stmt))
{
CASE_CFN_POW:
CASE_CFN_POWI:
break;
default:
return NULL;
}
base = gimple_call_arg (last_stmt, 0);
exp = gimple_call_arg (last_stmt, 1);
if (TREE_CODE (exp) != REAL_CST
&& TREE_CODE (exp) != INTEGER_CST)
{
if (flag_unsafe_math_optimizations
&& TREE_CODE (base) == REAL_CST
&& gimple_call_builtin_p (last_stmt, BUILT_IN_NORMAL))
{
combined_fn log_cfn;
built_in_function exp_bfn;
switch (DECL_FUNCTION_CODE (gimple_call_fndecl (last_stmt)))
{
case BUILT_IN_POW:
log_cfn = CFN_BUILT_IN_LOG;
exp_bfn = BUILT_IN_EXP;
break;
case BUILT_IN_POWF:
log_cfn = CFN_BUILT_IN_LOGF;
exp_bfn = BUILT_IN_EXPF;
break;
case BUILT_IN_POWL:
log_cfn = CFN_BUILT_IN_LOGL;
exp_bfn = BUILT_IN_EXPL;
break;
default:
return NULL;
}
tree logc = fold_const_call (log_cfn, TREE_TYPE (base), base);
tree exp_decl = builtin_decl_implicit (exp_bfn);
/* Optimize pow (C, x) as exp (log (C) * x). Normally match.pd
does that, but if C is a power of 2, we want to use
exp2 (log2 (C) * x) in the non-vectorized version, but for
vectorization we don't have vectorized exp2. */
if (logc
&& TREE_CODE (logc) == REAL_CST
&& exp_decl
&& lookup_attribute ("omp declare simd",
DECL_ATTRIBUTES (exp_decl)))
{
cgraph_node *node = cgraph_node::get_create (exp_decl);
if (node->simd_clones == NULL)
{
if (targetm.simd_clone.compute_vecsize_and_simdlen == NULL
|| node->definition)
return NULL;
expand_simd_clones (node);
if (node->simd_clones == NULL)
return NULL;
}
*type_out = get_vectype_for_scalar_type (vinfo, TREE_TYPE (base));
if (!*type_out)
return NULL;
tree def = vect_recog_temp_ssa_var (TREE_TYPE (base), NULL);
gimple *g = gimple_build_assign (def, MULT_EXPR, exp, logc);
append_pattern_def_seq (vinfo, stmt_vinfo, g);
tree res = vect_recog_temp_ssa_var (TREE_TYPE (base), NULL);
g = gimple_build_call (exp_decl, 1, def);
gimple_call_set_lhs (g, res);
return g;
}
}
return NULL;
}
/* We now have a pow or powi builtin function call with a constant
exponent. */
/* Catch squaring. */
if ((tree_fits_shwi_p (exp)
&& tree_to_shwi (exp) == 2)
|| (TREE_CODE (exp) == REAL_CST
&& real_equal (&TREE_REAL_CST (exp), &dconst2)))
{
if (!vect_supportable_direct_optab_p (vinfo, TREE_TYPE (base), MULT_EXPR,
TREE_TYPE (base), type_out))
return NULL;
var = vect_recog_temp_ssa_var (TREE_TYPE (base), NULL);
stmt = gimple_build_assign (var, MULT_EXPR, base, base);
return stmt;
}
/* Catch square root. */
if (TREE_CODE (exp) == REAL_CST
&& real_equal (&TREE_REAL_CST (exp), &dconsthalf))
{
*type_out = get_vectype_for_scalar_type (vinfo, TREE_TYPE (base));
if (*type_out
&& direct_internal_fn_supported_p (IFN_SQRT, *type_out,
OPTIMIZE_FOR_SPEED))
{
gcall *stmt = gimple_build_call_internal (IFN_SQRT, 1, base);
var = vect_recog_temp_ssa_var (TREE_TYPE (base), stmt);
gimple_call_set_lhs (stmt, var);
gimple_call_set_nothrow (stmt, true);
return stmt;
}
}
return NULL;
}
/* Function vect_recog_widen_sum_pattern
Try to find the following pattern:
type x_t;
TYPE x_T, sum = init;
loop:
sum_0 = phi <init, sum_1>
S1 x_t = *p;
S2 x_T = (TYPE) x_t;
S3 sum_1 = x_T + sum_0;
where type 'TYPE' is at least double the size of type 'type', i.e - we're
summing elements of type 'type' into an accumulator of type 'TYPE'. This is
a special case of a reduction computation.
Input:
* STMT_VINFO: The stmt from which the pattern search begins. In the example,
when this function is called with S3, the pattern {S2,S3} will be detected.
Output:
* TYPE_OUT: The type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. In this case it will be:
WIDEN_SUM <x_t, sum_0>
Note: The widening-sum idiom is a widening reduction pattern that is
vectorized without preserving all the intermediate results. It
produces only N/2 (widened) results (by summing up pairs of
intermediate results) rather than all N results. Therefore, we
cannot allow this pattern when we want to get all the results and in
the correct order (as is the case when this computation is in an
inner-loop nested in an outer-loop that us being vectorized). */
static gimple *
vect_recog_widen_sum_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo, tree *type_out)
{
gimple *last_stmt = stmt_vinfo->stmt;
tree oprnd0, oprnd1;
tree type;
gimple *pattern_stmt;
tree var;
/* Look for the following pattern
DX = (TYPE) X;
sum_1 = DX + sum_0;
In which DX is at least double the size of X, and sum_1 has been
recognized as a reduction variable.
*/
/* Starting from LAST_STMT, follow the defs of its uses in search
of the above pattern. */
if (!vect_reassociating_reduction_p (vinfo, stmt_vinfo, PLUS_EXPR,
&oprnd0, &oprnd1))
return NULL;
type = TREE_TYPE (gimple_get_lhs (last_stmt));
/* So far so good. Since last_stmt was detected as a (summation) reduction,
we know that oprnd1 is the reduction variable (defined by a loop-header
phi), and oprnd0 is an ssa-name defined by a stmt in the loop body.
Left to check that oprnd0 is defined by a cast from type 'type' to type
'TYPE'. */
vect_unpromoted_value unprom0;
if (!vect_look_through_possible_promotion (vinfo, oprnd0, &unprom0)
|| TYPE_PRECISION (unprom0.type) * 2 > TYPE_PRECISION (type))
return NULL;
vect_pattern_detected ("vect_recog_widen_sum_pattern", last_stmt);
if (!vect_supportable_direct_optab_p (vinfo, type, WIDEN_SUM_EXPR,
unprom0.type, type_out))
return NULL;
var = vect_recog_temp_ssa_var (type, NULL);
pattern_stmt = gimple_build_assign (var, WIDEN_SUM_EXPR, unprom0.op, oprnd1);
return pattern_stmt;
}
/* Function vect_recog_bitfield_ref_pattern
Try to find the following pattern:
bf_value = BIT_FIELD_REF (container, bitsize, bitpos);
result = (type_out) bf_value;
where type_out is a non-bitfield type, that is to say, it's precision matches
2^(TYPE_SIZE(type_out) - (TYPE_UNSIGNED (type_out) ? 1 : 2)).
Input:
* STMT_VINFO: The stmt from which the pattern search begins.
here it starts with:
result = (type_out) bf_value;
Output:
* TYPE_OUT: The vector type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. If the precision of type_out is bigger
than the precision type of _1 we perform the widening before the shifting,
since the new precision will be large enough to shift the value and moving
widening operations up the statement chain enables the generation of
widening loads. If we are widening and the operation after the pattern is
an addition then we mask first and shift later, to enable the generation of
shifting adds. In the case of narrowing we will always mask first, shift
last and then perform a narrowing operation. This will enable the
generation of narrowing shifts.
Widening with mask first, shift later:
container = (type_out) container;
masked = container & (((1 << bitsize) - 1) << bitpos);
result = patt2 >> masked;
Widening with shift first, mask last:
container = (type_out) container;
shifted = container >> bitpos;
result = shifted & ((1 << bitsize) - 1);
Narrowing:
masked = container & (((1 << bitsize) - 1) << bitpos);
result = masked >> bitpos;
result = (type_out) result;
The shifting is always optional depending on whether bitpos != 0.
*/
static gimple *
vect_recog_bitfield_ref_pattern (vec_info *vinfo, stmt_vec_info stmt_info,
tree *type_out)
{
gassign *first_stmt = dyn_cast <gassign *> (stmt_info->stmt);
if (!first_stmt)
return NULL;
gassign *bf_stmt;
if (CONVERT_EXPR_CODE_P (gimple_assign_rhs_code (first_stmt))
&& TREE_CODE (gimple_assign_rhs1 (first_stmt)) == SSA_NAME)
{
gimple *second_stmt
= SSA_NAME_DEF_STMT (gimple_assign_rhs1 (first_stmt));
bf_stmt = dyn_cast <gassign *> (second_stmt);
if (!bf_stmt
|| gimple_assign_rhs_code (bf_stmt) != BIT_FIELD_REF)
return NULL;
}
else
return NULL;
tree bf_ref = gimple_assign_rhs1 (bf_stmt);
tree container = TREE_OPERAND (bf_ref, 0);
if (!bit_field_offset (bf_ref).is_constant ()
|| !bit_field_size (bf_ref).is_constant ()
|| !tree_fits_uhwi_p (TYPE_SIZE (TREE_TYPE (container))))
return NULL;
if (!INTEGRAL_TYPE_P (TREE_TYPE (bf_ref))
|| !INTEGRAL_TYPE_P (TREE_TYPE (container))
|| TYPE_MODE (TREE_TYPE (container)) == E_BLKmode)
return NULL;
gimple *use_stmt, *pattern_stmt;
use_operand_p use_p;
tree ret = gimple_assign_lhs (first_stmt);
tree ret_type = TREE_TYPE (ret);
bool shift_first = true;
tree container_type = TREE_TYPE (container);
tree vectype = get_vectype_for_scalar_type (vinfo, container_type);
/* Calculate shift_n before the adjustments for widening loads, otherwise
the container may change and we have to consider offset change for
widening loads on big endianness. The shift_n calculated here can be
independent of widening. */
unsigned HOST_WIDE_INT shift_n = bit_field_offset (bf_ref).to_constant ();
unsigned HOST_WIDE_INT mask_width = bit_field_size (bf_ref).to_constant ();
unsigned HOST_WIDE_INT prec = tree_to_uhwi (TYPE_SIZE (container_type));
if (BYTES_BIG_ENDIAN)
shift_n = prec - shift_n - mask_width;
/* We move the conversion earlier if the loaded type is smaller than the
return type to enable the use of widening loads. */
if (TYPE_PRECISION (TREE_TYPE (container)) < TYPE_PRECISION (ret_type)
&& !useless_type_conversion_p (TREE_TYPE (container), ret_type))
{
pattern_stmt
= gimple_build_assign (vect_recog_temp_ssa_var (ret_type),
NOP_EXPR, container);
container = gimple_get_lhs (pattern_stmt);
container_type = TREE_TYPE (container);
prec = tree_to_uhwi (TYPE_SIZE (container_type));
vectype = get_vectype_for_scalar_type (vinfo, container_type);
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt, vectype);
}
else if (!useless_type_conversion_p (TREE_TYPE (container), ret_type))
/* If we are doing the conversion last then also delay the shift as we may
be able to combine the shift and conversion in certain cases. */
shift_first = false;
/* If the only use of the result of this BIT_FIELD_REF + CONVERT is a
PLUS_EXPR then do the shift last as some targets can combine the shift and
add into a single instruction. */
if (single_imm_use (gimple_assign_lhs (first_stmt), &use_p, &use_stmt))
{
if (gimple_code (use_stmt) == GIMPLE_ASSIGN
&& gimple_assign_rhs_code (use_stmt) == PLUS_EXPR)
shift_first = false;
}
/* If we don't have to shift we only generate the mask, so just fix the
code-path to shift_first. */
if (shift_n == 0)
shift_first = true;
tree result;
if (shift_first)
{
tree shifted = container;
if (shift_n)
{
pattern_stmt
= gimple_build_assign (vect_recog_temp_ssa_var (container_type),
RSHIFT_EXPR, container,
build_int_cst (sizetype, shift_n));
shifted = gimple_assign_lhs (pattern_stmt);
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt, vectype);
}
tree mask = wide_int_to_tree (container_type,
wi::mask (mask_width, false, prec));
pattern_stmt
= gimple_build_assign (vect_recog_temp_ssa_var (container_type),
BIT_AND_EXPR, shifted, mask);
result = gimple_assign_lhs (pattern_stmt);
}
else
{
tree mask = wide_int_to_tree (container_type,
wi::shifted_mask (shift_n, mask_width,
false, prec));
pattern_stmt
= gimple_build_assign (vect_recog_temp_ssa_var (container_type),
BIT_AND_EXPR, container, mask);
tree masked = gimple_assign_lhs (pattern_stmt);
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt, vectype);
pattern_stmt
= gimple_build_assign (vect_recog_temp_ssa_var (container_type),
RSHIFT_EXPR, masked,
build_int_cst (sizetype, shift_n));
result = gimple_assign_lhs (pattern_stmt);
}
if (!useless_type_conversion_p (TREE_TYPE (result), ret_type))
{
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt, vectype);
pattern_stmt
= gimple_build_assign (vect_recog_temp_ssa_var (ret_type),
NOP_EXPR, result);
}
*type_out = STMT_VINFO_VECTYPE (stmt_info);
vect_pattern_detected ("bitfield_ref pattern", stmt_info->stmt);
return pattern_stmt;
}
/* Function vect_recog_bit_insert_pattern
Try to find the following pattern:
written = BIT_INSERT_EXPR (container, value, bitpos);
Input:
* STMT_VINFO: The stmt we want to replace.
Output:
* TYPE_OUT: The vector type of the output of this pattern.
* Return value: A new stmt that will be used to replace the sequence of
stmts that constitute the pattern. In this case it will be:
value = (container_type) value; // Make sure
shifted = value << bitpos; // Shift value into place
masked = shifted & (mask << bitpos); // Mask off the non-relevant bits in
// the 'to-write value'.
cleared = container & ~(mask << bitpos); // Clearing the bits we want to
// write to from the value we want
// to write to.
written = cleared | masked; // Write bits.
where mask = ((1 << TYPE_PRECISION (value)) - 1), a mask to keep the number of
bits corresponding to the real size of the bitfield value we are writing to.
The shifting is always optional depending on whether bitpos != 0.
*/
static gimple *
vect_recog_bit_insert_pattern (vec_info *vinfo, stmt_vec_info stmt_info,
tree *type_out)
{
gassign *bf_stmt = dyn_cast <gassign *> (stmt_info->stmt);
if (!bf_stmt || gimple_assign_rhs_code (bf_stmt) != BIT_INSERT_EXPR)
return NULL;
tree container = gimple_assign_rhs1 (bf_stmt);
tree value = gimple_assign_rhs2 (bf_stmt);
tree shift = gimple_assign_rhs3 (bf_stmt);
tree bf_type = TREE_TYPE (value);
tree container_type = TREE_TYPE (container);
if (!INTEGRAL_TYPE_P (container_type)
|| !tree_fits_uhwi_p (TYPE_SIZE (container_type)))
return NULL;
gimple *pattern_stmt;
vect_unpromoted_value unprom;
unprom.set_op (value, vect_internal_def);
value = vect_convert_input (vinfo, stmt_info, container_type, &unprom,
get_vectype_for_scalar_type (vinfo,
container_type));
unsigned HOST_WIDE_INT mask_width = TYPE_PRECISION (bf_type);
unsigned HOST_WIDE_INT prec = tree_to_uhwi (TYPE_SIZE (container_type));
unsigned HOST_WIDE_INT shift_n = tree_to_uhwi (shift);
if (BYTES_BIG_ENDIAN)
{
shift_n = prec - shift_n - mask_width;
shift = build_int_cst (TREE_TYPE (shift), shift_n);
}
if (!useless_type_conversion_p (TREE_TYPE (value), container_type))
{
pattern_stmt =
gimple_build_assign (vect_recog_temp_ssa_var (container_type),
NOP_EXPR, value);
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt);
value = gimple_get_lhs (pattern_stmt);
}
/* Shift VALUE into place. */
tree shifted = value;
if (shift_n)
{
gimple_seq stmts = NULL;
shifted
= gimple_build (&stmts, LSHIFT_EXPR, container_type, value, shift);
if (!gimple_seq_empty_p (stmts))
append_pattern_def_seq (vinfo, stmt_info,
gimple_seq_first_stmt (stmts));
}
tree mask_t
= wide_int_to_tree (container_type,
wi::shifted_mask (shift_n, mask_width, false, prec));
/* Clear bits we don't want to write back from SHIFTED. */
gimple_seq stmts = NULL;
tree masked = gimple_build (&stmts, BIT_AND_EXPR, container_type, shifted,
mask_t);
if (!gimple_seq_empty_p (stmts))
{
pattern_stmt = gimple_seq_first_stmt (stmts);
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt);
}
/* Mask off the bits in the container that we are to write to. */
mask_t = wide_int_to_tree (container_type,
wi::shifted_mask (shift_n, mask_width, true, prec));
tree cleared = vect_recog_temp_ssa_var (container_type);
pattern_stmt = gimple_build_assign (cleared, BIT_AND_EXPR, container, mask_t);
append_pattern_def_seq (vinfo, stmt_info, pattern_stmt);
/* Write MASKED into CLEARED. */
pattern_stmt
= gimple_build_assign (vect_recog_temp_ssa_var (container_type),
BIT_IOR_EXPR, cleared, masked);
*type_out = STMT_VINFO_VECTYPE (stmt_info);
vect_pattern_detected ("bit_insert pattern", stmt_info->stmt);
return pattern_stmt;
}
/* Recognize cases in which an operation is performed in one type WTYPE
but could be done more efficiently in a narrower type NTYPE. For example,
if we have:
ATYPE a; // narrower than NTYPE
BTYPE b; // narrower than NTYPE
WTYPE aw = (WTYPE) a;
WTYPE bw = (WTYPE) b;
WTYPE res = aw + bw; // only uses of aw and bw
then it would be more efficient to do:
NTYPE an = (NTYPE) a;
NTYPE bn = (NTYPE) b;
NTYPE resn = an + bn;
WTYPE res = (WTYPE) resn;
Other situations include things like:
ATYPE a; // NTYPE or narrower
WTYPE aw = (WTYPE) a;
WTYPE res = aw + b;
when only "(NTYPE) res" is significant. In that case it's more efficient
to truncate "b" and do the operation on NTYPE instead:
NTYPE an = (NTYPE) a;
NTYPE bn = (NTYPE) b; // truncation
NTYPE resn = an + bn;
WTYPE res = (WTYPE) resn;
All users of "res" should then use "resn" instead, making the final
statement dead (not marked as relevant). The final statement is still
needed to maintain the type correctness of the IR.
vect_determine_precisions has already determined the minimum
precison of the operation and the minimum precision required
by users of the result. */
static gimple *
vect_recog_over_widening_pattern (vec_info *vinfo,
stmt_vec_info last_stmt_info, tree *type_out)
{
gassign *last_stmt = dyn_cast <gassign *> (last_stmt_info->stmt);
if (!last_stmt)
return NULL;
/* See whether we have found that this operation can be done on a
narrower type without changing its semantics. */
unsigned int new_precision = last_stmt_info->operation_precision;
if (!new_precision)
return NULL;
tree lhs = gimple_assign_lhs (last_stmt);
tree type = TREE_TYPE (lhs);
tree_code code = gimple_assign_rhs_code (last_stmt);
/* Punt for reductions where we don't handle the type conversions. */
if (STMT_VINFO_DEF_TYPE (last_stmt_info) == vect_reduction_def)
return NULL;
/* Keep the first operand of a COND_EXPR as-is: only the other two
operands are interesting. */
unsigned int first_op = (code == COND_EXPR ? 2 : 1);
/* Check the operands. */
unsigned int nops = gimple_num_ops (last_stmt) - first_op;
auto_vec <vect_unpromoted_value, 3> unprom (nops);
unprom.quick_grow (nops);
unsigned int min_precision = 0;
bool single_use_p = false;
for (unsigned int i = 0; i < nops; ++i)
{
tree op = gimple_op (last_stmt, first_op + i);
if (TREE_CODE (op) == INTEGER_CST)
unprom[i].set_op (op, vect_constant_def);
else if (TREE_CODE (op) == SSA_NAME)
{
bool op_single_use_p = true;
if (!vect_look_through_possible_promotion (vinfo, op, &unprom[i],
&op_single_use_p))
return NULL;
/* If:
(1) N bits of the result are needed;
(2) all inputs are widened from M<N bits; and
(3) one operand OP is a single-use SSA name
we can shift the M->N widening from OP to the output
without changing the number or type of extensions involved.
This then reduces the number of copies of STMT_INFO.
If instead of (3) more than one operand is a single-use SSA name,
shifting the extension to the output is even more of a win.
If instead:
(1) N bits of the result are needed;
(2) one operand OP2 is widened from M2<N bits;
(3) another operand OP1 is widened from M1<M2 bits; and
(4) both OP1 and OP2 are single-use
the choice is between:
(a) truncating OP2 to M1, doing the operation on M1,
and then widening the result to N
(b) widening OP1 to M2, doing the operation on M2, and then
widening the result to N
Both shift the M2->N widening of the inputs to the output.
(a) additionally shifts the M1->M2 widening to the output;
it requires fewer copies of STMT_INFO but requires an extra
M2->M1 truncation.
Which is better will depend on the complexity and cost of
STMT_INFO, which is hard to predict at this stage. However,
a clear tie-breaker in favor of (b) is the fact that the
truncation in (a) increases the length of the operation chain.
If instead of (4) only one of OP1 or OP2 is single-use,
(b) is still a win over doing the operation in N bits:
it still shifts the M2->N widening on the single-use operand
to the output and reduces the number of STMT_INFO copies.
If neither operand is single-use then operating on fewer than
N bits might lead to more extensions overall. Whether it does
or not depends on global information about the vectorization
region, and whether that's a good trade-off would again
depend on the complexity and cost of the statements involved,
as well as things like register pressure that are not normally
modelled at this stage. We therefore ignore these cases
and just optimize the clear single-use wins above.
Thus we take the maximum precision of the unpromoted operands
and record whether any operand is single-use. */
if (unprom[i].dt == vect_internal_def)
{
min_precision = MAX (min_precision,
TYPE_PRECISION (unprom[i].type));
single_use_p |= op_single_use_p;
}
}
else
return NULL;
}
/* Although the operation could be done in operation_precision, we have
to balance that against introducing extra truncations or extensions.
Calculate the minimum precision that can be handled efficiently.
The loop above determined that the operation could be handled
efficiently in MIN_PRECISION if SINGLE_USE_P; this would shift an
extension from the inputs to the output without introducing more
instructions, and would reduce the number of instructions required
for STMT_INFO itself.
vect_determine_precisions has also determined that the result only
needs min_output_precision bits. Truncating by a factor of N times
requires a tree of N - 1 instructions, so if TYPE is N times wider
than min_output_precision, doing the operation in TYPE and truncating
the result requires N + (N - 1) = 2N - 1 instructions per output vector.
In contrast:
- truncating the input to a unary operation and doing the operation
in the new type requires at most N - 1 + 1 = N instructions per
output vector
- doing the same for a binary operation requires at most
(N - 1) * 2 + 1 = 2N - 1 instructions per output vector
Both unary and binary operations require fewer instructions than
this if the operands were extended from a suitable truncated form.
Thus there is usually nothing to lose by doing operations in
min_output_precision bits, but there can be something to gain. */
if (!single_use_p)
min_precision = last_stmt_info->min_output_precision;
else
min_precision = MIN (min_precision, last_stmt_info->min_output_precision);
/* Apply the minimum efficient precision we just calculated. */
if (new_precision < min_precision)
new_precision = min_precision;
new_precision = vect_element_precision (new_precision);
if (new_precision >= TYPE_PRECISION (type))
return NULL;
vect_pattern_detected ("vect_recog_over_widening_pattern", last_stmt);
*type_out = get_vectype_for_scalar_type (vinfo, type);
if (!*type_out)
return NULL;
/* We've found a viable pattern. Get the new type of the operation. */
bool unsigned_p = (last_stmt_info->operation_sign == UNSIGNED);
tree new_type = build_nonstandard_integer_type (new_precision, unsigned_p);
/* If we're truncating an operation, we need to make sure that we
don't introduce new undefined overflow. The codes tested here are
a subset of those accepted by vect_truncatable_operation_p. */
tree op_type = new_type;
if (TYPE_OVERFLOW_UNDEFINED (new_type)
&& (code == PLUS_EXPR || code == MINUS_EXPR || code == MULT_EXPR))
op_type = build_nonstandard_integer_type (new_precision, true);
/* We specifically don't check here whether the target supports the
new operation, since it might be something that a later pattern
wants to rewrite anyway. If targets have a minimum element size
for some optabs, we should pattern-match smaller ops to larger ops
where beneficial. */
tree new_vectype = get_vectype_for_scalar_type (vinfo, new_type);
tree op_vectype = get_vectype_for_scalar_type (vinfo, op_type);
if (!new_vectype || !op_vectype)
return NULL;
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location, "demoting %T to %T\n",
type, new_type);
/* Calculate the rhs operands for an operation on OP_TYPE. */
tree ops[3] = {};
for (unsigned int i = 1; i < first_op; ++i)
ops[i - 1] = gimple_op (last_stmt, i);
vect_convert_inputs (vinfo, last_stmt_info, nops, &ops[first_op - 1],
op_type, &unprom[0], op_vectype);
/* Use the operation to produce a result of type OP_TYPE. */
tree new_var = vect_recog_temp_ssa_var (op_type, NULL);
gimple *pattern_stmt = gimple_build_assign (new_var, code,
ops[0], ops[1], ops[2]);
gimple_set_location (pattern_stmt, gimple_location (last_stmt));
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location,
"created pattern stmt: %G", pattern_stmt);
/* Convert back to the original signedness, if OP_TYPE is different
from NEW_TYPE. */
if (op_type != new_type)
pattern_stmt = vect_convert_output (vinfo, last_stmt_info, new_type,
pattern_stmt, op_vectype);
/* Promote the result to the original type. */
pattern_stmt = vect_convert_output (vinfo, last_stmt_info, type,
pattern_stmt, new_vectype);
return pattern_stmt;
}
/* Recognize the following patterns:
ATYPE a; // narrower than TYPE
BTYPE b; // narrower than TYPE
1) Multiply high with scaling
TYPE res = ((TYPE) a * (TYPE) b) >> c;
Here, c is bitsize (TYPE) / 2 - 1.
2) ... or also with rounding
TYPE res = (((TYPE) a * (TYPE) b) >> d + 1) >> 1;
Here, d is bitsize (TYPE) / 2 - 2.
3) Normal multiply high
TYPE res = ((TYPE) a * (TYPE) b) >> e;
Here, e is bitsize (TYPE) / 2.
where only the bottom half of res is used. */
static gimple *
vect_recog_mulhs_pattern (vec_info *vinfo,
stmt_vec_info last_stmt_info, tree *type_out)
{
/* Check for a right shift. */
gassign *last_stmt = dyn_cast <gassign *> (last_stmt_info->stmt);
if (!last_stmt
|| gimple_assign_rhs_code (last_stmt) != RSHIFT_EXPR)
return NULL;
/* Check that the shift result is wider than the users of the
result need (i.e. that narrowing would be a natural choice). */
tree lhs_type = TREE_TYPE (gimple_assign_lhs (last_stmt));
unsigned int target_precision
= vect_element_precision (last_stmt_info->min_output_precision);
if (!INTEGRAL_TYPE_P (lhs_type)
|| target_precision >= TYPE_PRECISION (lhs_type))
return NULL;
/* Look through any change in sign on the outer shift input. */
vect_unpromoted_value unprom_rshift_input;
tree rshift_input = vect_look_through_possible_promotion
(vinfo, gimple_assign_rhs1 (last_stmt), &unprom_rshift_input);
if (!rshift_input
|| TYPE_PRECISION (TREE_TYPE (rshift_input))
!= TYPE_PRECISION (lhs_type))
return NULL;
/* Get the definition of the shift input. */
stmt_vec_info rshift_input_stmt_info
= vect_get_internal_def (vinfo, rshift_input);
if (!rshift_input_stmt_info)
return NULL;
gassign *rshift_input_stmt
= dyn_cast <gassign *> (rshift_input_stmt_info->stmt);
if (!rshift_input_stmt)
return NULL;
stmt_vec_info mulh_stmt_info;
tree scale_term;
bool rounding_p = false;
/* Check for the presence of the rounding term. */
if (gimple_assign_rhs_code (rshift_input_stmt) == PLUS_EXPR)
{
/* Check that the outer shift was by 1. */
if (!integer_onep (gimple_assign_rhs2 (last_stmt)))
return NULL;
/* Check that the second operand of the PLUS_EXPR is 1. */
if (!integer_onep (gimple_assign_rhs2 (rshift_input_stmt)))
return NULL;
/* Look through any change in sign on the addition input. */
vect_unpromoted_value unprom_plus_input;
tree plus_input = vect_look_through_possible_promotion
(vinfo, gimple_assign_rhs1 (rshift_input_stmt), &unprom_plus_input);
if (!plus_input
|| TYPE_PRECISION (TREE_TYPE (plus_input))
!= TYPE_PRECISION (TREE_TYPE (rshift_input)))
return NULL;
/* Get the definition of the multiply-high-scale part. */
stmt_vec_info plus_input_stmt_info
= vect_get_internal_def (vinfo, plus_input);
if (!plus_input_stmt_info)
return NULL;
gassign *plus_input_stmt
= dyn_cast <gassign *> (plus_input_stmt_info->stmt);
if (!plus_input_stmt
|| gimple_assign_rhs_code (plus_input_stmt) != RSHIFT_EXPR)
return NULL;
/* Look through any change in sign on the scaling input. */
vect_unpromoted_value unprom_scale_input;
tree scale_input = vect_look_through_possible_promotion
(vinfo, gimple_assign_rhs1 (plus_input_stmt), &unprom_scale_input);
if (!scale_input
|| TYPE_PRECISION (TREE_TYPE (scale_input))
!= TYPE_PRECISION (TREE_TYPE (plus_input)))
return NULL;
/* Get the definition of the multiply-high part. */
mulh_stmt_info = vect_get_internal_def (vinfo, scale_input);
if (!mulh_stmt_info)
return NULL;
/* Get the scaling term. */
scale_term = gimple_assign_rhs2 (plus_input_stmt);
rounding_p = true;
}
else
{
mulh_stmt_info = rshift_input_stmt_info;
scale_term = gimple_assign_rhs2 (last_stmt);
}
/* Check that the scaling factor is constant. */
if (TREE_CODE (scale_term) != INTEGER_CST)
return NULL;
/* Check whether the scaling input term can be seen as two widened
inputs multiplied together. */
vect_unpromoted_value unprom_mult[2];
tree new_type;
unsigned int nops
= vect_widened_op_tree (vinfo, mulh_stmt_info, MULT_EXPR, WIDEN_MULT_EXPR,
false, 2, unprom_mult, &new_type);
if (nops != 2)
return NULL;
/* Adjust output precision. */
if (TYPE_PRECISION (new_type) < target_precision)
new_type = build_nonstandard_integer_type
(target_precision, TYPE_UNSIGNED (new_type));
unsigned mult_precision = TYPE_PRECISION (new_type);
internal_fn ifn;
/* Check that the scaling factor is expected. Instead of
target_precision, we should use the one that we actually
use for internal function. */
if (rounding_p)
{
/* Check pattern 2). */
if (wi::to_widest (scale_term) + mult_precision + 2
!= TYPE_PRECISION (lhs_type))
return NULL;
ifn = IFN_MULHRS;
}
else
{
/* Check for pattern 1). */
if (wi::to_widest (scale_term) + mult_precision + 1
== TYPE_PRECISION (lhs_type))
ifn = IFN_MULHS;
/* Check for pattern 3). */
else if (wi::to_widest (scale_term) + mult_precision
== TYPE_PRECISION (lhs_type))
ifn = IFN_MULH;
else
return NULL;
}
vect_pattern_detected ("vect_recog_mulhs_pattern", last_stmt);
/* Check for target support. */
tree new_vectype = get_vectype_for_scalar_type (vinfo, new_type);
if (!new_vectype
|| !direct_internal_fn_supported_p
(ifn, new_vectype, OPTIMIZE_FOR_SPEED))
return NULL;
/* The IR requires a valid vector type for the cast result, even though
it's likely to be discarded. */
*type_out = get_vectype_for_scalar_type (vinfo, lhs_type);
if (!*type_out)
return NULL;
/* Generate the IFN_MULHRS call. */
tree new_var = vect_recog_temp_ssa_var (new_type, NULL);
tree new_ops[2];
vect_convert_inputs (vinfo, last_stmt_info, 2, new_ops, new_type,
unprom_mult, new_vectype);
gcall *mulhrs_stmt
= gimple_build_call_internal (ifn, 2, new_ops[0], new_ops[1]);
gimple_call_set_lhs (mulhrs_stmt, new_var);
gimple_set_location (mulhrs_stmt, gimple_location (last_stmt));
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location,
"created pattern stmt: %G", (gimple *) mulhrs_stmt);
return vect_convert_output (vinfo, last_stmt_info, lhs_type,
mulhrs_stmt, new_vectype);
}
/* Recognize the patterns:
ATYPE a; // narrower than TYPE
BTYPE b; // narrower than TYPE
(1) TYPE avg = ((TYPE) a + (TYPE) b) >> 1;
or (2) TYPE avg = ((TYPE) a + (TYPE) b + 1) >> 1;
where only the bottom half of avg is used. Try to transform them into:
(1) NTYPE avg' = .AVG_FLOOR ((NTYPE) a, (NTYPE) b);
or (2) NTYPE avg' = .AVG_CEIL ((NTYPE) a, (NTYPE) b);
followed by:
TYPE avg = (TYPE) avg';
where NTYPE is no wider than half of TYPE. Since only the bottom half
of avg is used, all or part of the cast of avg' should become redundant.
If there is no target support available, generate code to distribute rshift
over plus and add a carry. */
static gimple *
vect_recog_average_pattern (vec_info *vinfo,
stmt_vec_info last_stmt_info, tree *type_out)
{
/* Check for a shift right by one bit. */
gassign *last_stmt = dyn_cast <gassign *> (last_stmt_info->stmt);
if (!last_stmt
|| gimple_assign_rhs_code (last_stmt) != RSHIFT_EXPR
|| !integer_onep (gimple_assign_rhs2 (last_stmt)))
return NULL;
/* Check that the shift result is wider than the users of the
result need (i.e. that narrowing would be a natural choice). */
tree lhs = gimple_assign_lhs (last_stmt);
tree type = TREE_TYPE (lhs);
unsigned int target_precision
= vect_element_precision (last_stmt_info->min_output_precision);
if (!INTEGRAL_TYPE_P (type) || target_precision >= TYPE_PRECISION (type))
return NULL;
/* Look through any change in sign on the shift input. */
tree rshift_rhs = gimple_assign_rhs1 (last_stmt);
vect_unpromoted_value unprom_plus;
rshift_rhs = vect_look_through_possible_promotion (vinfo, rshift_rhs,
&unprom_plus);
if (!rshift_rhs
|| TYPE_PRECISION (TREE_TYPE (rshift_rhs)) != TYPE_PRECISION (type))
return NULL;
/* Get the definition of the shift input. */
stmt_vec_info plus_stmt_info = vect_get_internal_def (vinfo, rshift_rhs);
if (!plus_stmt_info)
return NULL;
/* Check whether the shift input can be seen as a tree of additions on
2 or 3 widened inputs.
Note that the pattern should be a win even if the result of one or
more additions is reused elsewhere: if the pattern matches, we'd be
replacing 2N RSHIFT_EXPRs and N VEC_PACK_*s with N IFN_AVG_*s. */
internal_fn ifn = IFN_AVG_FLOOR;
vect_unpromoted_value unprom[3];
tree new_type;
unsigned int nops = vect_widened_op_tree (vinfo, plus_stmt_info, PLUS_EXPR,
WIDEN_PLUS_EXPR, false, 3,
unprom, &new_type);
if (nops == 0)
return NULL;
if (nops == 3)
{
/* Check that one operand is 1. */
unsigned int i;
for (i = 0; i < 3; ++i)
if (integer_onep (unprom[i].op))
break;
if (i == 3)
return NULL;
/* Throw away the 1 operand and keep the other two. */
if (i < 2)
unprom[i] = unprom[2];
ifn = IFN_AVG_CEIL;
}
vect_pattern_detected ("vect_recog_average_pattern", last_stmt);
/* We know that:
(a) the operation can be viewed as:
TYPE widened0 = (TYPE) UNPROM[0];
TYPE widened1 = (TYPE) UNPROM[1];
TYPE tmp1 = widened0 + widened1 {+ 1};
TYPE tmp2 = tmp1 >> 1; // LAST_STMT_INFO
(b) the first two statements are equivalent to:
TYPE widened0 = (TYPE) (NEW_TYPE) UNPROM[0];
TYPE widened1 = (TYPE) (NEW_TYPE) UNPROM[1];
(c) vect_recog_over_widening_pattern has already tried to narrow TYPE
where sensible;
(d) all the operations can be performed correctly at twice the width of
NEW_TYPE, due to the nature of the average operation; and
(e) users of the result of the right shift need only TARGET_PRECISION
bits, where TARGET_PRECISION is no more than half of TYPE's
precision.
Under these circumstances, the only situation in which NEW_TYPE
could be narrower than TARGET_PRECISION is if widened0, widened1
and an addition result are all used more than once. Thus we can
treat any widening of UNPROM[0] and UNPROM[1] to TARGET_PRECISION
as "free", whereas widening the result of the average instruction
from NEW_TYPE to TARGET_PRECISION would be a new operation. It's
therefore better not to go narrower than TARGET_PRECISION. */
if (TYPE_PRECISION (new_type) < target_precision)
new_type = build_nonstandard_integer_type (target_precision,
TYPE_UNSIGNED (new_type));
/* Check for target support. */
tree new_vectype = get_vectype_for_scalar_type (vinfo, new_type);
if (!new_vectype)
return NULL;
bool fallback_p = false;
if (direct_internal_fn_supported_p (ifn, new_vectype, OPTIMIZE_FOR_SPEED))
;
else if (TYPE_UNSIGNED (new_type)
&& optab_for_tree_code (RSHIFT_EXPR, new_vectype, optab_scalar)
&& optab_for_tree_code (PLUS_EXPR, new_vectype, optab_default)
&& optab_for_tree_code (BIT_IOR_EXPR, new_vectype, optab_default)
&& optab_for_tree_code (BIT_AND_EXPR, new_vectype, optab_default))
fallback_p = true;
else
return NULL;
/* The IR requires a valid vector type for the cast result, even though
it's likely to be discarded. */
*type_out = get_vectype_for_scalar_type (vinfo, type);
if (!*type_out)
return NULL;
tree new_var = vect_recog_temp_ssa_var (new_type, NULL);
tree new_ops[2];
vect_convert_inputs (vinfo, last_stmt_info, 2, new_ops, new_type,
unprom, new_vectype);
if (fallback_p)
{
/* As a fallback, generate code for following sequence:
shifted_op0 = new_ops[0] >> 1;
shifted_op1 = new_ops[1] >> 1;
sum_of_shifted = shifted_op0 + shifted_op1;
unmasked_carry = new_ops[0] and/or new_ops[1];
carry = unmasked_carry & 1;
new_var = sum_of_shifted + carry;
*/
tree one_cst = build_one_cst (new_type);
gassign *g;
tree shifted_op0 = vect_recog_temp_ssa_var (new_type, NULL);
g = gimple_build_assign (shifted_op0, RSHIFT_EXPR, new_ops[0], one_cst);
append_pattern_def_seq (vinfo, last_stmt_info, g, new_vectype);
tree shifted_op1 = vect_recog_temp_ssa_var (new_type, NULL);
g = gimple_build_assign (shifted_op1, RSHIFT_EXPR, new_ops[1], one_cst);
append_pattern_def_seq (vinfo, last_stmt_info, g, new_vectype);
tree sum_of_shifted = vect_recog_temp_ssa_var (new_type, NULL);
g = gimple_build_assign (sum_of_shifted, PLUS_EXPR,
shifted_op0, shifted_op1);
append_pattern_def_seq (vinfo, last_stmt_info, g, new_vectype);
tree unmasked_carry = vect_recog_temp_ssa_var (new_type, NULL);
tree_code c = (ifn == IFN_AVG_CEIL) ? BIT_IOR_EXPR : BIT_AND_EXPR;
g = gimple_build_assign (unmasked_carry, c, new_ops[0], new_ops[1]);
append_pattern_def_seq (vinfo, last_stmt_info, g, new_vectype);
tree carry = vect_recog_temp_ssa_var (new_type, NULL);
g = gimple_build_assign (carry, BIT_AND_EXPR, unmasked_carry, one_cst);
append_pattern_def_seq (vinfo, last_stmt_info, g, new_vectype);
g = gimple_build_assign (new_var, PLUS_EXPR, sum_of_shifted, carry);
return vect_convert_output (vinfo, last_stmt_info, type, g, new_vectype);
}
/* Generate the IFN_AVG* call. */
gcall *average_stmt = gimple_build_call_internal (ifn, 2, new_ops[0],
new_ops[1]);
gimple_call_set_lhs (average_stmt, new_var);
gimple_set_location (average_stmt, gimple_location (last_stmt));
if (dump_enabled_p ())
dump_printf_loc (MSG_NOTE, vect_location,
"created pattern stmt: %G", (gimple *) average_stmt);
return vect_convert_output (vinfo, last_stmt_info,
type, average_stmt, new_vectype);
}
/* Recognize cases in which the input to a cast is wider than its
output, and the input is fed by a widening operation. Fold this
by removing the unnecessary intermediate widening. E.g.:
unsigned char a;
unsigned int b = (unsigned int) a;
unsigned short c = (unsigned short) b;
-->
unsigned short c = (unsigned short) a;
Although this is rare in input IR, it is an expected side-effect
of the over-widening pattern above.
This is beneficial also for integer-to-float conversions, if the
widened integer has more bits than the float, and if the unwidened
input doesn't. */
static gimple *
vect_recog_cast_forwprop_pattern (vec_info *vinfo,
stmt_vec_info last_stmt_info, tree *type_out)
{
/* Check for a cast, including an integer-to-float conversion. */
gassign *last_stmt = dyn_cast <gassign *> (last_stmt_info->stmt);
if (!last_stmt)
return NULL;
tree_code code = gimple_assign_rhs_code (last_stmt);
if (!CONVERT_EXPR_CODE_P (code) && code != FLOAT_EXPR)
return NULL;
/* Make sure that the rhs is a scalar with a natural bitsize. */
tree lhs = gimple_assign_lhs (last_stmt);
if (!lhs)
return NULL;
tree lhs_type = TREE_TYPE (lhs);
scalar_mode lhs_mode;
if (VECT_SCALAR_BOOLEAN_TYPE_P (lhs_type)
|| !is_a <scalar_mode> (TYPE_MODE (lhs_type), &lhs_mode))
return NULL;
/* Check for a narrowing operation (from a vector point of view). */
tree rhs = gimple_assign_rhs1 (last_stmt);
tree rhs_type = TREE_TYPE (rhs);
if (!INTEGRAL_TYPE_P (rhs_type)
|| VECT_SCALAR_BOOLEAN_TYPE_P (rhs_type)
|| TYPE_PRECISION (rhs_type) <= GET_MODE_BITSIZE (lhs_mode))
return NULL;
/* Try to find an unpromoted input. */
vect_unpromoted_value unprom;
if (!vect_look_through_possible_promotion (vinfo, rhs, &unprom)
|| TYPE_PRECISION (unprom.type) >= TYPE_PRECISION (rhs_type))
return NULL;
/* If the bits above RHS_TYPE matter, make sure that they're the
same when extending from UNPROM as they are when extending from RHS. */
if (!INTEGRAL_TYPE_P (lhs_type)
&& TYPE_SIGN (rhs_type) != TYPE_SIGN (unprom.type))
return NULL;
/* We can get the same result by casting UNPROM directly, to avoid
the unnecessary widening and narrowing. */
vect_pattern_detected ("vect_recog_cast_forwprop_pattern", last_stmt);
*type_out = get_vectype_for_scalar_type (vinfo, lhs_type);
if (!*type_out)
return NULL;
tree new_var = vect_recog_temp_ssa_var (lhs_type, NULL);
gimple *pattern_stmt = gimple_build_assign (new_var, code, unprom.op);
gimple_set_location (pattern_stmt, gimple_location (last_stmt));
return pattern_stmt;
}
/* Try to detect a shift left of a widened input, converting LSHIFT_EXPR
to WIDEN_LSHIFT_EXPR. See vect_recog_widen_op_pattern for details. */
static gimple *
vect_recog_widen_shift_pattern (vec_info *vinfo,
stmt_vec_info last_stmt_info, tree *type_out)
{
return vect_recog_widen_op_pattern (vinfo, last_stmt_info, type_out,
LSHIFT_EXPR, WIDEN_LSHIFT_EXPR, true,
"vect_recog_widen_shift_pattern");
}
/* Detect a rotate pattern wouldn't be otherwise vectorized:
type a_t, b_t, c_t;
S0 a_t = b_t r<< c_t;
Input/Output:
* STMT_VINFO: The stmt from which the pattern search begins,
i.e. the shift/rotate stmt. The original stmt (S0) is replaced
with a sequence:
S1 d_t = -c_t;
S2 e_t = d_t & (B - 1);
S3 f_t = b_t << c_t;
S4 g_t = b_t >> e_t;
S0 a_t = f_t | g_t;
where B is element bitsize of type.
Output:
* TYPE_OUT: The type of the output of this pattern.
* Return value: A new stmt that will be used to replace the rotate
S0 stmt. */
static gimple *
vect_recog_rotate_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo, tree *type_out)
{
gimple *last_stmt = stmt_vinfo->stmt;
tree oprnd0, oprnd1, lhs, var, var1, var2, vectype, type, stype, def, def2;
gimple *pattern_stmt, *def_stmt;
enum tree_code rhs_code;
enum vect_def_type dt;
optab optab1, optab2;
edge ext_def = NULL;
bool bswap16_p = false;
if (is_gimple_assign (last_stmt))
{
rhs_code = gimple_assign_rhs_code (last_stmt);
switch (rhs_code)
{
case LROTATE_EXPR:
case RROTATE_EXPR:
break;
default:
return NULL;
}
lhs = gimple_assign_lhs (last_stmt);
oprnd0 = gimple_assign_rhs1 (last_stmt);
type = TREE_TYPE (oprnd0);
oprnd1 = gimple_assign_rhs2 (last_stmt);
}
else if (gimple_call_builtin_p (last_stmt, BUILT_IN_BSWAP16))
{
/* __builtin_bswap16 (x) is another form of x r>> 8.
The vectorizer has bswap support, but only if the argument isn't
promoted. */
lhs = gimple_call_lhs (last_stmt);
oprnd0 = gimple_call_arg (last_stmt, 0);
type = TREE_TYPE (oprnd0);
if (!lhs
|| TYPE_PRECISION (TREE_TYPE (lhs)) != 16
|| TYPE_PRECISION (type) <= 16
|| TREE_CODE (oprnd0) != SSA_NAME
|| BITS_PER_UNIT != 8)
return NULL;
stmt_vec_info def_stmt_info;
if (!vect_is_simple_use (oprnd0, vinfo, &dt, &def_stmt_info, &def_stmt))
return NULL;
if (dt != vect_internal_def)
return NULL;
if (gimple_assign_cast_p (def_stmt))
{
def = gimple_assign_rhs1 (def_stmt);
if (INTEGRAL_TYPE_P (TREE_TYPE (def))
&& TYPE_PRECISION (TREE_TYPE (def)) == 16)
oprnd0 = def;
}
type = TREE_TYPE (lhs);
vectype = get_vectype_for_scalar_type (vinfo, type);
if (vectype == NULL_TREE)
return NULL;
if (tree char_vectype = get_same_sized_vectype (char_type_node, vectype))
{
/* The encoding uses one stepped pattern for each byte in the
16-bit word. */
vec_perm_builder elts (TYPE_VECTOR_SUBPARTS (char_vectype), 2, 3);
for (unsigned i = 0; i < 3; ++i)
for (unsigned j = 0; j < 2; ++j)
elts.quick_push ((i + 1) * 2 - j - 1);
vec_perm_indices indices (elts, 1,
TYPE_VECTOR_SUBPARTS (char_vectype));
machine_mode vmode = TYPE_MODE (char_vectype);
if (can_vec_perm_const_p (vmode, vmode, indices))
{
/* vectorizable_bswap can handle the __builtin_bswap16 if we
undo the argument promotion. */
if (!useless_type_conversion_p (type, TREE_TYPE (oprnd0)))
{
def = vect_recog_temp_ssa_var (type, NULL);
def_stmt = gimple_build_assign (def, NOP_EXPR, oprnd0);
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt);
oprnd0 = def;
}
/* Pattern detected. */
vect_pattern_detected ("vect_recog_rotate_pattern", last_stmt);
*type_out = vectype;
/* Pattern supported. Create a stmt to be used to replace the
pattern, with the unpromoted argument. */
var = vect_recog_temp_ssa_var (type, NULL);
pattern_stmt = gimple_build_call (gimple_call_fndecl (last_stmt),
1, oprnd0);
gimple_call_set_lhs (pattern_stmt, var);
gimple_call_set_fntype (as_a <gcall *> (pattern_stmt),
gimple_call_fntype (last_stmt));
return pattern_stmt;
}
}
oprnd1 = build_int_cst (integer_type_node, 8);
rhs_code = LROTATE_EXPR;
bswap16_p = true;
}
else
return NULL;
if (TREE_CODE (oprnd0) != SSA_NAME
|| TYPE_PRECISION (TREE_TYPE (lhs)) != TYPE_PRECISION (type)
|| !INTEGRAL_TYPE_P (type))
return NULL;
stmt_vec_info def_stmt_info;
if (!vect_is_simple_use (oprnd1, vinfo, &dt, &def_stmt_info, &def_stmt))
return NULL;
if (dt != vect_internal_def
&& dt != vect_constant_def
&& dt != vect_external_def)
return NULL;
vectype = get_vectype_for_scalar_type (vinfo, type);
if (vectype == NULL_TREE)
return NULL;
/* If vector/vector or vector/scalar rotate is supported by the target,
don't do anything here. */
optab1 = optab_for_tree_code (rhs_code, vectype, optab_vector);
if (optab1
&& optab_handler (optab1, TYPE_MODE (vectype)) != CODE_FOR_nothing)
{
use_rotate:
if (bswap16_p)
{
if (!useless_type_conversion_p (type, TREE_TYPE (oprnd0)))
{
def = vect_recog_temp_ssa_var (type, NULL);
def_stmt = gimple_build_assign (def, NOP_EXPR, oprnd0);
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt);
oprnd0 = def;
}
/* Pattern detected. */
vect_pattern_detected ("vect_recog_rotate_pattern", last_stmt);
*type_out = vectype;
/* Pattern supported. Create a stmt to be used to replace the
pattern. */
var = vect_recog_temp_ssa_var (type, NULL);
pattern_stmt = gimple_build_assign (var, LROTATE_EXPR, oprnd0,
oprnd1);
return pattern_stmt;
}
return NULL;
}
if (is_a <bb_vec_info> (vinfo) || dt != vect_internal_def)
{
optab2 = optab_for_tree_code (rhs_code, vectype, optab_scalar);
if (optab2
&& optab_handler (optab2, TYPE_MODE (vectype)) != CODE_FOR_nothing)
goto use_rotate;
}
tree utype = unsigned_type_for (type);
tree uvectype = get_vectype_for_scalar_type (vinfo, utype);
if (!uvectype)
return NULL;
/* If vector/vector or vector/scalar shifts aren't supported by the target,
don't do anything here either. */
optab1 = optab_for_tree_code (LSHIFT_EXPR, uvectype, optab_vector);
optab2 = optab_for_tree_code (RSHIFT_EXPR, uvectype, optab_vector);
if (!optab1
|| optab_handler (optab1, TYPE_MODE (uvectype)) == CODE_FOR_nothing
|| !optab2
|| optab_handler (optab2, TYPE_MODE (uvectype)) == CODE_FOR_nothing)
{
if (! is_a <bb_vec_info> (vinfo) && dt == vect_internal_def)
return NULL;
optab1 = optab_for_tree_code (LSHIFT_EXPR, uvectype, optab_scalar);
optab2 = optab_for_tree_code (RSHIFT_EXPR, uvectype, optab_scalar);
if (!optab1
|| optab_handler (optab1, TYPE_MODE (uvectype)) == CODE_FOR_nothing
|| !optab2
|| optab_handler (optab2, TYPE_MODE (uvectype)) == CODE_FOR_nothing)
return NULL;
}
*type_out = vectype;
if (!useless_type_conversion_p (utype, TREE_TYPE (oprnd0)))
{
def = vect_recog_temp_ssa_var (utype, NULL);
def_stmt = gimple_build_assign (def, NOP_EXPR, oprnd0);
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt);
oprnd0 = def;
}
if (dt == vect_external_def && TREE_CODE (oprnd1) == SSA_NAME)
ext_def = vect_get_external_def_edge (vinfo, oprnd1);
def = NULL_TREE;
scalar_int_mode mode = SCALAR_INT_TYPE_MODE (utype);
if (dt != vect_internal_def || TYPE_MODE (TREE_TYPE (oprnd1)) == mode)
def = oprnd1;
else if (def_stmt && gimple_assign_cast_p (def_stmt))
{
tree rhs1 = gimple_assign_rhs1 (def_stmt);
if (TYPE_MODE (TREE_TYPE (rhs1)) == mode
&& TYPE_PRECISION (TREE_TYPE (rhs1))
== TYPE_PRECISION (type))
def = rhs1;
}
if (def == NULL_TREE)
{
def = vect_recog_temp_ssa_var (utype, NULL);
def_stmt = gimple_build_assign (def, NOP_EXPR, oprnd1);
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt);
}
stype = TREE_TYPE (def);
if (TREE_CODE (def) == INTEGER_CST)
{
if (!tree_fits_uhwi_p (def)
|| tree_to_uhwi (def) >= GET_MODE_PRECISION (mode)
|| integer_zerop (def))
return NULL;
def2 = build_int_cst (stype,
GET_MODE_PRECISION (mode) - tree_to_uhwi (def));
}
else
{
tree vecstype = get_vectype_for_scalar_type (vinfo, stype);
if (vecstype == NULL_TREE)
return NULL;
def2 = vect_recog_temp_ssa_var (stype, NULL);
def_stmt = gimple_build_assign (def2, NEGATE_EXPR, def);
if (ext_def)
{
basic_block new_bb
= gsi_insert_on_edge_immediate (ext_def, def_stmt);
gcc_assert (!new_bb);
}
else
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt, vecstype);
def2 = vect_recog_temp_ssa_var (stype, NULL);
tree mask = build_int_cst (stype, GET_MODE_PRECISION (mode) - 1);
def_stmt = gimple_build_assign (def2, BIT_AND_EXPR,
gimple_assign_lhs (def_stmt), mask);
if (ext_def)
{
basic_block new_bb
= gsi_insert_on_edge_immediate (ext_def, def_stmt);
gcc_assert (!new_bb);
}
else
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt, vecstype);
}
var1 = vect_recog_temp_ssa_var (utype, NULL);
def_stmt = gimple_build_assign (var1, rhs_code == LROTATE_EXPR
? LSHIFT_EXPR : RSHIFT_EXPR,
oprnd0, def);
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt);
var2 = vect_recog_temp_ssa_var (utype, NULL);
def_stmt = gimple_build_assign (var2, rhs_code == LROTATE_EXPR
? RSHIFT_EXPR : LSHIFT_EXPR,
oprnd0, def2);
append_pattern_def_seq (vinfo, stmt_vinfo, def_stmt);
/* Pattern detected. */
vect_pattern_detected ("vect_recog_rotate_pattern", last_stmt);
/* Pattern supported. Create a stmt to be used to replace the pattern. */
var = vect_recog_temp_ssa_var (utype, NULL);
pattern_stmt = gimple_build_assign (var, BIT_IOR_EXPR, var1, var2);
if (!useless_type_conversion_p (type, utype))
{
append_pattern_def_seq (vinfo, stmt_vinfo, pattern_stmt);
tree result = vect_recog_temp_ssa_var (type, NULL);
pattern_stmt = gimple_build_assign (result, NOP_EXPR, var);
}
return pattern_stmt;
}
/* Detect a vector by vector shift pattern that wouldn't be otherwise
vectorized:
type a_t;
TYPE b_T, res_T;
S1 a_t = ;
S2 b_T = ;
S3 res_T = b_T op a_t;
where type 'TYPE' is a type with different size than 'type',
and op is <<, >> or rotate.
Also detect cases:
type a_t;
TYPE b_T, c_T, res_T;
S0 c_T = ;
S1 a_t = (type) c_T;
S2 b_T = ;
S3 res_T = b_T op a_t;
Input/Output:
* STMT_VINFO: The stmt from which the pattern search begins,
i.e. the shift/rotate stmt. The original stmt (S3) is replaced
with a shift/rotate which has same type on both operands, in the
second case just b_T op c_T, in the first case with added cast
from a_t to c_T in STMT_VINFO_PATTERN_DEF_SEQ.
Output:
* TYPE_OUT: The type of the output of this pattern.
* Return value: A new stmt that will be used to replace the shift/rotate
S3 stmt. */
static gimple *
vect_recog_vector_vector_shift_pattern (vec_info *vinfo,
stmt_vec_info stmt_vinfo,
tree *type_out)
{
gimple *last_stmt = stmt_vinfo->stmt;
tree oprnd0, oprnd1, lhs, var;
gimple *pattern_stmt;
enum tree_code rhs_code;
if (!is_gimple_assign (last_stmt))
return NULL;
rhs_code = gimple_assign_rhs_code (last_stmt);
switch (rhs_code)
{
case LSHIFT_EXPR:
case RSHIFT_EXPR:
case LROTATE_EXPR:
case RROTATE_EXPR:
break;
default:
return NULL;
}
lhs = gimple_assign_lhs (last_stmt);
oprnd0 = gimple_assign_rhs1 (last_stmt);
oprnd1 = gimple_assign_rhs2 (last_stmt);
if (TREE_CODE (oprnd0) != SSA_NAME
|| TREE_CODE (oprnd1) != SSA_NAME
|| TYPE_MODE (TREE_TYPE (oprnd0)) == TYPE_MODE (TREE_TYPE (oprnd1))
|| !type_has_mode_precision_p (TREE_TYPE (oprnd1))
|| TYPE_PRECISION (TREE_TYPE (lhs))
!= TYPE_PRECISION (TREE_TYPE (oprnd0)))
return NULL;
stmt_vec_info def_vinfo = vect_get_internal_def (vinfo, oprnd1);
if (!def_vinfo)
return NULL;
*type_out = get_vectype_for_scalar_type (vinfo, TREE_TYPE (oprnd0));
if (*type_out == NULL_TREE)
return NULL;