[PATCH v2] x86: Remove __mmask intrinsics in strstr-avx512.c

Sunil Pandey skpgkp2@gmail.com
Mon Jul 11 23:32:28 GMT 2022


On Mon, Jul 11, 2022 at 3:38 PM Noah Goldstein via Libc-alpha
<libc-alpha@sourceware.org> wrote:
>
> Using standard operators generates and the same code and __mmask
> instrinsics are not available before GCC7.
>
> Removed:
>     _cvtmask64_u64
>     _kshiftri_mask64
>     _kand_mask64
> ---
>  sysdeps/x86_64/multiarch/strstr-avx512.c | 16 ++++++++++------
>  1 file changed, 10 insertions(+), 6 deletions(-)
>
> diff --git a/sysdeps/x86_64/multiarch/strstr-avx512.c b/sysdeps/x86_64/multiarch/strstr-avx512.c
> index 2ab9e96db8..e44c1a05dc 100644
> --- a/sysdeps/x86_64/multiarch/strstr-avx512.c
> +++ b/sysdeps/x86_64/multiarch/strstr-avx512.c
> @@ -26,6 +26,10 @@
>  #define ZMM_SIZE_IN_BYTES 64
>  #define PAGESIZE 4096
>
> +#define cvtmask64_u64(...) (uint64_t) (__VA_ARGS__)
> +#define kshiftri_mask64(x, y) ((x) >> (y))
> +#define kand_mask64(x, y) ((x) & (y))
> +
>  /*
>   Returns the index of the first edge within the needle, returns 0 if no edge
>   is found. Example: 'ab' is the first edge in 'aaaaaaaaaabaarddg'
> @@ -133,15 +137,15 @@ __strstr_avx512 (const char *haystack, const char *ned)
>    __m512i hay0 = _mm512_maskz_loadu_epi8 (loadmask, haystack + hay_index);
>    /* Search for NULL and compare only till null char */
>    uint64_t nullmask
> -      = _cvtmask64_u64 (_mm512_mask_testn_epi8_mask (loadmask, hay0, hay0));
> +      = cvtmask64_u64 (_mm512_mask_testn_epi8_mask (loadmask, hay0, hay0));
>    uint64_t cmpmask = nullmask ^ (nullmask - ONE_64BIT);
> -  cmpmask = cmpmask & _cvtmask64_u64 (loadmask);
> +  cmpmask = cmpmask & cvtmask64_u64 (loadmask);
>    /* Search for the 2 charaters of needle */
>    __mmask64 k0 = _mm512_cmpeq_epi8_mask (hay0, ned0);
>    __mmask64 k1 = _mm512_cmpeq_epi8_mask (hay0, ned1);
> -  k1 = _kshiftri_mask64 (k1, 1);
> +  k1 = kshiftri_mask64 (k1, 1);
>    /* k2 masks tell us if both chars from needle match */
> -  uint64_t k2 = _cvtmask64_u64 (_kand_mask64 (k0, k1)) & cmpmask;
> +  uint64_t k2 = cvtmask64_u64 (kand_mask64 (k0, k1)) & cmpmask;
>    /* For every match, search for the entire needle for a full match */
>    while (k2)
>      {
> @@ -178,13 +182,13 @@ __strstr_avx512 (const char *haystack, const char *ned)
>        hay0 = _mm512_loadu_si512 (haystack + hay_index);
>        hay1 = _mm512_load_si512 (haystack + hay_index
>                                  + 1); // Always 64 byte aligned
> -      nullmask = _cvtmask64_u64 (_mm512_testn_epi8_mask (hay1, hay1));
> +      nullmask = cvtmask64_u64 (_mm512_testn_epi8_mask (hay1, hay1));
>        /* Compare only till null char */
>        cmpmask = nullmask ^ (nullmask - ONE_64BIT);
>        k0 = _mm512_cmpeq_epi8_mask (hay0, ned0);
>        k1 = _mm512_cmpeq_epi8_mask (hay1, ned1);
>        /* k2 masks tell us if both chars from needle match */
> -      k2 = _cvtmask64_u64 (_kand_mask64 (k0, k1)) & cmpmask;
> +      k2 = cvtmask64_u64 (kand_mask64 (k0, k1)) & cmpmask;
>        /* For every match, compare full strings for potential match */
>        while (k2)
>          {
> --
> 2.34.1
>

LGTM

Reviewed-by: Sunil K Pandey <skpgkp2@gmail.com>


More information about the Libc-alpha mailing list