From: Szabolcs Nagy <Szabolcs.Nagy@arm.com>
To: Wilco Dijkstra <Wilco.Dijkstra@arm.com>
Cc: 'GNU C Library' <libc-alpha@sourceware.org>
Subject: Re: [PATCH] AArch64: Optimize strnlen
Date: Fri, 13 Jan 2023 12:26:54 +0000 [thread overview]
Message-ID: <Y8FODpWH2v9+lT7R@arm.com> (raw)
In-Reply-To: <PAWPR08MB898209A07333BBB09A5A0D3083FD9@PAWPR08MB8982.eurprd08.prod.outlook.com>
The 01/12/2023 15:54, Wilco Dijkstra wrote:
> Optimize strnlen using the shrn instruction and improve the main loop.
> Small strings are around 10% faster, large strings are 40% faster on
> modern CPUs. Passes regress.
please commit it, thanks.
Reviewed-by: Szabolcs Nagy <szabolcs.nagy@arm.com>
>
> ---
>
> diff --git a/sysdeps/aarch64/strnlen.S b/sysdeps/aarch64/strnlen.S
> index 35fd14804d42ab90573f995b20cf65ba75042978..21112fbf760b7a99a6d153c00f9cd6b6bc144f3a 100644
> --- a/sysdeps/aarch64/strnlen.S
> +++ b/sysdeps/aarch64/strnlen.S
> @@ -44,19 +44,16 @@
>
> /*
> Core algorithm:
> -
> - For each 16-byte chunk we calculate a 64-bit nibble mask value with four bits
> - per byte. We take 4 bits of every comparison byte with shift right and narrow
> - by 4 instruction. Since the bits in the nibble mask reflect the order in
> - which things occur in the original string, counting trailing zeros identifies
> - exactly which byte matched. */
> + Process the string in 16-byte aligned chunks. Compute a 64-bit mask with
> + four bits per byte using the shrn instruction. A count trailing zeros then
> + identifies the first zero byte. */
>
> ENTRY (__strnlen)
> PTR_ARG (0)
> SIZE_ARG (1)
> bic src, srcin, 15
> cbz cntin, L(nomatch)
> - ld1 {vdata.16b}, [src], 16
> + ld1 {vdata.16b}, [src]
> cmeq vhas_chr.16b, vdata.16b, 0
> lsl shift, srcin, 2
> shrn vend.8b, vhas_chr.8h, 4 /* 128->64 */
> @@ -71,36 +68,40 @@ L(finish):
> csel result, cntin, result, ls
> ret
>
> +L(nomatch):
> + mov result, cntin
> + ret
> +
> L(start_loop):
> sub tmp, src, srcin
> + add tmp, tmp, 17
> subs cntrem, cntin, tmp
> - b.ls L(nomatch)
> + b.lo L(nomatch)
>
> /* Make sure that it won't overread by a 16-byte chunk */
> - add tmp, cntrem, 15
> - tbnz tmp, 4, L(loop32_2)
> -
> + tbz cntrem, 4, L(loop32_2)
> + sub src, src, 16
> .p2align 5
> L(loop32):
> - ldr qdata, [src], 16
> + ldr qdata, [src, 32]!
> cmeq vhas_chr.16b, vdata.16b, 0
> umaxp vend.16b, vhas_chr.16b, vhas_chr.16b /* 128->64 */
> fmov synd, dend
> cbnz synd, L(end)
> L(loop32_2):
> - ldr qdata, [src], 16
> + ldr qdata, [src, 16]
> subs cntrem, cntrem, 32
> cmeq vhas_chr.16b, vdata.16b, 0
> - b.ls L(end)
> + b.lo L(end_2)
> umaxp vend.16b, vhas_chr.16b, vhas_chr.16b /* 128->64 */
> fmov synd, dend
> cbz synd, L(loop32)
> -
> +L(end_2):
> + add src, src, 16
> L(end):
> shrn vend.8b, vhas_chr.8h, 4 /* 128->64 */
> - sub src, src, 16
> - mov synd, vend.d[0]
> sub result, src, srcin
> + fmov synd, dend
> #ifndef __AARCH64EB__
> rbit synd, synd
> #endif
> @@ -110,10 +111,6 @@ L(end):
> csel result, cntin, result, ls
> ret
>
> -L(nomatch):
> - mov result, cntin
> - ret
> -
> END (__strnlen)
> libc_hidden_def (__strnlen)
> weak_alias (__strnlen, strnlen)
>
prev parent reply other threads:[~2023-01-13 12:27 UTC|newest]
Thread overview: 2+ messages / expand[flat|nested] mbox.gz Atom feed top
2023-01-12 15:54 Wilco Dijkstra
2023-01-13 12:26 ` Szabolcs Nagy [this message]
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=Y8FODpWH2v9+lT7R@arm.com \
--to=szabolcs.nagy@arm.com \
--cc=Wilco.Dijkstra@arm.com \
--cc=libc-alpha@sourceware.org \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for read-only IMAP folder(s) and NNTP newsgroup(s).