[libc-commits] [libc] ac763b9 - [libc] Add support for fma in riscv

[Mikhail R. Gadelha via libc-commits]

Author: Mikhail R. Gadelha
Date: 2023-03-08T16:15:24-03:00
New Revision: ac763b9fdf5b8b6cc563b30c3fe90499f3706b13

URL: https://github.com/llvm/llvm-project/commit/ac763b9fdf5b8b6cc563b30c3fe90499f3706b13
DIFF: https://github.com/llvm/llvm-project/commit/ac763b9fdf5b8b6cc563b30c3fe90499f3706b13.diff

LOG: [libc] Add support for fma in riscv

Reviewed By: lntue

Differential Revision: https://reviews.llvm.org/D145592

Added: 
    libc/src/__support/FPUtil/riscv64/FMA.h

Modified: 
    libc/src/__support/FPUtil/FMA.h
    libc/src/__support/macros/properties/cpu_features.h

Removed: 
    


################################################################################
diff  --git a/libc/src/__support/FPUtil/FMA.h b/libc/src/__support/FPUtil/FMA.h
index 05df9cb52668b..79a465a5be45a 100644
--- a/libc/src/__support/FPUtil/FMA.h
+++ b/libc/src/__support/FPUtil/FMA.h
@@ -18,6 +18,8 @@
 #include "x86_64/FMA.h"
 #elif defined(LIBC_TARGET_ARCH_IS_AARCH64)
 #include "aarch64/FMA.h"
+#elif defined(LIBC_TARGET_ARCH_IS_RISCV64)
+#include "riscv64/FMA.h"
 #endif
 
 #else

diff  --git a/libc/src/__support/FPUtil/riscv64/FMA.h b/libc/src/__support/FPUtil/riscv64/FMA.h
new file mode 100644
index 0000000000000..9b003f10adbf2
--- /dev/null
+++ b/libc/src/__support/FPUtil/riscv64/FMA.h
@@ -0,0 +1,49 @@
+//===-- RISCV64 implementations of the fma function -------------*- 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 LLVM_LIBC_SRC_SUPPORT_FPUTIL_RISCV64_FMA_H
+#define LLVM_LIBC_SRC_SUPPORT_FPUTIL_RISCV64_FMA_H
+
+#include "src/__support/macros/properties/architectures.h"
+#include "src/__support/macros/properties/cpu_features.h" // LIBC_TARGET_CPU_HAS_FMA
+
+#if !defined(LIBC_TARGET_ARCH_IS_RISCV64)
+#error "Invalid include"
+#endif
+
+#if !defined(LIBC_TARGET_CPU_HAS_FMA)
+#error "FMA instructions are not supported"
+#endif
+
+#include "src/__support/CPP/type_traits.h"
+
+namespace __llvm_libc {
+namespace fputil {
+
+template <typename T>
+cpp::enable_if_t<cpp::is_same_v<T, float>, T> fma(T x, T y, T z) {
+  float result;
+  __asm__ __volatile__("fmadd.s %0, %1, %2, %3\n\t"
+                       : "=f"(result)
+                       : "f"(x), "f"(y), "f"(z));
+  return result;
+}
+
+template <typename T>
+cpp::enable_if_t<cpp::is_same_v<T, double>, T> fma(T x, T y, T z) {
+  double result;
+  __asm__ __volatile__("fmadd.d %0, %1, %2, %3\n\t"
+                       : "=f"(result)
+                       : "f"(x), "f"(y), "f"(z));
+  return result;
+}
+
+} // namespace fputil
+} // namespace __llvm_libc
+
+#endif // LLVM_LIBC_SRC_SUPPORT_FPUTIL_RISCV64_FMA_H

diff  --git a/libc/src/__support/macros/properties/cpu_features.h b/libc/src/__support/macros/properties/cpu_features.h
index b986351ecbec7..9884d561195f4 100644
--- a/libc/src/__support/macros/properties/cpu_features.h
+++ b/libc/src/__support/macros/properties/cpu_features.h
@@ -36,7 +36,8 @@
 #define LIBC_TARGET_CPU_HAS_AVX512BW
 #endif
 
-#if defined(__ARM_FEATURE_FMA) || defined(__AVX2__) || defined(__FMA__)
+#if defined(__ARM_FEATURE_FMA) || defined(__AVX2__) || defined(__FMA__) ||     \
+    defined(__riscv)
 #define LIBC_TARGET_CPU_HAS_FMA
 #endif