mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-05 05:52:05 +00:00
420 lines
11 KiB
ArmAsm
420 lines
11 KiB
ArmAsm
/* memmove/memcpy/mempcpy optimized with AVX512 for KNL hardware.
|
|
Copyright (C) 2016-2020 Free Software Foundation, Inc.
|
|
This file is part of the GNU C Library.
|
|
|
|
The GNU C Library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 of the License, or (at your option) any later version.
|
|
|
|
The GNU C Library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with the GNU C Library; if not, see
|
|
<https://www.gnu.org/licenses/>. */
|
|
|
|
#include "sysdep.h"
|
|
|
|
#if 1
|
|
|
|
# include "asm-syntax.h"
|
|
|
|
.section .text.avx512,"ax",@progbits
|
|
ENTRY (__mempcpy_chk_avx512_no_vzeroupper)
|
|
cmp %RDX_LP, %RCX_LP
|
|
jb HIDDEN_JUMPTARGET (__chk_fail)
|
|
END (__mempcpy_chk_avx512_no_vzeroupper)
|
|
|
|
ENTRY (__mempcpy_avx512_no_vzeroupper)
|
|
mov %RDI_LP, %RAX_LP
|
|
add %RDX_LP, %RAX_LP
|
|
jmp L(start)
|
|
END (__mempcpy_avx512_no_vzeroupper)
|
|
|
|
ENTRY (__memmove_chk_avx512_no_vzeroupper)
|
|
cmp %RDX_LP, %RCX_LP
|
|
jb HIDDEN_JUMPTARGET (__chk_fail)
|
|
END (__memmove_chk_avx512_no_vzeroupper)
|
|
|
|
ENTRY (__memmove_avx512_no_vzeroupper)
|
|
mov %RDI_LP, %RAX_LP
|
|
# ifdef USE_AS_MEMPCPY
|
|
add %RDX_LP, %RAX_LP
|
|
# endif
|
|
L(start):
|
|
# ifdef __ILP32__
|
|
/* Clear the upper 32 bits. */
|
|
mov %edx, %edx
|
|
# endif
|
|
lea (%rsi, %rdx), %rcx
|
|
lea (%rdi, %rdx), %r9
|
|
cmp $512, %rdx
|
|
ja L(512bytesormore)
|
|
|
|
L(check):
|
|
cmp $16, %rdx
|
|
jbe L(less_16bytes)
|
|
cmp $256, %rdx
|
|
jb L(less_256bytes)
|
|
vmovups (%rsi), %zmm0
|
|
vmovups 0x40(%rsi), %zmm1
|
|
vmovups 0x80(%rsi), %zmm2
|
|
vmovups 0xC0(%rsi), %zmm3
|
|
vmovups -0x100(%rcx), %zmm4
|
|
vmovups -0xC0(%rcx), %zmm5
|
|
vmovups -0x80(%rcx), %zmm6
|
|
vmovups -0x40(%rcx), %zmm7
|
|
vmovups %zmm0, (%rdi)
|
|
vmovups %zmm1, 0x40(%rdi)
|
|
vmovups %zmm2, 0x80(%rdi)
|
|
vmovups %zmm3, 0xC0(%rdi)
|
|
vmovups %zmm4, -0x100(%r9)
|
|
vmovups %zmm5, -0xC0(%r9)
|
|
vmovups %zmm6, -0x80(%r9)
|
|
vmovups %zmm7, -0x40(%r9)
|
|
ret
|
|
|
|
L(less_256bytes):
|
|
cmp $128, %dl
|
|
jb L(less_128bytes)
|
|
vmovups (%rsi), %zmm0
|
|
vmovups 0x40(%rsi), %zmm1
|
|
vmovups -0x80(%rcx), %zmm2
|
|
vmovups -0x40(%rcx), %zmm3
|
|
vmovups %zmm0, (%rdi)
|
|
vmovups %zmm1, 0x40(%rdi)
|
|
vmovups %zmm2, -0x80(%r9)
|
|
vmovups %zmm3, -0x40(%r9)
|
|
ret
|
|
|
|
L(less_128bytes):
|
|
cmp $64, %dl
|
|
jb L(less_64bytes)
|
|
vmovdqu (%rsi), %ymm0
|
|
vmovdqu 0x20(%rsi), %ymm1
|
|
vmovdqu -0x40(%rcx), %ymm2
|
|
vmovdqu -0x20(%rcx), %ymm3
|
|
vmovdqu %ymm0, (%rdi)
|
|
vmovdqu %ymm1, 0x20(%rdi)
|
|
vmovdqu %ymm2, -0x40(%r9)
|
|
vmovdqu %ymm3, -0x20(%r9)
|
|
ret
|
|
|
|
L(less_64bytes):
|
|
cmp $32, %dl
|
|
jb L(less_32bytes)
|
|
vmovdqu (%rsi), %ymm0
|
|
vmovdqu -0x20(%rcx), %ymm1
|
|
vmovdqu %ymm0, (%rdi)
|
|
vmovdqu %ymm1, -0x20(%r9)
|
|
ret
|
|
|
|
L(less_32bytes):
|
|
vmovdqu (%rsi), %xmm0
|
|
vmovdqu -0x10(%rcx), %xmm1
|
|
vmovdqu %xmm0, (%rdi)
|
|
vmovdqu %xmm1, -0x10(%r9)
|
|
ret
|
|
|
|
L(less_16bytes):
|
|
cmp $8, %dl
|
|
jb L(less_8bytes)
|
|
movq (%rsi), %rsi
|
|
movq -0x8(%rcx), %rcx
|
|
movq %rsi, (%rdi)
|
|
movq %rcx, -0x8(%r9)
|
|
ret
|
|
|
|
L(less_8bytes):
|
|
cmp $4, %dl
|
|
jb L(less_4bytes)
|
|
mov (%rsi), %esi
|
|
mov -0x4(%rcx), %ecx
|
|
mov %esi, (%rdi)
|
|
mov %ecx, -0x4(%r9)
|
|
ret
|
|
|
|
L(less_4bytes):
|
|
cmp $2, %dl
|
|
jb L(less_2bytes)
|
|
mov (%rsi), %si
|
|
mov -0x2(%rcx), %cx
|
|
mov %si, (%rdi)
|
|
mov %cx, -0x2(%r9)
|
|
ret
|
|
|
|
L(less_2bytes):
|
|
cmp $1, %dl
|
|
jb L(less_1bytes)
|
|
mov (%rsi), %cl
|
|
mov %cl, (%rdi)
|
|
L(less_1bytes):
|
|
ret
|
|
|
|
L(512bytesormore):
|
|
# ifdef SHARED_CACHE_SIZE_HALF
|
|
mov $SHARED_CACHE_SIZE_HALF, %r8
|
|
# else
|
|
mov __x86_shared_cache_size_half(%rip), %r8
|
|
# endif
|
|
cmp %r8, %rdx
|
|
jae L(preloop_large)
|
|
cmp $1024, %rdx
|
|
ja L(1024bytesormore)
|
|
prefetcht1 (%rsi)
|
|
prefetcht1 0x40(%rsi)
|
|
prefetcht1 0x80(%rsi)
|
|
prefetcht1 0xC0(%rsi)
|
|
prefetcht1 0x100(%rsi)
|
|
prefetcht1 0x140(%rsi)
|
|
prefetcht1 0x180(%rsi)
|
|
prefetcht1 0x1C0(%rsi)
|
|
prefetcht1 -0x200(%rcx)
|
|
prefetcht1 -0x1C0(%rcx)
|
|
prefetcht1 -0x180(%rcx)
|
|
prefetcht1 -0x140(%rcx)
|
|
prefetcht1 -0x100(%rcx)
|
|
prefetcht1 -0xC0(%rcx)
|
|
prefetcht1 -0x80(%rcx)
|
|
prefetcht1 -0x40(%rcx)
|
|
vmovups (%rsi), %zmm0
|
|
vmovups 0x40(%rsi), %zmm1
|
|
vmovups 0x80(%rsi), %zmm2
|
|
vmovups 0xC0(%rsi), %zmm3
|
|
vmovups 0x100(%rsi), %zmm4
|
|
vmovups 0x140(%rsi), %zmm5
|
|
vmovups 0x180(%rsi), %zmm6
|
|
vmovups 0x1C0(%rsi), %zmm7
|
|
vmovups -0x200(%rcx), %zmm8
|
|
vmovups -0x1C0(%rcx), %zmm9
|
|
vmovups -0x180(%rcx), %zmm10
|
|
vmovups -0x140(%rcx), %zmm11
|
|
vmovups -0x100(%rcx), %zmm12
|
|
vmovups -0xC0(%rcx), %zmm13
|
|
vmovups -0x80(%rcx), %zmm14
|
|
vmovups -0x40(%rcx), %zmm15
|
|
vmovups %zmm0, (%rdi)
|
|
vmovups %zmm1, 0x40(%rdi)
|
|
vmovups %zmm2, 0x80(%rdi)
|
|
vmovups %zmm3, 0xC0(%rdi)
|
|
vmovups %zmm4, 0x100(%rdi)
|
|
vmovups %zmm5, 0x140(%rdi)
|
|
vmovups %zmm6, 0x180(%rdi)
|
|
vmovups %zmm7, 0x1C0(%rdi)
|
|
vmovups %zmm8, -0x200(%r9)
|
|
vmovups %zmm9, -0x1C0(%r9)
|
|
vmovups %zmm10, -0x180(%r9)
|
|
vmovups %zmm11, -0x140(%r9)
|
|
vmovups %zmm12, -0x100(%r9)
|
|
vmovups %zmm13, -0xC0(%r9)
|
|
vmovups %zmm14, -0x80(%r9)
|
|
vmovups %zmm15, -0x40(%r9)
|
|
ret
|
|
|
|
L(1024bytesormore):
|
|
cmp %rsi, %rdi
|
|
ja L(1024bytesormore_bkw)
|
|
sub $512, %r9
|
|
vmovups -0x200(%rcx), %zmm8
|
|
vmovups -0x1C0(%rcx), %zmm9
|
|
vmovups -0x180(%rcx), %zmm10
|
|
vmovups -0x140(%rcx), %zmm11
|
|
vmovups -0x100(%rcx), %zmm12
|
|
vmovups -0xC0(%rcx), %zmm13
|
|
vmovups -0x80(%rcx), %zmm14
|
|
vmovups -0x40(%rcx), %zmm15
|
|
prefetcht1 (%rsi)
|
|
prefetcht1 0x40(%rsi)
|
|
prefetcht1 0x80(%rsi)
|
|
prefetcht1 0xC0(%rsi)
|
|
prefetcht1 0x100(%rsi)
|
|
prefetcht1 0x140(%rsi)
|
|
prefetcht1 0x180(%rsi)
|
|
prefetcht1 0x1C0(%rsi)
|
|
|
|
/* Loop with unaligned memory access. */
|
|
L(gobble_512bytes_loop):
|
|
vmovups (%rsi), %zmm0
|
|
vmovups 0x40(%rsi), %zmm1
|
|
vmovups 0x80(%rsi), %zmm2
|
|
vmovups 0xC0(%rsi), %zmm3
|
|
vmovups 0x100(%rsi), %zmm4
|
|
vmovups 0x140(%rsi), %zmm5
|
|
vmovups 0x180(%rsi), %zmm6
|
|
vmovups 0x1C0(%rsi), %zmm7
|
|
add $512, %rsi
|
|
prefetcht1 (%rsi)
|
|
prefetcht1 0x40(%rsi)
|
|
prefetcht1 0x80(%rsi)
|
|
prefetcht1 0xC0(%rsi)
|
|
prefetcht1 0x100(%rsi)
|
|
prefetcht1 0x140(%rsi)
|
|
prefetcht1 0x180(%rsi)
|
|
prefetcht1 0x1C0(%rsi)
|
|
vmovups %zmm0, (%rdi)
|
|
vmovups %zmm1, 0x40(%rdi)
|
|
vmovups %zmm2, 0x80(%rdi)
|
|
vmovups %zmm3, 0xC0(%rdi)
|
|
vmovups %zmm4, 0x100(%rdi)
|
|
vmovups %zmm5, 0x140(%rdi)
|
|
vmovups %zmm6, 0x180(%rdi)
|
|
vmovups %zmm7, 0x1C0(%rdi)
|
|
add $512, %rdi
|
|
cmp %r9, %rdi
|
|
jb L(gobble_512bytes_loop)
|
|
vmovups %zmm8, (%r9)
|
|
vmovups %zmm9, 0x40(%r9)
|
|
vmovups %zmm10, 0x80(%r9)
|
|
vmovups %zmm11, 0xC0(%r9)
|
|
vmovups %zmm12, 0x100(%r9)
|
|
vmovups %zmm13, 0x140(%r9)
|
|
vmovups %zmm14, 0x180(%r9)
|
|
vmovups %zmm15, 0x1C0(%r9)
|
|
ret
|
|
|
|
L(1024bytesormore_bkw):
|
|
add $512, %rdi
|
|
vmovups 0x1C0(%rsi), %zmm8
|
|
vmovups 0x180(%rsi), %zmm9
|
|
vmovups 0x140(%rsi), %zmm10
|
|
vmovups 0x100(%rsi), %zmm11
|
|
vmovups 0xC0(%rsi), %zmm12
|
|
vmovups 0x80(%rsi), %zmm13
|
|
vmovups 0x40(%rsi), %zmm14
|
|
vmovups (%rsi), %zmm15
|
|
prefetcht1 -0x40(%rcx)
|
|
prefetcht1 -0x80(%rcx)
|
|
prefetcht1 -0xC0(%rcx)
|
|
prefetcht1 -0x100(%rcx)
|
|
prefetcht1 -0x140(%rcx)
|
|
prefetcht1 -0x180(%rcx)
|
|
prefetcht1 -0x1C0(%rcx)
|
|
prefetcht1 -0x200(%rcx)
|
|
|
|
/* Backward loop with unaligned memory access. */
|
|
L(gobble_512bytes_loop_bkw):
|
|
vmovups -0x40(%rcx), %zmm0
|
|
vmovups -0x80(%rcx), %zmm1
|
|
vmovups -0xC0(%rcx), %zmm2
|
|
vmovups -0x100(%rcx), %zmm3
|
|
vmovups -0x140(%rcx), %zmm4
|
|
vmovups -0x180(%rcx), %zmm5
|
|
vmovups -0x1C0(%rcx), %zmm6
|
|
vmovups -0x200(%rcx), %zmm7
|
|
sub $512, %rcx
|
|
prefetcht1 -0x40(%rcx)
|
|
prefetcht1 -0x80(%rcx)
|
|
prefetcht1 -0xC0(%rcx)
|
|
prefetcht1 -0x100(%rcx)
|
|
prefetcht1 -0x140(%rcx)
|
|
prefetcht1 -0x180(%rcx)
|
|
prefetcht1 -0x1C0(%rcx)
|
|
prefetcht1 -0x200(%rcx)
|
|
vmovups %zmm0, -0x40(%r9)
|
|
vmovups %zmm1, -0x80(%r9)
|
|
vmovups %zmm2, -0xC0(%r9)
|
|
vmovups %zmm3, -0x100(%r9)
|
|
vmovups %zmm4, -0x140(%r9)
|
|
vmovups %zmm5, -0x180(%r9)
|
|
vmovups %zmm6, -0x1C0(%r9)
|
|
vmovups %zmm7, -0x200(%r9)
|
|
sub $512, %r9
|
|
cmp %rdi, %r9
|
|
ja L(gobble_512bytes_loop_bkw)
|
|
vmovups %zmm8, -0x40(%rdi)
|
|
vmovups %zmm9, -0x80(%rdi)
|
|
vmovups %zmm10, -0xC0(%rdi)
|
|
vmovups %zmm11, -0x100(%rdi)
|
|
vmovups %zmm12, -0x140(%rdi)
|
|
vmovups %zmm13, -0x180(%rdi)
|
|
vmovups %zmm14, -0x1C0(%rdi)
|
|
vmovups %zmm15, -0x200(%rdi)
|
|
ret
|
|
|
|
L(preloop_large):
|
|
cmp %rsi, %rdi
|
|
ja L(preloop_large_bkw)
|
|
vmovups (%rsi), %zmm4
|
|
vmovups 0x40(%rsi), %zmm5
|
|
|
|
mov %rdi, %r11
|
|
/* Align destination for access with non-temporal stores in the loop. */
|
|
mov %rdi, %r8
|
|
and $-0x80, %rdi
|
|
add $0x80, %rdi
|
|
sub %rdi, %r8
|
|
sub %r8, %rsi
|
|
add %r8, %rdx
|
|
L(gobble_256bytes_nt_loop):
|
|
prefetcht1 0x200(%rsi)
|
|
prefetcht1 0x240(%rsi)
|
|
prefetcht1 0x280(%rsi)
|
|
prefetcht1 0x2C0(%rsi)
|
|
prefetcht1 0x300(%rsi)
|
|
prefetcht1 0x340(%rsi)
|
|
prefetcht1 0x380(%rsi)
|
|
prefetcht1 0x3C0(%rsi)
|
|
vmovdqu64 (%rsi), %zmm0
|
|
vmovdqu64 0x40(%rsi), %zmm1
|
|
vmovdqu64 0x80(%rsi), %zmm2
|
|
vmovdqu64 0xC0(%rsi), %zmm3
|
|
vmovntdq %zmm0, (%rdi)
|
|
vmovntdq %zmm1, 0x40(%rdi)
|
|
vmovntdq %zmm2, 0x80(%rdi)
|
|
vmovntdq %zmm3, 0xC0(%rdi)
|
|
sub $256, %rdx
|
|
add $256, %rsi
|
|
add $256, %rdi
|
|
cmp $256, %rdx
|
|
ja L(gobble_256bytes_nt_loop)
|
|
sfence
|
|
vmovups %zmm4, (%r11)
|
|
vmovups %zmm5, 0x40(%r11)
|
|
jmp L(check)
|
|
|
|
L(preloop_large_bkw):
|
|
vmovups -0x80(%rcx), %zmm4
|
|
vmovups -0x40(%rcx), %zmm5
|
|
|
|
/* Align end of destination for access with non-temporal stores. */
|
|
mov %r9, %r8
|
|
and $-0x80, %r9
|
|
sub %r9, %r8
|
|
sub %r8, %rcx
|
|
sub %r8, %rdx
|
|
add %r9, %r8
|
|
L(gobble_256bytes_nt_loop_bkw):
|
|
prefetcht1 -0x400(%rcx)
|
|
prefetcht1 -0x3C0(%rcx)
|
|
prefetcht1 -0x380(%rcx)
|
|
prefetcht1 -0x340(%rcx)
|
|
prefetcht1 -0x300(%rcx)
|
|
prefetcht1 -0x2C0(%rcx)
|
|
prefetcht1 -0x280(%rcx)
|
|
prefetcht1 -0x240(%rcx)
|
|
vmovdqu64 -0x100(%rcx), %zmm0
|
|
vmovdqu64 -0xC0(%rcx), %zmm1
|
|
vmovdqu64 -0x80(%rcx), %zmm2
|
|
vmovdqu64 -0x40(%rcx), %zmm3
|
|
vmovntdq %zmm0, -0x100(%r9)
|
|
vmovntdq %zmm1, -0xC0(%r9)
|
|
vmovntdq %zmm2, -0x80(%r9)
|
|
vmovntdq %zmm3, -0x40(%r9)
|
|
sub $256, %rdx
|
|
sub $256, %rcx
|
|
sub $256, %r9
|
|
cmp $256, %rdx
|
|
ja L(gobble_256bytes_nt_loop_bkw)
|
|
sfence
|
|
vmovups %zmm4, -0x80(%r8)
|
|
vmovups %zmm5, -0x40(%r8)
|
|
jmp L(check)
|
|
END (__memmove_avx512_no_vzeroupper)
|
|
|
|
strong_alias (__memmove_avx512_no_vzeroupper, __memcpy_avx512_no_vzeroupper)
|
|
strong_alias (__memmove_chk_avx512_no_vzeroupper, __memcpy_chk_avx512_no_vzeroupper)
|
|
#endif
|