+2019-09-20 Richard Biener <rguenther@suse.de>
+
+ PR tree-optimization/91821
+ * tree-vect-loop.c (check_reduction_path): Check we can compute
+ reduc_idx.
+ (vect_is_simple_reduction): Set STMT_VINFO_REDUC_IDX.
+ * tree-vect-patterns.c (vect_reassociating_reduction_p): Return
+ operands in canonical order.
+ * tree-vectorizer.c (vec_info::new_stmt_vec_info): Initialize
+ STMT_VINFO_REDUC_IDX.
+ * tree-vectorizer.h (_stmt_vec_info::reduc_idx): New.
+ (STMT_VINFO_REDUC_IDX): Likewise.
+
2019-09-20 Eric Botcazou <ebotcazou@adacore.com>
PR target/91269
gimple *use_stmt = USE_STMT (path[i].second);
tree op = USE_FROM_PTR (path[i].second);
if (! has_single_use (op)
- || ! is_gimple_assign (use_stmt))
+ || ! is_gimple_assign (use_stmt)
+ /* The following make sure we can compute the operand index
+ easily plus it mostly disallows chaining via COND_EXPR condition
+ operands. */
+ || (gimple_assign_rhs1 (use_stmt) != op
+ && gimple_assign_rhs2 (use_stmt) != op
+ && gimple_assign_rhs3 (use_stmt) != op))
{
fail = true;
break;
|| !flow_bb_inside_loop_p (loop, gimple_bb (def1_info->stmt))
|| vect_valid_reduction_input_p (def1_info)))
{
+ STMT_VINFO_REDUC_IDX (def_stmt_info) = 1;
if (dump_enabled_p ())
report_vect_op (MSG_NOTE, def_stmt, "detected reduction: ");
return def_stmt_info;
|| !flow_bb_inside_loop_p (loop, gimple_bb (def2_info->stmt))
|| vect_valid_reduction_input_p (def2_info)))
{
+ STMT_VINFO_REDUC_IDX (def_stmt_info) = 0;
if (dump_enabled_p ())
report_vect_op (MSG_NOTE, def_stmt, "detected reduction: ");
return def_stmt_info;
restriction is that all operations in the chain are the same. */
auto_vec<stmt_vec_info, 8> reduc_chain;
unsigned i;
+ bool is_slp_reduc = !nested_in_vect_loop && code != COND_EXPR;
for (i = path.length () - 1; i >= 1; --i)
{
gimple *stmt = USE_STMT (path[i].second);
if (gimple_assign_rhs_code (stmt) != code)
- break;
- reduc_chain.safe_push (loop_info->lookup_stmt (stmt));
+ is_slp_reduc = false;
+ stmt_vec_info stmt_info = loop_info->lookup_stmt (stmt);
+ STMT_VINFO_REDUC_IDX (stmt_info)
+ = path[i].second->use - gimple_assign_rhs1_ptr (stmt);
+ reduc_chain.safe_push (stmt_info);
}
- if (i == 0
- && ! nested_in_vect_loop
- && code != COND_EXPR)
+ if (is_slp_reduc)
{
for (unsigned i = 0; i < reduc_chain.length () - 1; ++i)
{
vect_force_simple_reduction. */
enum vect_reduction_type reduc_type;
+ /* On a stmt participating in the reduction the index of the operand
+ on the reduction SSA cycle. */
+ int reduc_idx;
+
/* On a reduction PHI the def returned by vect_force_simple_reduction.
On the def returned by vect_force_simple_reduction the
corresponding PHI. */
#define STMT_VINFO_SIMD_LANE_ACCESS_P(S) (S)->simd_lane_access_p
#define STMT_VINFO_VEC_REDUCTION_TYPE(S) (S)->v_reduc_type
#define STMT_VINFO_VEC_CONST_COND_REDUC_CODE(S) (S)->const_cond_reduc_code
+#define STMT_VINFO_REDUC_IDX(S) (S)->reduc_idx
#define STMT_VINFO_DR_WRT_VEC_LOOP(S) (S)->dr_wrt_vec_loop
#define STMT_VINFO_DR_BASE_ADDRESS(S) (S)->dr_wrt_vec_loop.base_address