libcxx/include/__math/fma.h

   1 //===----------------------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef _LIBCPP___MATH_FMA_H
  10 #define _LIBCPP___MATH_FMA_H
  11
  12 #include <__config>
  13 #include <__type_traits/enable_if.h>
  14 #include <__type_traits/is_arithmetic.h>
  15 #include <__type_traits/is_same.h>
  16 #include <__type_traits/promote.h>
  17
  18 #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
  19 #  pragma GCC system_header
  20 #endif
  21
  22 _LIBCPP_BEGIN_NAMESPACE_STD
  23
  24 namespace __math {
  25
  26 inline _LIBCPP_HIDE_FROM_ABI float fma(float __x, float __y, float __z) _NOEXCEPT {
  27   return __builtin_fmaf(__x, __y, __z);
  28 }
  29
  30 template <class = int>
  31 _LIBCPP_HIDE_FROM_ABI double fma(double __x, double __y, double __z) _NOEXCEPT {
  32   return __builtin_fma(__x, __y, __z);
  33 }
  34
  35 inline _LIBCPP_HIDE_FROM_ABI long double fma(long double __x, long double __y, long double __z) _NOEXCEPT {
  36   return __builtin_fmal(__x, __y, __z);
  37 }
  38
  39 template <class _A1,
  40           class _A2,
  41           class _A3,
  42           __enable_if_t<is_arithmetic<_A1>::value && is_arithmetic<_A2>::value && is_arithmetic<_A3>::value, int> = 0>
  43 inline _LIBCPP_HIDE_FROM_ABI typename __promote<_A1, _A2, _A3>::type fma(_A1 __x, _A2 __y, _A3 __z) _NOEXCEPT {
  44   using __result_type = typename __promote<_A1, _A2, _A3>::type;
  45   static_assert(
  46       !(_IsSame<_A1, __result_type>::value && _IsSame<_A2, __result_type>::value && _IsSame<_A3, __result_type>::value),
  47       "");
  48   return __builtin_fma((__result_type)__x, (__result_type)__y, (__result_type)__z);
  49 }
  50
  51 } // namespace __math
  52
  53 _LIBCPP_END_NAMESPACE_STD
  54
  55 #endif // _LIBCPP___MATH_FMA_H