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
|
/***************************************************************************
* Copyright (c) Johan Mabille, Sylvain Corlay, Wolf Vollprecht and *
* Martin Renou *
* Copyright (c) QuantStack *
* Copyright (c) Serge Guelton *
* *
* Distributed under the terms of the BSD 3-Clause License. *
* *
* The full license is in the file LICENSE, distributed with this software. *
****************************************************************************/
#include "xsimd/xsimd.hpp"
#ifndef XSIMD_NO_SUPPORTED_ARCHITECTURE
#include "test_utils.hpp"
template <class B>
struct poly_evaluation_test
{
using batch_type = B;
using value_type = typename B::value_type;
static constexpr size_t size = B::size;
using vector_type = std::vector<value_type>;
size_t nb_input;
vector_type input;
vector_type horner_res;
vector_type estrin_res;
poly_evaluation_test()
{
nb_input = size * 10000;
input.resize(nb_input);
for (size_t i = 0; i < nb_input; ++i)
{
input[i] = value_type(i) / 4 + value_type(1.2) * std::sqrt(value_type(i + 0.25));
}
horner_res.resize(nb_input);
estrin_res.resize(nb_input);
}
void test_poly_evaluation()
{
batch_type in, out;
for (size_t i = 0; i < nb_input; i += size)
{
detail::load_batch(in, input, i);
out = xsimd::kernel::horner<typename batch_type::value_type, typename batch_type::arch_type, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16>(in);
detail::store_batch(out, horner_res, i);
out = xsimd::kernel::estrin<typename batch_type::value_type, typename batch_type::arch_type, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16>(in);
detail::store_batch(out, estrin_res, i);
}
size_t diff = detail::get_nb_diff(horner_res, estrin_res);
CHECK_EQ(diff, 0);
}
};
TEST_CASE_TEMPLATE("[poly evaluation]", B, BATCH_FLOAT_TYPES)
{
poly_evaluation_test<B> Test;
Test.test_poly_evaluation();
}
#endif
|
