Free cookie consent management tool by TermsFeed Policy Generator

source: trunk/HeuristicLab.ExtLibs/HeuristicLab.NativeInterpreter/0.1/NativeInterpreter-0.1/src/vector_operations.h @ 18091

Last change on this file since 18091 was 16905, checked in by gkronber, 6 years ago

#2915: Corrected calculation of cuberoot function in all interpreters and updated all formatters.
I tested a run with all interpreters and got the same results with all of them.

File size: 4.7 KB
Line 
1#ifndef VECTOR_OPERATIONS_H
2#define VECTOR_OPERATIONS_H
3
4#define _USE_MATH_DEFINES
5#include <cmath>
6#include <cstring>
7
8#ifdef USE_VDT
9#include "vdt/vdtMath.h"
10#include "vdt/stdwrap.h"
11#define hl_exp vdt::fast_exp
12#define hl_log vdt::fast_log
13#define hl_sin vdt::fast_sin
14#define hl_cos vdt::fast_cos
15#define hl_tan vdt::fast_tan
16#define hl_tanh vdt::fast_tanh
17#define hl_sqrt vdt::fast_sqrt
18#define hl_pow vdt::fast_pow
19#define hl_cbrt std::cbrt
20#define hl_round vdt::fast_round
21#define hl_inv vdt::fast_inv
22#else
23#define hl_exp std::exp
24#define hl_log std::log
25#define hl_sin std::sin
26#define hl_cos std::cos
27#define hl_tan std::tan
28#define hl_tanh std::tanh
29#define hl_sqrt std::sqrt
30#define hl_pow std::pow
31#define hl_cbrt std::cbrt
32#define hl_round std::round
33#define hl_inv(x) 1. / x;
34
35#endif
36
37constexpr int BATCHSIZE = 64;
38
39#define FOR(i) for(int i = 0; i < BATCHSIZE; ++i)
40
41// When auto-vectorizing without __restrict,
42// gcc and clang check for overlap (with a bunch of integer code)
43// before running the vectorized loop
44
45// vector - vector operations
46inline void load(double* __restrict a, double const * __restrict b) noexcept { std::memcpy(a, b, BATCHSIZE * sizeof(double)); }
47inline void add(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] += b[i]; }
48inline void sub(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] -= b[i]; }
49inline void mul(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] *= b[i]; }
50inline void div(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] /= b[i]; }
51inline void exp(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_exp(b[i]); }
52inline void log(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_log(b[i]); }
53inline void sin(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_sin(b[i]); }
54inline void cos(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_cos(b[i]); }
55inline void tan(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_tan(b[i]); }
56inline void tanh(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_tanh(b[i]); }
57inline void sqrt(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_sqrt(b[i]); }
58inline void pow(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_pow(a[i], hl_round(b[i])); };
59inline void root(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_pow(a[i], 1. / hl_round(b[i])); };
60inline void cbrt(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_cbrt(b[i]); };
61inline void square(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_pow(b[i], 2.); };
62inline void inv(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = hl_inv(b[i]); }
63inline void neg(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = -b[i]; }
64inline void abs(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] = std::fabs(b[i]); }
65inline void analytical_quotient(double* __restrict a, double const * __restrict b) noexcept { FOR(i) a[i] /= hl_sqrt(b[i]*b[i] + 1.); }
66
67// vector - scalar operations
68inline void load(double* __restrict a, double s) noexcept { FOR(i) a[i] = s; }
69inline void add(double* __restrict a, double s) noexcept { FOR(i) a[i] += s; }
70inline void sub(double* __restrict a, double s) noexcept { FOR(i) a[i] -= s; }
71inline void mul(double* __restrict a, double s) noexcept { FOR(i) a[i] *= s; }
72inline void div(double* __restrict a, double s) noexcept { FOR(i) a[i] /= s; }
73inline void pow(double* __restrict dst, double const * __restrict src, double s) noexcept { FOR(i) dst[i] = hl_pow(src[i], s); }
74
75// vector operations
76inline void neg(double* __restrict a) noexcept { FOR(i) a[i] = -a[i]; }
77inline void inv(double* __restrict a) noexcept { FOR(i) a[i] = hl_inv(a[i]); }
78inline void exp(double* __restrict a) noexcept { FOR(i) a[i] = hl_exp(a[i]); }
79inline void log(double* __restrict a) noexcept { FOR(i) a[i] = hl_log(a[i]); }
80inline void sin(double* __restrict a) noexcept { FOR(i) a[i] = hl_sin(a[i]); }
81inline void cos(double* __restrict a) noexcept { FOR(i) a[i] = hl_cos(a[i]); }
82inline void tan(double* __restrict a) noexcept { FOR(i) a[i] = hl_tan(a[i]); }
83inline void tanh(double* __restrict a) noexcept { FOR(i) a[i] = hl_tanh(a[i]); }
84inline void round(double* __restrict a) noexcept { FOR(i) a[i] = hl_round(a[i]); }
85inline void square(double* __restrict a) noexcept { FOR(i) a[i] = hl_pow(a[i], 2.); }
86inline void cbrt(double* __restrict a) noexcept { FOR(i) a[i] = hl_cbrt(a[i]); }
87
88#undef FOR
89#endif
Note: See TracBrowser for help on using the repository browser.