forked from timblechmann/nova-simd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
simd_unary_arithmetic.hpp
85 lines (69 loc) · 3.18 KB
/
simd_unary_arithmetic.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
// unary arithmetic simd functions
// Copyright (C) 2010 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_UNARY_ARITHMETIC_HPP
#define SIMD_UNARY_ARITHMETIC_HPP
#include "vec.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 {
namespace detail {
#define DEFINE_UNARY_FUNCTOR(NAME, VEC_NAME) \
struct NAME##_ \
{ \
template <typename FloatType> \
always_inline FloatType operator()(FloatType arg) const \
{ \
return NAME(arg); \
} \
template <typename FloatType> \
always_inline vec<FloatType> operator()(vec<FloatType> arg) const \
{ \
return VEC_NAME(arg); \
} \
};
DEFINE_UNARY_FUNCTOR(fabs, abs)
DEFINE_UNARY_FUNCTOR(sign, sign)
DEFINE_UNARY_FUNCTOR(square, square)
DEFINE_UNARY_FUNCTOR(cube, cube)
DEFINE_UNARY_FUNCTOR(reciprocal, reciprocal)
DEFINE_UNARY_FUNCTOR(round, round)
DEFINE_UNARY_FUNCTOR(frac, frac)
DEFINE_UNARY_FUNCTOR(ceil, ceil)
DEFINE_UNARY_FUNCTOR(floor, floor)
DEFINE_UNARY_FUNCTOR(trunc, trunc)
DEFINE_UNARY_FUNCTOR(undenormalize, undenormalize)
} /* namespace detail */
NOVA_SIMD_DEFINE_UNARY_WRAPPER(abs, detail::fabs_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(sgn, detail::sign_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(square, detail::square_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(cube, detail::cube_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(round, detail::round_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(frac, detail::frac_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(ceil, detail::ceil_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(floor, detail::floor_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(trunc, detail::trunc_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(undenormalize, detail::undenormalize_)
NOVA_SIMD_DEFINE_UNARY_WRAPPER(reciprocal, detail::reciprocal_)
} /* namespace nova */
#undef always_inline
#undef DEFINE_UNARY_FUNCTOR
#endif /* SIMD_UNARY_ARITHMETIC_HPP */