[PATCH] XOR implementation for ARMv8
Ard Biesheuvel
ard.biesheuvel at linaro.org
Thu Jun 25 03:21:07 PDT 2015
On 25 June 2015 at 11:22, 刘晓东 <liuxiaodong at nudt.edu.cn> wrote:
>> On Wed, Jun 24, 2015 at 11:51 AM, Ard Biesheuvel <ard.biesheuvel at linaro.org> wrote:
>> > On 24 June 2015 at 10:29, Jérôme Forissier <jerome.forissier at linaro.org> wrote:
>> >>
>> >>
>> >> On 06/24/2015 09:00 AM, 刘晓东 wrote:
>> >>> Use the 128-bit SIMD registers and SIMD arithmetic instructions for XOR calculation in assembly language.
>> >>
>> >> Don't you need kernel_neon_begin()/kernel_neon_end() somewhere? (see
>> >> Documentation/arm/kernel_mode_neon.txt).
>> >>
>> >
>> > Jerome is right: use of this driver will corrupt the FP/SIMD state of
>> > arbitrary userland tasks if you don't explicitly claim the NEON for
>> > in-kernel use by calling kernel_neon_begin)_ and end()
>> >
>> > Since XOR may be called in interrupt context, this could add a fixed
>> > overhead to each call, even if you are calling the function many times
>> > in a row. This means you may be better off using even fewer registers,
>> > and use kernel_neon_begin_partial() instead.
>> >
>> > May I ask what kind of core you tested this on?
>>
>> And if Xiaodong Liu isn't subscribed to linux arm mail list then he will not get this email chain.
>> It was removed by Jerome from to/cc list. Please don't do that.
>>
>> (restoring back Xiaodong Liu email)
>>
>> --
>> Best regards, Klimov Alexey
>
> According to your suggestion, I have revised the code by using kernel_neon_begin and kernel_neon_end to avoid contaminating the SIMD registers in interrupt context.
> BTW, I use Phytium FT-1500A SoC which is armv8 compatible for tests.
>
Very nice! I tested it on my Cortex-A57 based system:
8regs : 4810.400 MB/sec
32regs : 4365.200 MB/sec
arm64regs8: 5146.400 MB/sec
neon_arm64: 17314.800 MB/sec
so that is quite a lovely speedup. However, after spotting and
removing a bug in your asm code, I get this instead
xor: measuring software checksum speed
8regs : 4651.600 MB/sec
32regs : 4234.800 MB/sec
arm64regs8: 5099.600 MB/sec
neon_arm64: 5321.200 MB/sec
Unless you are getting much better results on your system after fixing
your code, I am not convinced this is worth the trouble, to be honest.
Perhaps if you can show improved performance on a real-world
benchmark?
I have a couple of review comments below. But could you also make sure
that you submit the patch according to the rules?
So please:
- prepend 'arm64' to the subject
- don't use lines longer than 75 columns in the commit message
- please trim all whitespace from the end of lines.
scripts/checkpatch.pl will help you verify if you have done that
correctly.
> ---
> include/asm/xor.h | 197 ++++++++++++++++++++++++++++++++++++++++++++
> kernel/arm64ksyms.c | 13 ++
> lib/Makefile | 4
> lib/xor-neon.c | 30 ++++++
> lib/xor.S | 228 ++++++++++++++++++++++++++++++++++++++++++++++++++++
> 5 files changed, 472 insertions(+)
> ---
> diff -pruN -X dontdiff linux-4.0.5-orig/arch/arm64/include/asm/xor.h linux-4.0.5-mod/arch/arm64/include/asm/xor.h
> --- linux-4.0.5-orig/arch/arm64/include/asm/xor.h 1970-01-01 08:00:00.000000000 +0800
> +++ linux-4.0.5-mod/arch/arm64/include/asm/xor.h 2015-06-25 16:59:19.527197817 +0800
> @@ -0,0 +1,197 @@
> +/*
> + * arch/arm64/include/asm/xor.h
> + *
> + * Copyright (C) Xiaodong Liu <liuxiaodong at nudt.edu.cn>, Changsha, P.R. China
> + *
> + * This program is free software; you can redistribute it and/or modify
> + * it under the terms of the GNU General Public License version 2 as
> + * published by the Free Software Foundation.
> + */
> +#include <linux/hardirq.h>
> +#include <asm-generic/xor.h>
> +#include <asm/hwcap.h>
> +#include <asm/neon.h>
> +
> +#define __XOR(a1, a2) a1 ^= a2
> +
> +#define GET_BLOCK_8(dst) \
> + __asm__("ldp %1, %2, [%0], #16;\n\t" \
> + "ldp %3, %4, [%0], #16;\n\t" \
> + "ldp %5, %6, [%0], #16;\n\t" \
> + "ldp %7, %8, [%0], #16;\n\t" \
> + : "=r" (dst), "=r" (a1), "=r" (a2), "=r" (a3), "=r" (a4), "=r" (a5), "=r" (a6), "=r" (a7), "=r" (a8) \
> + : "0" (dst))
> +
> +#define XOR_BLOCK_8(src) \
> + __asm__("ldp %1, %2, [%0], #16;\n\t" \
> + "ldp %3, %4, [%0], #16;\n\t" \
> + "ldp %5, %6, [%0], #16;\n\t" \
> + "ldp %7, %8, [%0], #16;\n\t" \
> + : "=r" (src), "=r" (b1), "=r" (b2), "=r" (b3), "=r" (b4), "=r" (b5), "=r" (b6), "=r" (b7), "=r" (b8) \
> + : "0" (src)); \
> + __XOR(a1, b1); __XOR(a2, b2); __XOR(a3, b3); __XOR(a4, b4); __XOR(a5, b5); __XOR(a6, b6); __XOR(a7, b7); __XOR(a8, b8)
> +
> +#define PUT_BLOCK_8(dst) \
> + __asm__ __volatile__("stp %1, %2, [%0], #16;\n\t" \
> + "stp %3, %4, [%0], #16;\n\t" \
> + "stp %5, %6, [%0], #16;\n\t" \
> + "stp %7, %8, [%0], #16;\n\t" \
> + : "=r" (dst) \
> + : "0" (dst), "r" (a1), "r" (a2), "r" (a3), "r" (a4), "r" (a5), "r" (a6), "r" (a7), "r" (a8));
> +
> +#define INIT_REGISTER() \
> + register unsigned long a1 __asm__("x4"); \
> + register unsigned long a2 __asm__("x5"); \
> + register unsigned long a3 __asm__("x6"); \
> + register unsigned long a4 __asm__("x7"); \
> + register unsigned long a5 __asm__("x8"); \
> + register unsigned long a6 __asm__("x9"); \
> + register unsigned long a7 __asm__("x10"); \
> + register unsigned long a8 __asm__("x11"); \
> + register unsigned long b1 __asm__("x12"); \
> + register unsigned long b2 __asm__("x13"); \
> + register unsigned long b3 __asm__("x14"); \
> + register unsigned long b4 __asm__("x15"); \
> + register unsigned long b5 __asm__("x16"); \
> + register unsigned long b6 __asm__("x17"); \
> + register unsigned long b7 __asm__("x18"); \
> + register unsigned long b8 __asm__("x19");
> +
> +static void
> +xor_arm8regs_2(unsigned long bytes, unsigned long *p1, unsigned long *p2)
> +{
> + unsigned long lines = bytes / sizeof(unsigned long) / 8;
> + INIT_REGISTER();
> +
> + do {
> + GET_BLOCK_8(p1);
> + XOR_BLOCK_8(p2);
> + PUT_BLOCK_8(p1);
> + } while(--lines);
> +}
> +
> +static void
> +xor_arm8regs_3(unsigned long bytes, unsigned long *p1, unsigned long *p2,
> + unsigned long *p3)
> +{
> + unsigned long lines = bytes / sizeof(unsigned long) / 8;
> + INIT_REGISTER();
> +
> + do {
> + GET_BLOCK_8(p1);
> + XOR_BLOCK_8(p2);
> + XOR_BLOCK_8(p3);
> + PUT_BLOCK_8(p1);
> + } while(--lines);
> +}
> +
> +static void
> +xor_arm8regs_4(unsigned long bytes, unsigned long *p1, unsigned long *p2,
> + unsigned long *p3, unsigned long *p4)
> +{
> + unsigned long lines = bytes / sizeof(unsigned long) / 8;
> + INIT_REGISTER();
> +
> + do {
> + GET_BLOCK_8(p1);
> + XOR_BLOCK_8(p2);
> + XOR_BLOCK_8(p3);
> + XOR_BLOCK_8(p4);
> + PUT_BLOCK_8(p1);
> + } while(--lines);
> +}
> +
> +static void
> +xor_arm8regs_5(unsigned long bytes, unsigned long *p1, unsigned long *p2,
> + unsigned long *p3, unsigned long *p4, unsigned long *p5)
> +{
> + unsigned long lines = bytes / sizeof(unsigned long) / 8;
> + INIT_REGISTER();
> +
> + do {
> + GET_BLOCK_8(p1);
> + XOR_BLOCK_8(p2);
> + XOR_BLOCK_8(p3);
> + XOR_BLOCK_8(p4);
> + XOR_BLOCK_8(p5);
> + PUT_BLOCK_8(p1);
> + } while(--lines);
> +}
> +
I think you can drop this implementation entirely (i.e., the non-NEON
one). NEON is mandatory on arm64, and usable in interrupt context (see
below), and this code is only slightly faster than the generic code
(according to my benchmark)
> +extern struct xor_block_template const xor_block_neon_arm64;
> +
> +static void
> +xor_neon_2(unsigned long bytes, unsigned long *p1, unsigned long *p2)
> +{
> + if(in_interrupt()) {
> + xor_arm8regs_2(bytes, p1, p2);
No need to fallback to integer code in interrupt context.
> + } else {
> + kernel_neon_begin();
... since kernel_neon_begin() is allowed in interrupt context on arm64.
However, since kernel_neon_begin() may be more costly in interrupt
context (since it stacks/unstacks the contents every time), you should
use the following instead
kernel_neon_begin_partial(16);
and reworks your assembly code so it uses q0 .. q15 instead of q16 .. q31.
> + xor_block_neon_arm64.do_2(bytes, p1, p2);
> + kernel_neon_end();
> + }
> +}
> +
> +static void
> +xor_neon_3(unsigned long bytes, unsigned long *p1, unsigned long *p2,
> + unsigned long *p3)
> +{
> + if(in_interrupt()) {
> + xor_arm8regs_3(bytes, p1, p2, p3);
likewise
> + } else {
> + kernel_neon_begin();
> + xor_block_neon_arm64.do_3(bytes, p1, p2, p3);
> + kernel_neon_end();
> + }
> +}
> +
> +static void
> +xor_neon_4(unsigned long bytes, unsigned long *p1, unsigned long *p2,
> + unsigned long *p3, unsigned long *p4)
> +{
> + if(in_interrupt()) {
> + xor_arm8regs_4(bytes, p1, p2, p3, p4);
likewise
> + } else {
> + kernel_neon_begin();
> + xor_block_neon_arm64.do_4(bytes, p1, p2, p3, p4);
> + kernel_neon_end();
> + }
> +}
> +
> +static void
> +xor_neon_5(unsigned long bytes, unsigned long *p1, unsigned long *p2,
> + unsigned long *p3, unsigned long *p4, unsigned long *p5)
> +{
> + if(in_interrupt()) {
> + xor_arm8regs_5(bytes, p1, p2, p3, p4, p5);
likewise
> + } else {
> + kernel_neon_begin();
> + xor_block_neon_arm64.do_5(bytes, p1, p2, p3, p4, p5);
> + kernel_neon_end();
> + }
> +}
> +
> +static struct xor_block_template xor_block_arm64regs8 = {
> + .name = "arm64regs8",
> + .do_2 = xor_arm8regs_2,
> + .do_3 = xor_arm8regs_3,
> + .do_4 = xor_arm8regs_4,
> + .do_5 = xor_arm8regs_5
> +};
> +
Remove
> +static struct xor_block_template xor_block_arm64 = {
> + .name = "neon_arm64",
> + .do_2 = xor_neon_2,
> + .do_3 = xor_neon_3,
> + .do_4 = xor_neon_4,
> + .do_5 = xor_neon_5
> +};
> +
> +#undef XOR_TRY_TEMPLATES
> +#define XOR_TRY_TEMPLATES \
> + do { \
> + xor_speed(&xor_block_8regs); \
> + xor_speed(&xor_block_32regs); \
> + xor_speed(&xor_block_arm64regs8); \
Remove
> + do { if (cpu_has_neon()) xor_speed(&xor_block_arm64); } while (0); \
> + } while(0)
> diff -pruN -X dontdiff linux-4.0.5-orig/arch/arm64/kernel/arm64ksyms.c linux-4.0.5-mod/arch/arm64/kernel/arm64ksyms.c
> --- linux-4.0.5-orig/arch/arm64/kernel/arm64ksyms.c 2015-06-06 23:21:22.000000000 +0800
> +++ linux-4.0.5-mod/arch/arm64/kernel/arm64ksyms.c 2015-06-25 11:40:07.537692040 +0800
> @@ -65,3 +65,16 @@ EXPORT_SYMBOL(test_and_change_bit);
> #ifdef CONFIG_FUNCTION_TRACER
> EXPORT_SYMBOL(_mcount);
> #endif
> +
> + /* xor ops */
> +extern void xor_arm64ldpregs16_2(unsigned long, unsigned long *, unsigned long *);
> +extern void xor_arm64ldpregs16_3(unsigned long, unsigned long *, unsigned long *,
> + unsigned long *);
> +extern void xor_arm64ldpregs16_4(unsigned long, unsigned long *, unsigned long *,
> + unsigned long *, unsigned long *);
> +extern void xor_arm64ldpregs16_5(unsigned long, unsigned long *, unsigned long *,
> + unsigned long *, unsigned long *, unsigned long *);
> +EXPORT_SYMBOL(xor_arm64ldpregs16_2);
> +EXPORT_SYMBOL(xor_arm64ldpregs16_3);
> +EXPORT_SYMBOL(xor_arm64ldpregs16_4);
> +EXPORT_SYMBOL(xor_arm64ldpregs16_5);
Please remove these. This breaks the build if you build xor-neon as a
module, and is not needed anyway.
> diff -pruN -X dontdiff linux-4.0.5-orig/arch/arm64/lib/Makefile linux-4.0.5-mod/arch/arm64/lib/Makefile
> --- linux-4.0.5-orig/arch/arm64/lib/Makefile 2015-06-06 23:21:22.000000000 +0800
> +++ linux-4.0.5-mod/arch/arm64/lib/Makefile 2015-06-25 16:47:44.051223943 +0800
> @@ -3,3 +3,7 @@ lib-y := bitops.o clear_user.o delay.o
> clear_page.o memchr.o memcpy.o memmove.o memset.o \
> memcmp.o strcmp.o strncmp.o strlen.o strnlen.o \
> strchr.o strrchr.o
> +
> +ifeq ($(CONFIG_KERNEL_MODE_NEON),y)
> + obj-$(CONFIG_XOR_BLOCKS) += xor-neon.o xor.o
> +endif
> diff -pruN -X dontdiff linux-4.0.5-orig/arch/arm64/lib/xor-neon.c linux-4.0.5-mod/arch/arm64/lib/xor-neon.c
> --- linux-4.0.5-orig/arch/arm64/lib/xor-neon.c 1970-01-01 08:00:00.000000000 +0800
> +++ linux-4.0.5-mod/arch/arm64/lib/xor-neon.c 2015-06-25 16:53:36.319210709 +0800
> @@ -0,0 +1,30 @@
> +/*
> + * arch/arm64/lib/xor-neon.c
> + *
> + * Copyright (C) Xiaodong Liu <liuxiaodong at nudt.edu.cn>, Changsha, P.R. China
> + *
> + * This program is free software; you can redistribute it and/or modify
> + * it under the terms of the GNU General Public License version 2 as
> + * published by the Free Software Foundation.
> + */
> +
> +#include <linux/raid/xor.h>
> +#include <linux/module.h>
> +
> +extern void xor_arm64ldpregs16_2(unsigned long, unsigned long *, unsigned long *);
> +extern void xor_arm64ldpregs16_3(unsigned long, unsigned long *, unsigned long *,
> + unsigned long *);
> +extern void xor_arm64ldpregs16_4(unsigned long, unsigned long *, unsigned long *,
> + unsigned long *, unsigned long *);
> +extern void xor_arm64ldpregs16_5(unsigned long, unsigned long *, unsigned long *,
> + unsigned long *, unsigned long *, unsigned long *);
> +
Could you please use better names here? arm64ldp does not suggest it
uses NEON instructions.
> +struct xor_block_template const xor_block_neon_arm64 = {
> + .name = "ARM64LDPregs16",
Same here, although the name does not really matter
> + .do_2 = xor_arm64ldpregs16_2,
> + .do_3 = xor_arm64ldpregs16_3,
> + .do_4 = xor_arm64ldpregs16_4,
> + .do_5 = xor_arm64ldpregs16_5,
> +};
> +
> +EXPORT_SYMBOL(xor_block_neon_arm64);
> diff -pruN -X dontdiff linux-4.0.5-orig/arch/arm64/lib/xor.S linux-4.0.5-mod/arch/arm64/lib/xor.S
> --- linux-4.0.5-orig/arch/arm64/lib/xor.S 1970-01-01 08:00:00.000000000 +0800
> +++ linux-4.0.5-mod/arch/arm64/lib/xor.S 2015-06-24 09:25:49.969256540 +0800
> @@ -0,0 +1,228 @@
> +/*
> + * arch/arm64/lib/xor.S
> + *
> + * Copyright (C) Xiaodong Liu <liuxiaodong at nudt.edu.cn>, Changsha, P.R. China
> + *
> + * This program is free software; you can redistribute it and/or modify
> + * it under the terms of the GNU General Public License version 2 as
> + * published by the Free Software Foundation.
> + */
> +
> +#include <linux/linkage.h>
> +#include <asm/assembler.h>
newline here please
> +.macro xor_vectorregs16
> + eor v24.16b, v24.16b, v16.16b
> + eor v25.16b, v25.16b, v17.16b
> + eor v26.16b, v26.16b, v18.16b
> + eor v27.16b, v27.16b, v19.16b
> + eor v28.16b, v28.16b, v20.16b
> + eor v29.16b, v29.16b, v21.16b
> + eor v30.16b, v30.16b, v22.16b
> + eor v31.16b, v31.16b, v23.16b
> +.endm
> +
as mentioned above, please use q0 .. q15 so you only need to preserve
16 registers instead of all 32
> +.align 4
.align and .p2align mean the same thing on arm, so please choose one
and stick with it
> +
> +/*
> + * void xor_arm64ldpregs16_2(unsigned long size, unsigned long * dst, unsigned long *src);
> + *
> + * Parameters:
> + * x0 - size
> + * x1 - dst
> + * x2 - src
> + */
> +ENTRY(xor_arm64ldpregs16_2)
> +
> + lsr x0, x0, #10
You are dividing the size by 1024, but only xor'ing 128 bytes per
iteration. This means your code is utterly broken, and the benchmarks
are meaningless.
> +
> +.p2align 4
If you are aligning on a cache line here, it makes more sense to use 5
(64 bytes) instead. But you could apply it to the entire function
instead of here, I don't think it will make a lot of difference.
> +Loop23:
> + ldp q16, q17, [x2], #32
> + ldp q18, q19, [x2], #32
> + ldp q20, q21, [x2], #32
> + ldp q22, q23, [x2], #32
> +
> + mov x3,x1
> +
> + ldp q24, q25, [x1], #32
> + ldp q26, q27, [x1], #32
> + ldp q27, q29, [x1], #32
> + ldp q30, q31, [x1], #32
> +
> + xor_vectorregs16
> +
> + stp q24, q25, [x3], #32
> + stp q26, q27, [x3], #32
> + stp q27, q29, [x3], #32
> + stp q30, q31, [x3], #32
> +
> + subs x0, x0, #1
> + cbnz x0, Loop23
> +
> + ret
> +ENDPROC(xor_arm64ldpregs16_2)
> +
Same applies to the other versions.
Regards,
Ard.
[...]
More information about the linux-arm-kernel
mailing list