xref: /llvm-project/libc/src/string/memory_utils/inline_memset.h (revision 2cc97951400ca2ab79d6bdeccffa6e431882a86e)
11f578347SGuillaume Chatelet //===-- Implementation of memset and bzero --------------------------------===//
21f578347SGuillaume Chatelet //
31f578347SGuillaume Chatelet // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
41f578347SGuillaume Chatelet // See https://llvm.org/LICENSE.txt for license information.
51f578347SGuillaume Chatelet // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
61f578347SGuillaume Chatelet //
71f578347SGuillaume Chatelet //===----------------------------------------------------------------------===//
81f578347SGuillaume Chatelet 
91f578347SGuillaume Chatelet #ifndef LLVM_LIBC_SRC_STRING_MEMORY_UTILS_INLINE_MEMSET_H
101f578347SGuillaume Chatelet #define LLVM_LIBC_SRC_STRING_MEMORY_UTILS_INLINE_MEMSET_H
111f578347SGuillaume Chatelet 
12*2cc97951Sc8ef #include "src/__support/macros/attributes.h"               // LIBC_INLINE
131f578347SGuillaume Chatelet #include "src/__support/macros/properties/architectures.h" // LIBC_TARGET_ARCH_IS_
141f578347SGuillaume Chatelet #include "src/string/memory_utils/utils.h"                 // Ptr, CPtr
151f578347SGuillaume Chatelet 
161f578347SGuillaume Chatelet #include <stddef.h> // size_t
171f578347SGuillaume Chatelet 
181f578347SGuillaume Chatelet #if defined(LIBC_TARGET_ARCH_IS_X86)
191f578347SGuillaume Chatelet #include "src/string/memory_utils/x86_64/inline_memset.h"
201f578347SGuillaume Chatelet #define LIBC_SRC_STRING_MEMORY_UTILS_MEMSET inline_memset_x86
211f578347SGuillaume Chatelet #elif defined(LIBC_TARGET_ARCH_IS_AARCH64)
221f578347SGuillaume Chatelet #include "src/string/memory_utils/aarch64/inline_memset.h"
231f578347SGuillaume Chatelet #define LIBC_SRC_STRING_MEMORY_UTILS_MEMSET inline_memset_aarch64
241f578347SGuillaume Chatelet #elif defined(LIBC_TARGET_ARCH_IS_ANY_RISCV)
251f578347SGuillaume Chatelet #include "src/string/memory_utils/riscv/inline_memset.h"
261f578347SGuillaume Chatelet #define LIBC_SRC_STRING_MEMORY_UTILS_MEMSET inline_memset_riscv
27452fa6b8SJoseph Huber #elif defined(LIBC_TARGET_ARCH_IS_ARM)
281f578347SGuillaume Chatelet #include "src/string/memory_utils/generic/byte_per_byte.h"
291f578347SGuillaume Chatelet #define LIBC_SRC_STRING_MEMORY_UTILS_MEMSET inline_memset_byte_per_byte
30452fa6b8SJoseph Huber #elif defined(LIBC_TARGET_ARCH_IS_GPU)
31452fa6b8SJoseph Huber #include "src/string/memory_utils/generic/builtin.h"
32452fa6b8SJoseph Huber #define LIBC_SRC_STRING_MEMORY_UTILS_MEMSET inline_memset_builtin
331f578347SGuillaume Chatelet #else
341f578347SGuillaume Chatelet #error "Unsupported architecture"
351f578347SGuillaume Chatelet #endif
361f578347SGuillaume Chatelet 
375ff3ff33SPetr Hosek namespace LIBC_NAMESPACE_DECL {
381f578347SGuillaume Chatelet 
391f578347SGuillaume Chatelet LIBC_INLINE static void inline_memset(void *dst, uint8_t value, size_t count) {
401f578347SGuillaume Chatelet   LIBC_SRC_STRING_MEMORY_UTILS_MEMSET(reinterpret_cast<Ptr>(dst), value, count);
411f578347SGuillaume Chatelet }
421f578347SGuillaume Chatelet 
435ff3ff33SPetr Hosek } // namespace LIBC_NAMESPACE_DECL
441f578347SGuillaume Chatelet 
451f578347SGuillaume Chatelet #undef LIBC_SRC_STRING_MEMORY_UTILS_MEMSET
461f578347SGuillaume Chatelet 
471f578347SGuillaume Chatelet #endif // LLVM_LIBC_SRC_STRING_MEMORY_UTILS_INLINE_MEMSET_H
48