]> sourceware.org Git - glibc.git/commitdiff
x86: Optimize memcmp-evex-movbe.S
authorNoah Goldstein <goldstein.w.n@gmail.com>
Mon, 17 May 2021 17:57:24 +0000 (13:57 -0400)
committerNoah Goldstein <goldstein.w.n@gmail.com>
Wed, 19 May 2021 02:57:51 +0000 (22:57 -0400)
No bug. This commit optimizes memcmp-evex.S. The optimizations include
adding a new vec compare path for small sizes, reorganizing the entry
control flow, removing some unnecissary ALU instructions from the main
loop, and most importantly replacing the heavy use of vpcmp + kand
logic with vpxor + vptern. test-memcmp and test-wmemcmp are both
passing.

Signed-off-by: Noah Goldstein <goldstein.w.n@gmail.com>
Reviewed-by: H.J. Lu <hjl.tools@gmail.com>
sysdeps/x86_64/multiarch/memcmp-evex-movbe.S

index 9c093972e11fc88b19e65729dce03ca05dd13f67..654dc7ac8ccb9445b2c7107a7cf2d9f6ce4b1010 100644 (file)
 #if IS_IN (libc)
 
 /* memcmp/wmemcmp is implemented as:
-   1. For size from 2 to 7 bytes, load as big endian with movbe and bswap
-      to avoid branches.
-   2. Use overlapping compare to avoid branch.
-   3. Use vector compare when size >= 4 bytes for memcmp or size >= 8
-      bytes for wmemcmp.
-   4. If size is 8 * VEC_SIZE or less, unroll the loop.
-   5. Compare 4 * VEC_SIZE at a time with the aligned first memory
+   1. Use ymm vector compares when possible. The only case where
+      vector compares is not possible for when size < CHAR_PER_VEC
+      and loading from either s1 or s2 would cause a page cross.
+   2. For size from 2 to 7 bytes on page cross, load as big endian
+      with movbe and bswap to avoid branches.
+   3. Use xmm vector compare when size >= 4 bytes for memcmp or
+      size >= 8 bytes for wmemcmp.
+   4. Optimistically compare up to first 4 * CHAR_PER_VEC one at a
+      to check for early mismatches. Only do this if its guranteed the
+      work is not wasted.
+   5. If size is 8 * VEC_SIZE or less, unroll the loop.
+   6. Compare 4 * VEC_SIZE at a time with the aligned first memory
       area.
-   6. Use 2 vector compares when size is 2 * VEC_SIZE or less.
-   7. Use 4 vector compares when size is 4 * VEC_SIZE or less.
-   8. Use 8 vector compares when size is 8 * VEC_SIZE or less.  */
+   7. Use 2 vector compares when size is 2 * CHAR_PER_VEC or less.
+   8. Use 4 vector compares when size is 4 * CHAR_PER_VEC or less.
+   9. Use 8 vector compares when size is 8 * CHAR_PER_VEC or less.  */
 
 # include <sysdep.h>
 
 # define VMOVU         vmovdqu64
 
 # ifdef USE_AS_WMEMCMP
-#  define VPCMPEQ      vpcmpeqd
+#  define CHAR_SIZE    4
+#  define VPCMP        vpcmpd
 # else
-#  define VPCMPEQ      vpcmpeqb
+#  define CHAR_SIZE    1
+#  define VPCMP        vpcmpub
 # endif
 
+# define VEC_SIZE      32
+# define PAGE_SIZE     4096
+# define CHAR_PER_VEC  (VEC_SIZE / CHAR_SIZE)
+
+# define XMM0          xmm16
+# define XMM1          xmm17
+# define XMM2          xmm18
+# define YMM0          ymm16
 # define XMM1          xmm17
 # define XMM2          xmm18
 # define YMM1          ymm17
 # define YMM5          ymm21
 # define YMM6          ymm22
 
-# define VEC_SIZE 32
-# ifdef USE_AS_WMEMCMP
-#  define VEC_MASK 0xff
-#  define XMM_MASK 0xf
-# else
-#  define VEC_MASK 0xffffffff
-#  define XMM_MASK 0xffff
-# endif
-
 /* Warning!
            wmemcmp has to use SIGNED comparison for elements.
            memcmp has to use UNSIGNED comparison for elemnts.
 
        .section .text.evex,"ax",@progbits
 ENTRY (MEMCMP)
-# ifdef USE_AS_WMEMCMP
-       shl     $2, %RDX_LP
-# elif defined __ILP32__
+# ifdef __ILP32__
        /* Clear the upper 32 bits.  */
        movl    %edx, %edx
 # endif
-       cmp     $VEC_SIZE, %RDX_LP
+       cmp     $CHAR_PER_VEC, %RDX_LP
        jb      L(less_vec)
 
        /* From VEC to 2 * VEC.  No branch when size == VEC_SIZE.  */
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k1
+       VMOVU   (%rsi), %YMM1
+       /* Use compare not equals to directly check for mismatch.  */
+       VPCMP   $4, (%rdi), %YMM1, %k1
        kmovd   %k1, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
-
-       cmpq    $(VEC_SIZE * 2), %rdx
-       jbe     L(last_vec)
-
-       /* More than 2 * VEC.  */
-       cmpq    $(VEC_SIZE * 8), %rdx
-       ja      L(more_8x_vec)
-       cmpq    $(VEC_SIZE * 4), %rdx
-       jb      L(last_4x_vec)
+       /* NB: eax must be destination register if going to
+          L(return_vec_[0,2]). For L(return_vec_3 destination register
+          must be ecx.  */
+       testl   %eax, %eax
+       jnz     L(return_vec_0)
 
-       /* From 4 * VEC to 8 * VEC, inclusively. */
-       VMOVU   (%rsi), %YMM1
-       VPCMPEQ (%rdi), %YMM1, %k1
+       cmpq    $(CHAR_PER_VEC * 2), %rdx
+       jbe     L(last_1x_vec)
 
+       /* Check second VEC no matter what.  */
        VMOVU   VEC_SIZE(%rsi), %YMM2
-       VPCMPEQ VEC_SIZE(%rdi), %YMM2, %k2
+       VPCMP   $4, VEC_SIZE(%rdi), %YMM2, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_1)
+
+       /* Less than 4 * VEC.  */
+       cmpq    $(CHAR_PER_VEC * 4), %rdx
+       jbe     L(last_2x_vec)
 
+       /* Check third and fourth VEC no matter what.  */
        VMOVU   (VEC_SIZE * 2)(%rsi), %YMM3
-       VPCMPEQ (VEC_SIZE * 2)(%rdi), %YMM3, %k3
+       VPCMP   $4, (VEC_SIZE * 2)(%rdi), %YMM3, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_2)
 
        VMOVU   (VEC_SIZE * 3)(%rsi), %YMM4
-       VPCMPEQ (VEC_SIZE * 3)(%rdi), %YMM4, %k4
+       VPCMP   $4, (VEC_SIZE * 3)(%rdi), %YMM4, %k1
+       kmovd   %k1, %ecx
+       testl   %ecx, %ecx
+       jnz     L(return_vec_3)
 
-       kandd   %k1, %k2, %k5
-       kandd   %k3, %k4, %k6
-       kandd   %k5, %k6, %k6
+       /* Zero YMM0. 4x VEC reduction is done with vpxor + vtern so
+          compare with zero to get a mask is needed.  */
+       vpxorq  %XMM0, %XMM0, %XMM0
 
-       kmovd   %k6, %eax
-       cmpl    $VEC_MASK, %eax
-       jne     L(4x_vec_end)
+       /* Go to 4x VEC loop.  */
+       cmpq    $(CHAR_PER_VEC * 8), %rdx
+       ja      L(more_8x_vec)
 
-       leaq    -(4 * VEC_SIZE)(%rdi, %rdx), %rdi
-       leaq    -(4 * VEC_SIZE)(%rsi, %rdx), %rsi
-       VMOVU   (%rsi), %YMM1
-       VPCMPEQ (%rdi), %YMM1, %k1
+       /* Handle remainder of size = 4 * VEC + 1 to 8 * VEC without any
+          branches.  */
 
-       VMOVU   VEC_SIZE(%rsi), %YMM2
-       VPCMPEQ VEC_SIZE(%rdi), %YMM2, %k2
-       kandd   %k1, %k2, %k5
+       /* Load first two VEC from s2 before adjusting addresses.  */
+       VMOVU   -(VEC_SIZE * 4)(%rsi, %rdx, CHAR_SIZE), %YMM1
+       VMOVU   -(VEC_SIZE * 3)(%rsi, %rdx, CHAR_SIZE), %YMM2
+       leaq    -(4 * VEC_SIZE)(%rdi, %rdx, CHAR_SIZE), %rdi
+       leaq    -(4 * VEC_SIZE)(%rsi, %rdx, CHAR_SIZE), %rsi
+
+       /* Wait to load from s1 until addressed adjust due to
+          unlamination of microfusion with complex address mode.  */
+
+       /* vpxor will be all 0s if s1 and s2 are equal. Otherwise it
+          will have some 1s.  */
+       vpxorq  (%rdi), %YMM1, %YMM1
+       vpxorq  (VEC_SIZE)(%rdi), %YMM2, %YMM2
 
        VMOVU   (VEC_SIZE * 2)(%rsi), %YMM3
-       VPCMPEQ (VEC_SIZE * 2)(%rdi), %YMM3, %k3
-       kandd   %k3, %k5, %k5
+       vpxorq  (VEC_SIZE * 2)(%rdi), %YMM3, %YMM3
+       /* Or together YMM1, YMM2, and YMM3 into YMM3.  */
+       vpternlogd $0xfe, %YMM1, %YMM2, %YMM3
 
        VMOVU   (VEC_SIZE * 3)(%rsi), %YMM4
-       VPCMPEQ (VEC_SIZE * 3)(%rdi), %YMM4, %k4
-       kandd   %k4, %k5, %k5
+       /* Ternary logic to xor (VEC_SIZE * 3)(%rdi) with YMM4 while
+          oring with YMM3. Result is stored in YMM4.  */
+       vpternlogd $0xde, (VEC_SIZE * 3)(%rdi), %YMM3, %YMM4
+       /* Compare YMM4 with 0. If any 1s s1 and s2 don't match.  */
+       VPCMP   $4, %YMM4, %YMM0, %k1
+       kmovd   %k1, %ecx
+       testl   %ecx, %ecx
+       jnz     L(return_vec_0_1_2_3)
+       /* NB: eax must be zero to reach here.  */
+       ret
 
-       kmovd   %k5, %eax
-       cmpl    $VEC_MASK, %eax
-       jne     L(4x_vec_end)
-       xorl    %eax, %eax
+       /* NB: aligning 32 here allows for the rest of the jump targets
+          to be tuned for 32 byte alignment. Most important this ensures
+          the L(more_8x_vec) loop is 32 byte aligned.  */
+       .p2align 5
+L(less_vec):
+       /* Check if one or less CHAR. This is necessary for size = 0 but
+          is also faster for size = CHAR_SIZE.  */
+       cmpl    $1, %edx
+       jbe     L(one_or_less)
+
+       /* Check if loading one VEC from either s1 or s2 could cause a
+          page cross. This can have false positives but is by far the
+          fastest method.  */
+       movl    %edi, %eax
+       orl     %esi, %eax
+       andl    $(PAGE_SIZE - 1), %eax
+       cmpl    $(PAGE_SIZE - VEC_SIZE), %eax
+       jg      L(page_cross_less_vec)
+
+       /* No page cross possible.  */
+       VMOVU   (%rsi), %YMM2
+       VPCMP   $4, (%rdi), %YMM2, %k1
+       kmovd   %k1, %eax
+       /* Create mask in ecx for potentially in bound matches.  */
+       bzhil   %edx, %eax, %eax
+       jnz     L(return_vec_0)
        ret
 
        .p2align 4
-L(last_2x_vec):
-       /* From VEC to 2 * VEC.  No branch when size == VEC_SIZE.  */
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k2
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
+L(return_vec_0):
+       tzcntl  %eax, %eax
+# ifdef USE_AS_WMEMCMP
+       movl    (%rdi, %rax, CHAR_SIZE), %ecx
+       xorl    %edx, %edx
+       cmpl    (%rsi, %rax, CHAR_SIZE), %ecx
+       /* NB: no partial register stall here because xorl zero idiom
+          above.  */
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
+# else
+       movzbl  (%rsi, %rax), %ecx
+       movzbl  (%rdi, %rax), %eax
+       subl    %ecx, %eax
+# endif
+       ret
 
-L(last_vec):
-       /* Use overlapping loads to avoid branches.  */
-       leaq    -VEC_SIZE(%rdi, %rdx), %rdi
-       leaq    -VEC_SIZE(%rsi, %rdx), %rsi
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k2
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
+       /* NB: No p2align necessary. Alignment  % 16 is naturally 1
+          which is good enough for a target not in a loop.  */
+L(return_vec_1):
+       tzcntl  %eax, %eax
+# ifdef USE_AS_WMEMCMP
+       movl    VEC_SIZE(%rdi, %rax, CHAR_SIZE), %ecx
+       xorl    %edx, %edx
+       cmpl    VEC_SIZE(%rsi, %rax, CHAR_SIZE), %ecx
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
+# else
+       movzbl  VEC_SIZE(%rsi, %rax), %ecx
+       movzbl  VEC_SIZE(%rdi, %rax), %eax
+       subl    %ecx, %eax
+# endif
        ret
 
-       .p2align 4
-L(first_vec):
-       /* A byte or int32 is different within 16 or 32 bytes.  */
-       tzcntl  %eax, %ecx
+       /* NB: No p2align necessary. Alignment  % 16 is naturally 2
+          which is good enough for a target not in a loop.  */
+L(return_vec_2):
+       tzcntl  %eax, %eax
 # ifdef USE_AS_WMEMCMP
-       xorl    %eax, %eax
-       movl    (%rdi, %rcx, 4), %edx
-       cmpl    (%rsi, %rcx, 4), %edx
-L(wmemcmp_return):
-       setl    %al
-       negl    %eax
-       orl     $1, %eax
+       movl    (VEC_SIZE * 2)(%rdi, %rax, CHAR_SIZE), %ecx
+       xorl    %edx, %edx
+       cmpl    (VEC_SIZE * 2)(%rsi, %rax, CHAR_SIZE), %ecx
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
 # else
-       movzbl  (%rdi, %rcx), %eax
-       movzbl  (%rsi, %rcx), %edx
-       sub     %edx, %eax
+       movzbl  (VEC_SIZE * 2)(%rsi, %rax), %ecx
+       movzbl  (VEC_SIZE * 2)(%rdi, %rax), %eax
+       subl    %ecx, %eax
 # endif
        ret
 
+       .p2align 4
+L(8x_return_vec_0_1_2_3):
+       /* Returning from L(more_8x_vec) requires restoring rsi.  */
+       addq    %rdi, %rsi
+L(return_vec_0_1_2_3):
+       VPCMP   $4, %YMM1, %YMM0, %k0
+       kmovd   %k0, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_0)
+
+       VPCMP   $4, %YMM2, %YMM0, %k0
+       kmovd   %k0, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_1)
+
+       VPCMP   $4, %YMM3, %YMM0, %k0
+       kmovd   %k0, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_2)
+L(return_vec_3):
+       tzcntl  %ecx, %ecx
 # ifdef USE_AS_WMEMCMP
+       movl    (VEC_SIZE * 3)(%rdi, %rcx, CHAR_SIZE), %eax
+       xorl    %edx, %edx
+       cmpl    (VEC_SIZE * 3)(%rsi, %rcx, CHAR_SIZE), %eax
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
+# else
+       movzbl  (VEC_SIZE * 3)(%rdi, %rcx), %eax
+       movzbl  (VEC_SIZE * 3)(%rsi, %rcx), %ecx
+       subl    %ecx, %eax
+# endif
+       ret
+
        .p2align 4
-L(4):
-       xorl    %eax, %eax
-       movl    (%rdi), %edx
-       cmpl    (%rsi), %edx
-       jne     L(wmemcmp_return)
+L(more_8x_vec):
+       /* Set end of s1 in rdx.  */
+       leaq    -(VEC_SIZE * 4)(%rdi, %rdx, CHAR_SIZE), %rdx
+       /* rsi stores s2 - s1. This allows loop to only update one
+          pointer.  */
+       subq    %rdi, %rsi
+       /* Align s1 pointer.  */
+       andq    $-VEC_SIZE, %rdi
+       /* Adjust because first 4x vec where check already.  */
+       subq    $-(VEC_SIZE * 4), %rdi
+       .p2align 4
+L(loop_4x_vec):
+       VMOVU   (%rsi, %rdi), %YMM1
+       vpxorq  (%rdi), %YMM1, %YMM1
+
+       VMOVU   VEC_SIZE(%rsi, %rdi), %YMM2
+       vpxorq  VEC_SIZE(%rdi), %YMM2, %YMM2
+
+       VMOVU   (VEC_SIZE * 2)(%rsi, %rdi), %YMM3
+       vpxorq  (VEC_SIZE * 2)(%rdi), %YMM3, %YMM3
+       vpternlogd $0xfe, %YMM1, %YMM2, %YMM3
+
+       VMOVU   (VEC_SIZE * 3)(%rsi, %rdi), %YMM4
+       vpternlogd $0xde, (VEC_SIZE * 3)(%rdi), %YMM3, %YMM4
+       VPCMP   $4, %YMM4, %YMM0, %k1
+       kmovd   %k1, %ecx
+       testl   %ecx, %ecx
+       jnz     L(8x_return_vec_0_1_2_3)
+       subq    $-(VEC_SIZE * 4), %rdi
+       cmpq    %rdx, %rdi
+       jb      L(loop_4x_vec)
+
+       subq    %rdx, %rdi
+       /* rdi has 4 * VEC_SIZE - remaining length.  */
+       cmpl    $(VEC_SIZE * 3), %edi
+       jae     L(8x_last_1x_vec)
+       /* Load regardless of branch.  */
+       VMOVU   (VEC_SIZE * 2)(%rsi, %rdx), %YMM3
+       cmpl    $(VEC_SIZE * 2), %edi
+       jae     L(8x_last_2x_vec)
+
+       VMOVU   (%rsi, %rdx), %YMM1
+       vpxorq  (%rdx), %YMM1, %YMM1
+
+       VMOVU   VEC_SIZE(%rsi, %rdx), %YMM2
+       vpxorq  VEC_SIZE(%rdx), %YMM2, %YMM2
+
+       vpxorq  (VEC_SIZE * 2)(%rdx), %YMM3, %YMM3
+       vpternlogd $0xfe, %YMM1, %YMM2, %YMM3
+
+       VMOVU   (VEC_SIZE * 3)(%rsi, %rdx), %YMM4
+       vpternlogd $0xde, (VEC_SIZE * 3)(%rdx), %YMM3, %YMM4
+       VPCMP   $4, %YMM4, %YMM0, %k1
+       kmovd   %k1, %ecx
+       /* Restore s1 pointer to rdi.  */
+       movq    %rdx, %rdi
+       testl   %ecx, %ecx
+       jnz     L(8x_return_vec_0_1_2_3)
+       /* NB: eax must be zero to reach here.  */
+       ret
+
+       /* Only entry is from L(more_8x_vec).  */
+       .p2align 4
+L(8x_last_2x_vec):
+       VPCMP   $4, (VEC_SIZE * 2)(%rdx), %YMM3, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(8x_return_vec_2)
+       /* Naturally aligned to 16 bytes.  */
+L(8x_last_1x_vec):
+       VMOVU   (VEC_SIZE * 3)(%rsi, %rdx), %YMM1
+       VPCMP   $4, (VEC_SIZE * 3)(%rdx), %YMM1, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(8x_return_vec_3)
+       ret
+
+       .p2align 4
+L(last_2x_vec):
+       /* Check second to last VEC.  */
+       VMOVU   -(VEC_SIZE * 2)(%rsi, %rdx, CHAR_SIZE), %YMM1
+       VPCMP   $4, -(VEC_SIZE * 2)(%rdi, %rdx, CHAR_SIZE), %YMM1, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_1_end)
+
+       /* Check last VEC.  */
+       .p2align 4
+L(last_1x_vec):
+       VMOVU   -(VEC_SIZE * 1)(%rsi, %rdx, CHAR_SIZE), %YMM1
+       VPCMP   $4, -(VEC_SIZE * 1)(%rdi, %rdx, CHAR_SIZE), %YMM1, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_0_end)
        ret
+
+       .p2align 4
+L(8x_return_vec_2):
+       subq    $VEC_SIZE, %rdx
+L(8x_return_vec_3):
+       tzcntl  %eax, %eax
+# ifdef USE_AS_WMEMCMP
+       leaq    (%rdx, %rax, CHAR_SIZE), %rax
+       movl    (VEC_SIZE * 3)(%rax), %ecx
+       xorl    %edx, %edx
+       cmpl    (VEC_SIZE * 3)(%rsi, %rax), %ecx
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
 # else
+       addq    %rdx, %rax
+       movzbl  (VEC_SIZE * 3)(%rsi, %rax), %ecx
+       movzbl  (VEC_SIZE * 3)(%rax), %eax
+       subl    %ecx, %eax
+# endif
+       ret
+
        .p2align 4
-L(between_4_7):
-       /* Load as big endian with overlapping movbe to avoid branches.  */
-       movbe   (%rdi), %eax
-       movbe   (%rsi), %ecx
-       shlq    $32, %rax
-       shlq    $32, %rcx
-       movbe   -4(%rdi, %rdx), %edi
-       movbe   -4(%rsi, %rdx), %esi
-       orq     %rdi, %rax
-       orq     %rsi, %rcx
-       subq    %rcx, %rax
-       je      L(exit)
-       sbbl    %eax, %eax
-       orl     $1, %eax
+L(return_vec_0_end):
+       tzcntl  %eax, %eax
+       addl    %edx, %eax
+# ifdef USE_AS_WMEMCMP
+       movl    -VEC_SIZE(%rdi, %rax, CHAR_SIZE), %ecx
+       xorl    %edx, %edx
+       cmpl    -VEC_SIZE(%rsi, %rax, CHAR_SIZE), %ecx
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
+# else
+       movzbl  -VEC_SIZE(%rsi, %rax), %ecx
+       movzbl  -VEC_SIZE(%rdi, %rax), %eax
+       subl    %ecx, %eax
+# endif
        ret
 
        .p2align 4
-L(exit):
+L(return_vec_1_end):
+       tzcntl  %eax, %eax
+       addl    %edx, %eax
+# ifdef USE_AS_WMEMCMP
+       movl    -(VEC_SIZE * 2)(%rdi, %rax, CHAR_SIZE), %ecx
+       xorl    %edx, %edx
+       cmpl    -(VEC_SIZE * 2)(%rsi, %rax, CHAR_SIZE), %ecx
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
+# else
+       movzbl  -(VEC_SIZE * 2)(%rsi, %rax), %ecx
+       movzbl  -(VEC_SIZE * 2)(%rdi, %rax), %eax
+       subl    %ecx, %eax
+# endif
        ret
 
+
        .p2align 4
+L(page_cross_less_vec):
+       /* if USE_AS_WMEMCMP it can only be 0, 4, 8, 12, 16, 20, 24, 28
+          bytes.  */
+       cmpl    $(16 / CHAR_SIZE), %edx
+       jae     L(between_16_31)
+# ifndef USE_AS_WMEMCMP
+       cmpl    $8, %edx
+       jae     L(between_8_15)
+       cmpl    $4, %edx
+       jae     L(between_4_7)
 L(between_2_3):
        /* Load as big endian to avoid branches.  */
        movzwl  (%rdi), %eax
@@ -217,224 +448,99 @@ L(between_2_3):
        shll    $8, %ecx
        bswap   %eax
        bswap   %ecx
-       movb    -1(%rdi, %rdx), %al
-       movb    -1(%rsi, %rdx), %cl
+       movzbl  -1(%rdi, %rdx), %edi
+       movzbl  -1(%rsi, %rdx), %esi
+       orl     %edi, %eax
+       orl     %esi, %ecx
        /* Subtraction is okay because the upper 8 bits are zero.  */
        subl    %ecx, %eax
        ret
-
        .p2align 4
-L(1):
-       movzbl  (%rdi), %eax
+L(one_or_less):
+       jb      L(zero)
        movzbl  (%rsi), %ecx
+       movzbl  (%rdi), %eax
        subl    %ecx, %eax
        ret
-# endif
-
-       .p2align 4
-L(zero):
-       xorl    %eax, %eax
-       ret
 
        .p2align 4
-L(less_vec):
-# ifdef USE_AS_WMEMCMP
-       /* It can only be 0, 4, 8, 12, 16, 20, 24, 28 bytes.  */
-       cmpb    $4, %dl
-       je      L(4)
-       jb      L(zero)
-# else
-       cmpb    $1, %dl
-       je      L(1)
-       jb      L(zero)
-       cmpb    $4, %dl
-       jb      L(between_2_3)
-       cmpb    $8, %dl
-       jb      L(between_4_7)
+L(between_8_15):
 # endif
-       cmpb    $16, %dl
-       jae     L(between_16_31)
-       /* It is between 8 and 15 bytes.  */
+       /* If USE_AS_WMEMCMP fall through into 8-15 byte case.  */
        vmovq   (%rdi), %XMM1
        vmovq   (%rsi), %XMM2
-       VPCMPEQ %XMM1, %XMM2, %k2
-       kmovw   %k2, %eax
-       subl    $XMM_MASK, %eax
-       jnz     L(first_vec)
+       VPCMP   $4, %XMM1, %XMM2, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_0)
        /* Use overlapping loads to avoid branches.  */
-       leaq    -8(%rdi, %rdx), %rdi
-       leaq    -8(%rsi, %rdx), %rsi
+       leaq    -8(%rdi, %rdx, CHAR_SIZE), %rdi
+       leaq    -8(%rsi, %rdx, CHAR_SIZE), %rsi
        vmovq   (%rdi), %XMM1
        vmovq   (%rsi), %XMM2
-       VPCMPEQ %XMM1, %XMM2, %k2
-       kmovw   %k2, %eax
-       subl    $XMM_MASK, %eax
-       jnz     L(first_vec)
+       VPCMP   $4, %XMM1, %XMM2, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_0)
        ret
 
        .p2align 4
-L(between_16_31):
-       /* From 16 to 31 bytes.  No branch when size == 16.  */
-       VMOVU   (%rsi), %XMM2
-       VPCMPEQ (%rdi), %XMM2, %k2
-       kmovw   %k2, %eax
-       subl    $XMM_MASK, %eax
-       jnz     L(first_vec)
-
-       /* Use overlapping loads to avoid branches.  */
-       leaq    -16(%rdi, %rdx), %rdi
-       leaq    -16(%rsi, %rdx), %rsi
-       VMOVU   (%rsi), %XMM2
-       VPCMPEQ (%rdi), %XMM2, %k2
-       kmovw   %k2, %eax
-       subl    $XMM_MASK, %eax
-       jnz     L(first_vec)
+L(zero):
+       xorl    %eax, %eax
        ret
 
        .p2align 4
-L(more_8x_vec):
-       /* More than 8 * VEC.  Check the first VEC.  */
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k2
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
-
-       /* Align the first memory area for aligned loads in the loop.
-          Compute how much the first memory area is misaligned.  */
-       movq    %rdi, %rcx
-       andl    $(VEC_SIZE - 1), %ecx
-       /* Get the negative of offset for alignment.  */
-       subq    $VEC_SIZE, %rcx
-       /* Adjust the second memory area.  */
-       subq    %rcx, %rsi
-       /* Adjust the first memory area which should be aligned now.  */
-       subq    %rcx, %rdi
-       /* Adjust length.  */
-       addq    %rcx, %rdx
-
-L(loop_4x_vec):
-       /* Compare 4 * VEC at a time forward.  */
-       VMOVU   (%rsi), %YMM1
-       VPCMPEQ (%rdi), %YMM1, %k1
-
-       VMOVU   VEC_SIZE(%rsi), %YMM2
-       VPCMPEQ VEC_SIZE(%rdi), %YMM2, %k2
-       kandd   %k2, %k1, %k5
-
-       VMOVU   (VEC_SIZE * 2)(%rsi), %YMM3
-       VPCMPEQ (VEC_SIZE * 2)(%rdi), %YMM3, %k3
-       kandd   %k3, %k5, %k5
-
-       VMOVU   (VEC_SIZE * 3)(%rsi), %YMM4
-       VPCMPEQ (VEC_SIZE * 3)(%rdi), %YMM4, %k4
-       kandd   %k4, %k5, %k5
-
-       kmovd   %k5, %eax
-       cmpl    $VEC_MASK, %eax
-       jne     L(4x_vec_end)
-
-       addq    $(VEC_SIZE * 4), %rdi
-       addq    $(VEC_SIZE * 4), %rsi
-
-       subq    $(VEC_SIZE * 4), %rdx
-       cmpq    $(VEC_SIZE * 4), %rdx
-       jae     L(loop_4x_vec)
-
-       /* Less than 4 * VEC.  */
-       cmpq    $VEC_SIZE, %rdx
-       jbe     L(last_vec)
-       cmpq    $(VEC_SIZE * 2), %rdx
-       jbe     L(last_2x_vec)
-
-L(last_4x_vec):
-       /* From 2 * VEC to 4 * VEC. */
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k2
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
-
-       addq    $VEC_SIZE, %rdi
-       addq    $VEC_SIZE, %rsi
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k2
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
+L(between_16_31):
+       /* From 16 to 31 bytes.  No branch when size == 16.  */
+       VMOVU   (%rsi), %XMM2
+       VPCMP   $4, (%rdi), %XMM2, %k1
+       kmovd   %k1, %eax
+       testl   %eax, %eax
+       jnz     L(return_vec_0)
 
        /* Use overlapping loads to avoid branches.  */
-       leaq    -(3 * VEC_SIZE)(%rdi, %rdx), %rdi
-       leaq    -(3 * VEC_SIZE)(%rsi, %rdx), %rsi
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k2
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
 
-       addq    $VEC_SIZE, %rdi
-       addq    $VEC_SIZE, %rsi
-       VMOVU   (%rsi), %YMM2
-       VPCMPEQ (%rdi), %YMM2, %k2
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
-       ret
-
-       .p2align 4
-L(4x_vec_end):
+       VMOVU   -16(%rsi, %rdx, CHAR_SIZE), %XMM2
+       leaq    -16(%rdi, %rdx, CHAR_SIZE), %rdi
+       leaq    -16(%rsi, %rdx, CHAR_SIZE), %rsi
+       VPCMP   $4, (%rdi), %XMM2, %k1
        kmovd   %k1, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec)
-       kmovd   %k2, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec_x1)
-       kmovd   %k3, %eax
-       subl    $VEC_MASK, %eax
-       jnz     L(first_vec_x2)
-       kmovd   %k4, %eax
-       subl    $VEC_MASK, %eax
-       tzcntl  %eax, %ecx
-# ifdef USE_AS_WMEMCMP
-       xorl    %eax, %eax
-       movl    (VEC_SIZE * 3)(%rdi, %rcx, 4), %edx
-       cmpl    (VEC_SIZE * 3)(%rsi, %rcx, 4), %edx
-       jmp     L(wmemcmp_return)
-# else
-       movzbl  (VEC_SIZE * 3)(%rdi, %rcx), %eax
-       movzbl  (VEC_SIZE * 3)(%rsi, %rcx), %edx
-       sub     %edx, %eax
-# endif
+       testl   %eax, %eax
+       jnz     L(return_vec_0)
        ret
 
-       .p2align 4
-L(first_vec_x1):
-       tzcntl  %eax, %ecx
 # ifdef USE_AS_WMEMCMP
-       xorl    %eax, %eax
-       movl    VEC_SIZE(%rdi, %rcx, 4), %edx
-       cmpl    VEC_SIZE(%rsi, %rcx, 4), %edx
-       jmp     L(wmemcmp_return)
-# else
-       movzbl  VEC_SIZE(%rdi, %rcx), %eax
-       movzbl  VEC_SIZE(%rsi, %rcx), %edx
-       sub     %edx, %eax
-# endif
+       .p2align 4
+L(one_or_less):
+       jb      L(zero)
+       movl    (%rdi), %ecx
+       xorl    %edx, %edx
+       cmpl    (%rsi), %ecx
+       je      L(zero)
+       setg    %dl
+       leal    -1(%rdx, %rdx), %eax
        ret
+# else
 
        .p2align 4
-L(first_vec_x2):
-       tzcntl  %eax, %ecx
-# ifdef USE_AS_WMEMCMP
-       xorl    %eax, %eax
-       movl    (VEC_SIZE * 2)(%rdi, %rcx, 4), %edx
-       cmpl    (VEC_SIZE * 2)(%rsi, %rcx, 4), %edx
-       jmp     L(wmemcmp_return)
-# else
-       movzbl  (VEC_SIZE * 2)(%rdi, %rcx), %eax
-       movzbl  (VEC_SIZE * 2)(%rsi, %rcx), %edx
-       sub     %edx, %eax
-# endif
+L(between_4_7):
+       /* Load as big endian with overlapping movbe to avoid branches.
+        */
+       movbe   (%rdi), %eax
+       movbe   (%rsi), %ecx
+       shlq    $32, %rax
+       shlq    $32, %rcx
+       movbe   -4(%rdi, %rdx), %edi
+       movbe   -4(%rsi, %rdx), %esi
+       orq     %rdi, %rax
+       orq     %rsi, %rcx
+       subq    %rcx, %rax
+       jz      L(zero_4_7)
+       sbbl    %eax, %eax
+       orl     $1, %eax
+L(zero_4_7):
        ret
+# endif
+
 END (MEMCMP)
 #endif
This page took 0.060946 seconds and 5 git commands to generate.