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
|
/*
* 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/storage/builder.hpp>
#include <iostream>
using namespace gridtools;
#ifdef GT_CUDACC
#include <gridtools/storage/gpu.hpp>
using traits_t = storage::gpu;
#else
#include <gridtools/storage/cpu_ifirst.hpp>
using traits_t = storage::cpu_ifirst;
#endif
int main() {
uint_t Ni = 10;
uint_t Nj = 12;
uint_t Nk = 20;
auto const builder = storage::builder<traits_t>.dimensions(Ni, Nj, Nk).id<0>();
auto phi = builder
.name("phi") //
.type<float>() //
.initializer([](int i, int j, int k) { return i + j + k; }) //
.build();
auto lap = builder.name("lap").type<double>().value(-1).build();
std::cout << phi->name() << "\n";
auto phi_view = phi->host_view();
phi_view(1, 2, 3) = 3.1415;
std::cout << "phi_view(1, 2, 3) = " << phi_view(1, 2, 3) << std::endl;
std::cout << "j length = " << phi->lengths()[1] << std::endl;
}
|
