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
|
/*
* 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
*/
#pragma once
#include <gridtools/stencil/cartesian.hpp>
#include <gtest/gtest.h>
#define GT_STENCIL_NAIVE
#include <stencil_select.hpp>
#include <test_environment.hpp>
namespace gridtools {
namespace stencil {
namespace cartesian {
using x_interval = axis<1>::full_interval;
struct copy_functor {
typedef in_accessor<0> in;
typedef inout_accessor<1> out;
typedef make_param_list<in, out> param_list;
template <typename Evaluation>
GT_FUNCTION static void apply(Evaluation &eval, x_interval) {
eval(out()) = eval(in());
}
};
struct copy_functor_with_expression {
typedef in_accessor<0> in;
typedef inout_accessor<1> out;
typedef make_param_list<in, out> param_list;
template <typename Evaluation>
GT_FUNCTION static void apply(Evaluation &eval, x_interval) {
using namespace expressions;
eval(out()) = eval(in() + 0.);
}
};
struct copy_functor_default_interval {
typedef in_accessor<0> in;
typedef inout_accessor<1> out;
typedef make_param_list<in, out> param_list;
template <class Eval>
GT_FUNCTION static void apply(Eval &&eval) {
eval(out()) = eval(in());
}
};
struct base_fixture : ::testing::Test {
using env_t = test_environment<1, axis<1>>::apply<naive, double, inlined_params<13, 9, 7>>;
template <class Fun, class... Storages>
void run_computation(Storages &&...storages) const {
run_single_stage(Fun(), naive(), env_t::make_grid(), std::forward<Storages>(storages)...);
}
using fun_t = std::function<double(int, int, int)>;
fun_t input = [](int i, int j, int k) { return i * 100 + j * 10 + k; };
fun_t shifted = [this](int i, int j, int k) { return input(i + 1, j + 1, k); };
};
} // namespace cartesian
} // namespace stencil
} // namespace gridtools
|
