From 3240de21d0073d09e19f70bc09c6fbcc53e34dcb Mon Sep 17 00:00:00 2001
From: "raphael.bach" <raphael.bach@etu.hesge.ch>
Date: Tue, 28 Jun 2022 20:11:13 +0200
Subject: [PATCH] Fix offset computation in `fmpi_partition_block_1d()`

---
 src/fmpi_domain.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/fmpi_domain.c b/src/fmpi_domain.c
index e44e914..6578769 100644
--- a/src/fmpi_domain.c
+++ b/src/fmpi_domain.c
@@ -93,7 +93,7 @@ static struct fmpi_data fmpi_partition_block_1d(
     const size_t rem = data->cnt % (size_t)ctx->mpi->size;
     const size_t cnt = (rank < rem) ? (cnt_per_proc + 1) : cnt_per_proc;
     const size_t size = cnt * data->type.size;
-    const size_t offset = (rank * size) + rem;
+    const size_t offset = (rank * size) + (rank < rem ? 0 : rem);
     return (struct fmpi_data){
         .type = data->type,
         .cnt = cnt,
-- 
GitLab