1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
auto phi = make_storage();
auto phi_new = make_storage();
halo_descriptor boundary_i(halo, halo, halo, Ni - halo - 1, Ni);
halo_descriptor boundary_j(halo, halo, halo, Nj - halo - 1, Nj);
auto grid = make_grid(boundary_i, boundary_j, axis_t{kmax, Nk - kmax});
auto const spec = [](auto phi, auto phi_new) {
GT_DECLARE_TMP(double, lap);
return execute_parallel() //
.stage(lap_function(), phi, lap) //
.stage(smoothing_function_3(), phi, lap, phi_new); //
};
run(spec, stencil_backend_t(), grid, phi, phi_new);
|
