Re: [RFC: PATCH] Extend vectorizer to handle nonlinear induction for neg, mul/lshift/rshift with a constant.

[Richard Sandiford <richard.sandiford@arm.com>]

Richard Biener <richard.guenther@gmail.com> writes:
>> +/* Create vector init for vectorized iv.  */
>> +static tree
>> +vect_create_nonlinear_iv_init (gimple_seq* stmts, tree init_expr,
>> +                              tree step_expr, poly_uint64 nunits,
>> +                              tree vectype,
>> +                              enum vect_induction_op_type induction_type)
>> +{
>> +  unsigned HOST_WIDE_INT const_nunits;
>> +  tree vec_shift, vec_init, new_name;
>> +  unsigned i;
>> +
>> +  /* iv_loop is the loop to be vectorized. Create:
>> +     vec_init = [X, X+S, X+2*S, X+3*S] (S = step_expr, X = init_expr).  */
>> +  new_name = init_expr;
>> +  switch (induction_type)
>> +    {
>> +    case vect_step_op_shr:
>> +    case vect_step_op_shl:
>> +      /* Build the Initial value from shift_expr.  */
>> +      vec_init = gimple_build_vector_from_val (stmts,
>> +                                              vectype,
>> +                                              new_name);
>> +      vec_shift = gimple_build (stmts, VEC_SERIES_EXPR, vectype,
>> +                               build_zero_cst (TREE_TYPE (step_expr)),
>> +                               step_expr);
>
> There might be a more canonical way to build the series expr - Richard?

build_vec_series is shorter if step_expr is known to be a constant.
The above looks good for the general case.

Thanks,
Richard