Add `fmpi_world_size()`

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
 ==============================================================================*/

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;
+}