Skip to content
Snippets Groups Projects
Commit dbef149e authored by Frederik Hennig's avatar Frederik Hennig
Browse files

update jacobi test case

parent 09a94bf5
No related branches found
No related tags found
1 merge request!5Extend mdspan interface and fix mdspan memory layout mapping
Pipeline #71121 passed
......@@ -5,21 +5,21 @@
namespace stdex = std::experimental;
using field_t = stdex::mdspan<double, stdex::extents< int64_t, std::dynamic_extent, std::dynamic_extent>>;
using scalar_field_t = stdex::mdspan<double, stdex::extents< int64_t, std::dynamic_extent, std::dynamic_extent, 1>>;
using field_t = stdex::mdspan<double, stdex::extents<int64_t, std::dynamic_extent, std::dynamic_extent>, stdex::layout_left>;
using scalar_field_t = stdex::mdspan<double, stdex::extents<int64_t, std::dynamic_extent, std::dynamic_extent, 1>, stdex::layout_left>;
int main(void)
{
auto data_f = std::make_unique<double[]>(64);
scalar_field_t f{ data_f.get(), 8, 8 };
scalar_field_t f{data_f.get(), 8, 8};
auto data_u = std::make_unique<double[]>(64);
field_t u{ data_u.get(), 8, 8 };
field_t u{data_u.get(), 8, 8};
auto data_u_tmp = std::make_unique<double[]>(64);
field_t u_tmp{ data_u_tmp.get(), 8, 8 };
field_t u_tmp{data_u_tmp.get(), 8, 8};
double h { 1.0 / 7.0 };
double h{1.0 / 7.0};
gen::jacobi_smooth(f, h, u_tmp, u);
}
......@@ -20,8 +20,8 @@ with SourceFileGenerator() as sfg:
poisson_kernel = sfg.kernels.create(poisson_jacobi)
sfg.function("jacobi_smooth")(
sfg.map_field(u_src, mdspan.from_field(u_src)),
sfg.map_field(u_dst, mdspan.from_field(u_dst)),
sfg.map_field(u_src, mdspan.from_field(u_src, layout_policy="layout_left")),
sfg.map_field(u_dst, mdspan.from_field(u_dst, layout_policy="layout_left")),
sfg.map_field(f, mdspan.from_field(f)),
sfg.call(poisson_kernel)
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment