forked from timblechmann/nova-simd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
simd_math.hpp
101 lines (78 loc) · 3.6 KB
/
simd_math.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// simdfied mathematical functions
// Copyright (C) 2009 Tim Blechmann
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; see the file COPYING. If not, write to
// the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
// Boston, MA 02111-1307, USA.
#ifndef SIMD_MATH_HPP
#define SIMD_MATH_HPP
#include "vec.hpp"
#include "detail/math.hpp"
#include "detail/unroll_helpers.hpp"
#include "detail/define_macros.hpp"
#if defined(__GNUC__) && defined(NDEBUG)
#define always_inline inline __attribute__((always_inline))
#else
#define always_inline inline
#endif
namespace nova {
#define DEFINE_UNARY_FUNCTOR(NAME) \
namespace detail { \
struct NAME##_ \
{ \
template <typename FloatType> \
always_inline FloatType operator()(FloatType arg) const \
{ \
return NAME(arg); \
} \
}; \
} // namespace detail
#define DEFINE_UNARY_MATH_FUNCTIONS(NAME) \
DEFINE_UNARY_FUNCTOR(NAME) \
NOVA_SIMD_DEFINE_UNARY_WRAPPER(NAME, detail::NAME##_)
DEFINE_UNARY_MATH_FUNCTIONS(sin)
DEFINE_UNARY_MATH_FUNCTIONS(cos)
DEFINE_UNARY_MATH_FUNCTIONS(tan)
DEFINE_UNARY_MATH_FUNCTIONS(asin)
DEFINE_UNARY_MATH_FUNCTIONS(acos)
DEFINE_UNARY_MATH_FUNCTIONS(atan)
DEFINE_UNARY_MATH_FUNCTIONS(tanh)
DEFINE_UNARY_MATH_FUNCTIONS(log)
DEFINE_UNARY_MATH_FUNCTIONS(log2)
DEFINE_UNARY_MATH_FUNCTIONS(log10)
DEFINE_UNARY_MATH_FUNCTIONS(exp)
DEFINE_UNARY_MATH_FUNCTIONS(signed_sqrt)
#define DEFINE_BINARY_MATH_FUNCTOR(NAME) \
namespace detail { \
struct NAME##_ \
{ \
template <typename FloatType> \
always_inline FloatType operator()(FloatType lhs, FloatType rhs) const \
{ \
return NAME(lhs, rhs); \
} \
}; \
\
}
DEFINE_BINARY_MATH_FUNCTOR(pow)
DEFINE_BINARY_MATH_FUNCTOR(signed_pow)
NOVA_SIMD_DEFINE_BINARY_WRAPPER(pow, detail::pow_)
NOVA_SIMD_DEFINE_BINARY_WRAPPER(spow, detail::signed_pow_)
}
#undef DEFINE_UNARY_FUNCTOR
#undef DEFINE_BINARY_MATH_FUNCTOR
#undef DEFINE_UNARY_MATH_FUNCTIONS
#undef DEFINE_BINARY_MATH_FUNCTIONS
#undef always_inline
#endif /* SIMD_MATH_HPP */