From a122bef13517d3db21d33754ef4158b326eb2d5a Mon Sep 17 00:00:00 2001
From: "raphael.bach" <raphael.bach@etu.hesge.ch>
Date: Fri, 1 Jul 2022 21:46:21 +0200
Subject: [PATCH] Add `fmpi_world_size()`
---
include/fmpi_core.h | 4 ++++
src/fmpi_core.c | 12 ++++++++++++
2 files changed, 16 insertions(+)
diff --git a/include/fmpi_core.h b/include/fmpi_core.h
index 9654644..8604f17 100644
--- a/include/fmpi_core.h
+++ b/include/fmpi_core.h
@@ -134,6 +134,10 @@ int fmpi_run_task(const struct fmpi_ctx * ctx, struct fmpi_task * task);
fmpi_sync()
------------------------------------------------------------------------------*/
int fmpi_sync(const struct fmpi_ctx * ctx);
+/*------------------------------------------------------------------------------
+ fmpi_world_size()
+------------------------------------------------------------------------------*/
+int fmpi_world_size(const struct fmpi_ctx * ctx);
/*==============================================================================
MACRO
==============================================================================*/
diff --git a/src/fmpi_core.c b/src/fmpi_core.c
index 8989b49..3898044 100644
--- a/src/fmpi_core.c
+++ b/src/fmpi_core.c
@@ -137,3 +137,15 @@ int fmpi_sync(const struct fmpi_ctx * const ctx)
}
return err;
}
+/*------------------------------------------------------------------------------
+ fmpi_world_size()
+------------------------------------------------------------------------------*/
+int fmpi_world_size(const struct fmpi_ctx * const ctx)
+{
+ assert(ctx != NULL);
+ const int size = fmpi_mpi_world_size(ctx->mpi);
+ if(size < FMPI_SUCCESS) {
+ FMPI_RAISE_ERROR(ctx->err_handler, "FMPI", "fmpi_mpi_world_size() failed!");
+ }
+ return size;
+}
--
GitLab