glibc/sysdeps/x86_64/multiarch/memset.S

/* Multiple versions of memset
   All versions must be listed in ifunc-impl-list.c.
   Copyright (C) 2014-2017 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
   <http://www.gnu.org/licenses/>.  */

#include <sysdep.h>
#include <shlib-compat.h>
#include <init-arch.h>

/* Define multiple versions only for the definition in lib.  */
#if IS_IN (libc)
ENTRY(memset)
	.type	memset, @gnu_indirect_function
	LOAD_RTLD_GLOBAL_RO_RDX
	lea	__memset_erms(%rip), %RAX_LP
	HAS_ARCH_FEATURE (Prefer_ERMS)
	jnz	2f
	lea	__memset_sse2_unaligned_erms(%rip), %RAX_LP
	HAS_CPU_FEATURE (ERMS)
	jnz	1f
	lea	__memset_sse2_unaligned(%rip), %RAX_LP
1:
	HAS_ARCH_FEATURE (AVX2_Usable)
	jz	2f
	lea	__memset_avx2_unaligned_erms(%rip), %RAX_LP
	HAS_CPU_FEATURE (ERMS)
	jnz	L(AVX512F)
	lea	__memset_avx2_unaligned(%rip), %RAX_LP
L(AVX512F):
	HAS_ARCH_FEATURE (Prefer_No_AVX512)
	jnz	2f
	HAS_ARCH_FEATURE (AVX512F_Usable)
	jz	2f
	lea	__memset_avx512_no_vzeroupper(%rip), %RAX_LP
	HAS_ARCH_FEATURE (Prefer_No_VZEROUPPER)
	jnz	2f
	lea	__memset_avx512_unaligned_erms(%rip), %RAX_LP
	HAS_CPU_FEATURE (ERMS)
	jnz	2f
	lea	__memset_avx512_unaligned(%rip), %RAX_LP
2:	ret
END(memset)
#endif

#if IS_IN (libc)
# define MEMSET_SYMBOL(p,s)	p##_sse2_##s
# define WMEMSET_SYMBOL(p,s)	p##_sse2_##s

# ifdef SHARED
#  undef libc_hidden_builtin_def
/* It doesn't make sense to send libc-internal memset calls through a PLT.
   The speedup we get from using SSE2 instructions is likely eaten away
   by the indirect call in the PLT.  */
#  define libc_hidden_builtin_def(name) \
	.globl __GI_memset; __GI_memset = __memset_sse2_unaligned; \
	.globl __GI_wmemset; __GI_wmemset = __wmemset_sse2_unaligned; \
	.globl __GI___wmemset; __GI___wmemset = __wmemset_sse2_unaligned
# endif

# undef weak_alias
# define weak_alias(original, alias) \
	.weak bzero; bzero = __bzero

# undef strong_alias
# define strong_alias(original, alias)
#endif

#include "../memset.S"