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
|
/*
* 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 <type_traits>
#include <gridtools/meta.hpp>
// fn backend
#if defined(GT_FN_NAIVE)
#ifndef GT_STENCIL_NAIVE
#define GT_STENCIL_NAIVE
#endif
#ifndef GT_STORAGE_CPU_KFIRST
#define GT_STORAGE_CPU_KFIRST
#endif
#ifndef GT_TIMER_DUMMY
#define GT_TIMER_DUMMY
#endif
#include <gridtools/fn/backend/naive.hpp>
namespace {
using fn_backend_t = gridtools::fn::backend::naive;
}
#elif defined(GT_FN_GPU)
#ifndef GT_STENCIL_GPU
#define GT_STENCIL_GPU
#endif
#ifndef GT_STORAGE_GPU
#define GT_STORAGE_GPU
#endif
#ifndef GT_TIMER_CUDA
#define GT_TIMER_CUDA
#endif
#include <gridtools/fn/backend/gpu.hpp>
namespace {
template <int... sizes>
using block_sizes_t =
gridtools::meta::zip<gridtools::meta::iseq_to_list<std::make_integer_sequence<int, sizeof...(sizes)>,
gridtools::meta::list,
gridtools::integral_constant>,
gridtools::meta::list<gridtools::integral_constant<int, sizes>...>>;
using fn_backend_t = gridtools::fn::backend::gpu<block_sizes_t<32, 8, 1>>;
} // namespace
#endif
#include "stencil_select.hpp"
#include "storage_select.hpp"
#include "timer_select.hpp"
namespace gridtools::fn::backend {
namespace naive_impl_ {
template <class ThreadPool>
struct naive_with_threadpool;
template <class ThreadPool>
storage::cpu_kfirst backend_storage_traits(naive_with_threadpool<ThreadPool>);
template <class ThreadPool>
timer_dummy backend_timer_impl(naive_with_threadpool<ThreadPool>);
template <class ThreadPool>
inline char const *backend_name(naive_with_threadpool<ThreadPool> const &) {
return "naive";
}
} // namespace naive_impl_
using naive_impl_::naive_with_threadpool;
namespace gpu_impl_ {
template <class, class>
struct gpu;
template <class ThreadBlockSizes, class LoopBlockSizes>
storage::gpu backend_storage_traits(gpu<ThreadBlockSizes, LoopBlockSizes>);
template <class ThreadBlockSizes, class LoopBlockSizes>
timer_cuda backend_timer_impl(gpu<ThreadBlockSizes, LoopBlockSizes>);
template <class ThreadBlockSizes, class LoopBlockSizes>
inline char const *backend_name(gpu<ThreadBlockSizes, LoopBlockSizes> const &) {
return "gpu";
}
} // namespace gpu_impl_
using gpu_impl_::gpu;
} // namespace gridtools::fn::backend
|
