diff options
Diffstat (limited to 'mg2d.c')
-rw-r--r-- | mg2d.c | 14 |
1 files changed, 12 insertions, 2 deletions
@@ -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]; |