On Tue, Sep 29, 2026 at 1:32 PM Stefan Schulze Frielinghaus
<[email protected]> wrote:
>
> From: Stefan Schulze Frielinghaus <[email protected]>
>
> A defined zero value for clz/ctz does not necessarily need to coinside
> for scalar and vector modes.
>
> In vect_recog_ctz_ffs_pattern() we test whether clz/ctz has a defined
> zero value for scalars while preparing for vector modes.  Similarly, in
> vect_recog_popcount_clz_ctz_ffs_pattern() when constructing an IFN call
> for a vector mode, the defined zero value for the vector mode must be
> considered and not the scalar one.
>
> With this patch we now call into target hooks
> C[LT]Z_DEFINED_VALUE_AT_ZERO with a vector mode for which targets must
> be prepared.  Targets which are prepared in the sense that they can
> process vector modes because they utilize GET_MODE_UNIT_BITSIZE are
> aarch64, arm, gcn, loongarch, mips, riscv, s390.
>
> Targets which need to be adapated are i386 and rs6000 due to usage of
> GET_MODE_BITSIZE.  This patch restores the previous behaviour for those
> targets by substituting GET_MODE_BITSIZE by GET_MODE_UNIT_BITSIZE.
>
> Targets nvptx, pru, vax use GET_MODE_BITSIZE but don't have vector
> support or in case of nvptx have no clz/ctz vector support.
>
> Targets alpha, arc, c6x, cris, csky, m68k, nds32, sparc, xtensa make use
> of constants only.  All other targets fall back to the default
> implementation.

I just looked at the vector pattern parts and I belive there's still a hole in
that you ask for a specific vector mode but the vector mode being used
is eventually another, esp. when performing BB vectorization.  Meaning
what's missing is a validation in vectorizable_call which should see the
final vector type/mode being used - but then the choice from pattern
recognition has to be visible as part of the internal function call - I'm not
sure it is?

That said, the assumption was probably the behavior would be consistent
with scalar/vector modes for a single target ...

> gcc/ChangeLog:
>
>         * config/i386/i386.h (CTZ_DEFINED_VALUE_AT_ZERO): Utilize
>         GET_MODE_UNIT_BITSIZE instead of GET_MODE_BITSIZE in order to
>         deal with vector modes.
>         * config/rs6000/rs6000.h (CLZ_DEFINED_VALUE_AT_ZERO): Ditto.
>         (CTZ_DEFINED_VALUE_AT_ZERO): Ditto.
>         * tree-vect-patterns.cc (vect_recog_ctz_ffs_pattern): Use
>         defined zero value for vector modes.
>         (vect_recog_popcount_clz_ctz_ffs_pattern): Test for a defined
>         zero value for vector modes.
> ---
>  gcc/config/i386/i386.h     |  4 ++--
>  gcc/config/rs6000/rs6000.h |  4 ++--
>  gcc/tree-vect-patterns.cc  | 22 ++++++++++------------
>  3 files changed, 14 insertions(+), 16 deletions(-)
>
> diff --git a/gcc/config/i386/i386.h b/gcc/config/i386/i386.h
> index 47725c56090..35563a6144a 100644
> --- a/gcc/config/i386/i386.h
> +++ b/gcc/config/i386/i386.h
> @@ -3060,9 +3060,9 @@ extern void debug_dispatch_window (int);
>  /* The value at zero is only defined for the BMI instructions
>     LZCNT and TZCNT, not the BSR/BSF insns in the original isa.  */
>  #define CTZ_DEFINED_VALUE_AT_ZERO(MODE, VALUE) \
> -       ((VALUE) = GET_MODE_BITSIZE (MODE), TARGET_BMI ? 2 : 0)
> +       ((VALUE) = GET_MODE_UNIT_BITSIZE (MODE), TARGET_BMI ? 2 : 0)
>  #define CLZ_DEFINED_VALUE_AT_ZERO(MODE, VALUE) \
> -       ((VALUE) = GET_MODE_BITSIZE (MODE), TARGET_LZCNT ? 2 : 0)
> +       ((VALUE) = GET_MODE_UNIT_BITSIZE (MODE), TARGET_LZCNT ? 2 : 0)
>
>
>  /* Flags returned by ix86_get_callcvt ().  */
> diff --git a/gcc/config/rs6000/rs6000.h b/gcc/config/rs6000/rs6000.h
> index bcd4c39d24d..98952f11189 100644
> --- a/gcc/config/rs6000/rs6000.h
> +++ b/gcc/config/rs6000/rs6000.h
> @@ -1749,14 +1749,14 @@ typedef struct rs6000_args
>
>  /* The cntlzw and cntlzd instructions return 32 and 64 for input of zero.  */
>  #define CLZ_DEFINED_VALUE_AT_ZERO(MODE, VALUE) \
> -  ((VALUE) = GET_MODE_BITSIZE (MODE), 2)
> +  ((VALUE) = GET_MODE_UNIT_BITSIZE (MODE), 2)
>
>  /* The CTZ patterns that are implemented in terms of CLZ return -1 for input 
> of
>     zero.  The hardware instructions added in Power9 and the sequences using
>     popcount return 32 or 64.  */
>  #define CTZ_DEFINED_VALUE_AT_ZERO(MODE, VALUE)                         \
>    (TARGET_CTZ || TARGET_POPCNTD                                              
>   \
> -   ? ((VALUE) = GET_MODE_BITSIZE (MODE), 2)                            \
> +   ? ((VALUE) = GET_MODE_UNIT_BITSIZE (MODE), 2)                       \
>     : ((VALUE) = -1, 2))
>
>  /* Specify the machine mode that pointers have.
> diff --git a/gcc/tree-vect-patterns.cc b/gcc/tree-vect-patterns.cc
> index 5fcc37f053c..38f65c0dafa 100644
> --- a/gcc/tree-vect-patterns.cc
> +++ b/gcc/tree-vect-patterns.cc
> @@ -1918,7 +1918,7 @@ vect_recog_ctz_ffs_pattern (vec_info *vinfo, 
> stmt_vec_info stmt_vinfo,
>      {
>        ifnnew = IFN_CTZ;
>        defined_at_zero_new
> -       = CTZ_DEFINED_VALUE_AT_ZERO (SCALAR_INT_TYPE_MODE (rhs_type),
> +       = CTZ_DEFINED_VALUE_AT_ZERO (TYPE_MODE (vec_rhs_type),
>                                      val_new) == 2;
>      }
>    else if (direct_internal_fn_supported_p (IFN_CLZ, vec_rhs_type,
> @@ -1926,7 +1926,7 @@ vect_recog_ctz_ffs_pattern (vec_info *vinfo, 
> stmt_vec_info stmt_vinfo,
>      {
>        ifnnew = IFN_CLZ;
>        defined_at_zero_new
> -       = CLZ_DEFINED_VALUE_AT_ZERO (SCALAR_INT_TYPE_MODE (rhs_type),
> +       = CLZ_DEFINED_VALUE_AT_ZERO (TYPE_MODE (vec_rhs_type),
>                                      val_new) == 2;
>      }
>    if ((ifnnew == IFN_LAST
> @@ -2219,6 +2219,10 @@ vect_recog_popcount_clz_ctz_ffs_pattern (vec_info 
> *vinfo,
>    if (TYPE_PRECISION (unprom_diff.type) != TYPE_PRECISION (lhs_type))
>      return NULL;
>
> +  vec_type = get_vectype_for_scalar_type (vinfo, lhs_type);
> +  if (!vec_type)
> +    return NULL;
> +
>    /* Also A should be unsigned or same precision as temp_in, otherwise
>       different builtins/internal functions have different behaviors.  */
>    if (TYPE_PRECISION (unprom_diff.type)
> @@ -2248,8 +2252,7 @@ vect_recog_popcount_clz_ctz_ffs_pattern (vec_info 
> *vinfo,
>             int val1, val2;
>             val1 = tree_to_shwi (gimple_call_arg (call_stmt, 1));
>             int d2
> -             = CLZ_DEFINED_VALUE_AT_ZERO (SCALAR_INT_TYPE_MODE (lhs_type),
> -                                          val2);
> +             = CLZ_DEFINED_VALUE_AT_ZERO (TYPE_MODE (vec_type), val2);
>             if (d2 != 2 || val1 != val2 + addend)
>               return NULL;
>           }
> @@ -2264,8 +2267,7 @@ vect_recog_popcount_clz_ctz_ffs_pattern (vec_info 
> *vinfo,
>             int val1, val2;
>             val1 = tree_to_shwi (gimple_call_arg (call_stmt, 1));
>             int d2
> -             = CTZ_DEFINED_VALUE_AT_ZERO (SCALAR_INT_TYPE_MODE (lhs_type),
> -                                          val2);
> +             = CTZ_DEFINED_VALUE_AT_ZERO (TYPE_MODE (vec_type), val2);
>             if (d2 != 2 || val1 != val2)
>               return NULL;
>           }
> @@ -2277,11 +2279,7 @@ vect_recog_popcount_clz_ctz_ffs_pattern (vec_info 
> *vinfo,
>         gcc_unreachable ();
>        }
>
> -  vec_type = get_vectype_for_scalar_type (vinfo, lhs_type);
>    /* Do it only if the backend has popcount<vector_mode>2 etc. pattern.  */
> -  if (!vec_type)
> -    return NULL;
> -
>    bool supported
>      = direct_internal_fn_supported_p (ifn, vec_type, OPTIMIZE_FOR_SPEED);
>    /* We cannot rely on the reduction check in vect_recog_ctz_ffs_pattern
> @@ -2334,11 +2332,11 @@ vect_recog_popcount_clz_ctz_ffs_pattern (vec_info 
> *vinfo,
>    tree arg2 = NULL_TREE;
>    int val;
>    if (ifn == IFN_CLZ
> -      && CLZ_DEFINED_VALUE_AT_ZERO (SCALAR_INT_TYPE_MODE (lhs_type),
> +      && CLZ_DEFINED_VALUE_AT_ZERO (TYPE_MODE (vec_type),
>                                     val) == 2)
>      arg2 = build_int_cst (integer_type_node, val);
>    else if (ifn == IFN_CTZ
> -          && CTZ_DEFINED_VALUE_AT_ZERO (SCALAR_INT_TYPE_MODE (lhs_type),
> +          && CTZ_DEFINED_VALUE_AT_ZERO (TYPE_MODE (vec_type),
>                                          val) == 2)
>      arg2 = build_int_cst (integer_type_node, val);
>    if (arg2)
> --
> 2.55.0
>

Reply via email to