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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/*
* GridTools
*
* Copyright (c) 2014-2023, ETH Zurich
* All rights reserved.
*
* Please, refer to the LICENSE file in the root directory.
* SPDX-License-Identifier: BSD-3-Clause
*/
#include <gridtools/sid/loop.hpp>
#include <gtest/gtest.h>
#include <gridtools/common/compose.hpp>
#include <gridtools/common/integral_constant.hpp>
#include <gridtools/common/tuple.hpp>
namespace gridtools {
namespace {
using namespace literals;
struct assignment_f {
double m_val;
template <class Strides>
GT_FUNCTION void operator()(double *ptr, Strides const &) const {
*ptr = m_val;
}
};
using i_t = integral_constant<int, 0>;
using j_t = integral_constant<int, 1>;
TEST(make_loop, smoke) {
double data[10][10] = {};
auto strides = tuple(10_c, 1_c);
double *ptr = &data[0][0];
sid::make_loop<i_t>(5, 1)(assignment_f{42})(ptr, strides);
for (int i = 0; i < 5; ++i)
EXPECT_EQ(42, data[i][0]) << " i:" << i;
ptr = &data[2][3];
sid::make_loop<j_t>(4_c, -1_c)(assignment_f{5})(ptr, strides);
for (int i = 0; i < 4; ++i)
EXPECT_EQ(5, data[2][i]) << " i:" << i;
ptr = &data[0][0];
sid::make_loop<i_t>(10_c)(sid::make_loop<j_t>(10_c)(assignment_f{88}))(ptr, strides);
for (int i = 0; i < 10; ++i)
for (int j = 0; j < 10; ++j)
EXPECT_EQ(88, data[i][j]) << " i:" << i << ", j:" << j;
// pass ptr as r-value
sid::make_loop<i_t>(10_c)(sid::make_loop<j_t>(10_c)(assignment_f{88}))(&data[0][0], strides);
for (int i = 0; i < 10; ++i)
for (int j = 0; j < 10; ++j)
EXPECT_EQ(88, data[i][j]) << " i:" << i << ", j:" << j;
}
TEST(make_unrolled_loop, smoke) {
double data[10][10] = {};
auto strides = tuple(10_c, 1_c);
double *ptr = &data[0][0];
sid::make_unrolled_loop<i_t, 2>(5, 1)(assignment_f{42})(ptr, strides);
for (int i = 0; i < 5; ++i)
EXPECT_EQ(42, data[i][0]) << " i:" << i;
ptr = &data[2][3];
sid::make_unrolled_loop<j_t, 3>(4_c, -1_c)(assignment_f{5})(ptr, strides);
for (int i = 0; i < 4; ++i)
EXPECT_EQ(5, data[2][i]) << " i:" << i;
ptr = &data[0][0];
sid::make_unrolled_loop<i_t, 2>(10_c)(sid::make_unrolled_loop<j_t, 7>(10_c)(assignment_f{88}))(
ptr, strides);
for (int i = 0; i < 10; ++i)
for (int j = 0; j < 10; ++j)
EXPECT_EQ(88, data[i][j]) << " i:" << i << ", j:" << j;
// pass ptr as r-value
sid::make_unrolled_loop<i_t, 3>(10_c)(sid::make_unrolled_loop<j_t, 1>(10_c)(assignment_f{88}))(
&data[0][0], strides);
for (int i = 0; i < 10; ++i)
for (int j = 0; j < 10; ++j)
EXPECT_EQ(88, data[i][j]) << " i:" << i << ", j:" << j;
}
TEST(nest_loops, smoke) {
double data[10][10] = {};
double *ptr = &data[0][0];
auto strides = tuple(10_c, 1_c);
auto testee = compose(sid::make_loop<i_t>(10_c), sid::make_loop<j_t>(10_c));
testee(assignment_f{42})(ptr, strides);
for (int i = 0; i < 10; ++i)
for (int j = 0; j < 10; ++j)
EXPECT_EQ(42, data[i][j]) << " i:" << i << ", j:" << j;
}
TEST(range, smoke) {
double data[10][10] = {};
double *ptr = &data[0][0];
auto strides = tuple(10_c, 1_c);
auto testee = sid::make_range(ptr, strides, sid::make_loop<i_t>(10_c), sid::make_loop<j_t>(10_c));
for (auto &&val : testee)
val = 42;
for (int i = 0; i < 10; ++i)
for (int j = 0; j < 10; ++j)
EXPECT_EQ(42, data[i][j]) << " i:" << i << ", j:" << j;
}
} // namespace
} // namespace gridtools
|
