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 |
56 | |