From 786ecb7e5a95d84c26c14de0656dce050e7a8478 Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Sun, 2 Jun 2019 17:28:05 +0200 Subject: mg2d: export the local component extents in public API --- mg2d.c | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) (limited to 'mg2d.c') diff --git a/mg2d.c b/mg2d.c index a0865c2..578dfe0 100644 --- a/mg2d.c +++ b/mg2d.c @@ -1430,6 +1430,12 @@ MG2DContext *mg2d_solver_alloc(size_t domain_size) ctx->domain_size = domain_size; + ctx->local_start[0] = 0; + ctx->local_start[1] = 0; + + ctx->local_size[0] = domain_size; + ctx->local_size[1] = domain_size; + return ctx; fail: mg2d_solver_free(&ctx); @@ -1509,6 +1515,10 @@ MG2DContext *mg2d_solver_alloc_mpi(MPI_Comm comm, const size_t local_start[2], ctx->priv->dg = dg; ctx->priv->local_component = rank; ctx->domain_size = dg->domain_size[0]; + ctx->local_start[0] = local_start[0]; + ctx->local_start[1] = local_start[1]; + ctx->local_size[0] = local_size[0]; + ctx->local_size[1] = local_size[1]; return ctx; fail: @@ -1716,8 +1726,8 @@ int mg2d_init_guess(MG2DContext *ctx, const double *src, priv->transfer_init->src.step[0] = src_step[0]; priv->transfer_init->src.step[1] = src_step[1]; - priv->transfer_init->dst.size[0] = ctx->domain_size; - priv->transfer_init->dst.size[1] = ctx->domain_size; + priv->transfer_init->dst.size[0] = ctx->local_size[0]; + priv->transfer_init->dst.size[1] = ctx->local_size[1]; priv->transfer_init->dst.step[0] = ctx->step[0]; priv->transfer_init->dst.step[1] = ctx->step[1]; -- cgit v1.2.3