diff --git a/libc/src/string/memory_utils/CMakeLists.txt b/libc/src/string/memory_utils/CMakeLists.txt --- a/libc/src/string/memory_utils/CMakeLists.txt +++ b/libc/src/string/memory_utils/CMakeLists.txt @@ -18,6 +18,7 @@ op_generic.h op_x86.h riscv/bcmp_implementations.h + riscv/memcmp_implementations.h riscv/memcpy_implementations.h utils.h x86_64/bcmp_implementations.h diff --git a/libc/src/string/memory_utils/memcmp_implementations.h b/libc/src/string/memory_utils/memcmp_implementations.h --- a/libc/src/string/memory_utils/memcmp_implementations.h +++ b/libc/src/string/memory_utils/memcmp_implementations.h @@ -9,44 +9,36 @@ #ifndef LLVM_LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP_IMPLEMENTATIONS_H #define LLVM_LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP_IMPLEMENTATIONS_H -#include "src/__support/common.h" -#include "src/__support/macros/optimization.h" // LIBC_UNLIKELY LIBC_LOOP_NOUNROLL -#include "src/__support/macros/properties/architectures.h" -#include "src/string/memory_utils/generic/aligned_access.h" -#include "src/string/memory_utils/generic/byte_per_byte.h" -#include "src/string/memory_utils/op_generic.h" -#include "src/string/memory_utils/op_riscv.h" -#include "src/string/memory_utils/utils.h" // CPtr MemcmpReturnType +#include "src/__support/macros/config.h" // LIBC_INLINE +#include "src/__support/macros/properties/architectures.h" // LIBC_TARGET_ARCH_IS_ +#include "src/string/memory_utils/utils.h" // Ptr, CPtr #include // size_t #if defined(LIBC_TARGET_ARCH_IS_X86) #include "src/string/memory_utils/x86_64/memcmp_implementations.h" +#define LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP inline_memcmp_x86 #elif defined(LIBC_TARGET_ARCH_IS_AARCH64) #include "src/string/memory_utils/aarch64/memcmp_implementations.h" +#define LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP inline_memcmp_aarch64 +#elif defined(LIBC_TARGET_ARCH_IS_ANY_RISCV) +#include "src/string/memory_utils/riscv/memcmp_implementations.h" +#define LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP inline_memcmp_riscv +#else +// We may want to error instead of defaulting to suboptimal implementation. +#include "src/string/memory_utils/generic/byte_per_byte.h" +#define LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP inline_memcmp_byte_per_byte #endif namespace __llvm_libc { -LIBC_INLINE MemcmpReturnType inline_memcmp(CPtr p1, CPtr p2, size_t count) { -#if defined(LIBC_TARGET_ARCH_IS_X86) - return inline_memcmp_x86(p1, p2, count); -#elif defined(LIBC_TARGET_ARCH_IS_AARCH64) - return inline_memcmp_aarch64(p1, p2, count); -#elif defined(LIBC_TARGET_ARCH_IS_RISCV64) - return inline_memcmp_aligned_access_64bit(p1, p2, count); -#elif defined(LIBC_TARGET_ARCH_IS_RISCV32) - return inline_memcmp_aligned_access_32bit(p1, p2, count); -#else - return inline_memcmp_byte_per_byte(p1, p2, count); -#endif -} - LIBC_INLINE int inline_memcmp(const void *p1, const void *p2, size_t count) { - return static_cast(inline_memcmp(reinterpret_cast(p1), - reinterpret_cast(p2), count)); + return static_cast(LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP( + reinterpret_cast(p1), reinterpret_cast(p2), count)); } } // namespace __llvm_libc +#undef LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP + #endif // LLVM_LIBC_SRC_STRING_MEMORY_UTILS_MEMCMP_IMPLEMENTATIONS_H diff --git a/libc/src/string/memory_utils/riscv/memcmp_implementations.h b/libc/src/string/memory_utils/riscv/memcmp_implementations.h new file mode 100644 --- /dev/null +++ b/libc/src/string/memory_utils/riscv/memcmp_implementations.h @@ -0,0 +1,33 @@ +//===-- Memcmp implementation for riscv -------------------------*- C++ -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +#ifndef LIBC_SRC_STRING_MEMORY_UTILS_RISCV_MEMCMP_IMPLEMENTATIONS_H +#define LIBC_SRC_STRING_MEMORY_UTILS_RISCV_MEMCMP_IMPLEMENTATIONS_H + +#include "src/__support/macros/attributes.h" // LIBC_INLINE +#include "src/__support/macros/properties/architectures.h" // LIBC_TARGET_ARCH_IS_RISCV64 +#include "src/string/memory_utils/generic/aligned_access.h" +#include "src/string/memory_utils/utils.h" // Ptr, CPtr + +#include // size_t + +namespace __llvm_libc { + +[[maybe_unused]] LIBC_INLINE MemcmpReturnType +inline_memcmp_riscv(CPtr p1, CPtr p2, size_t count) { +#if defined(LIBC_TARGET_ARCH_IS_RISCV64) + return inline_memcmp_aligned_access_64bit(p1, p2, count); +#elif defined(LIBC_TARGET_ARCH_IS_RISCV32) + return inline_memcmp_aligned_access_32bit(p1, p2, count); +#else +#error "Unimplemented" +#endif +} + +} // namespace __llvm_libc + +#endif // LIBC_SRC_STRING_MEMORY_UTILS_RISCV_MEMCMP_IMPLEMENTATIONS_H diff --git a/utils/bazel/llvm-project-overlay/libc/BUILD.bazel b/utils/bazel/llvm-project-overlay/libc/BUILD.bazel --- a/utils/bazel/llvm-project-overlay/libc/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/libc/BUILD.bazel @@ -2022,6 +2022,7 @@ "src/string/memory_utils/memmove_implementations.h", "src/string/memory_utils/memset_implementations.h", "src/string/memory_utils/riscv/bcmp_implementations.h", + "src/string/memory_utils/riscv/memcmp_implementations.h", "src/string/memory_utils/riscv/memcpy_implementations.h", "src/string/memory_utils/strcmp_implementations.h", "src/string/memory_utils/strstr_implementations.h",