Throw MPI_Abort if the number of tasks is not at least 2

matthewcarbone · Jan 19, 2024 · f725383 · f725383
1 parent 6d09a0b
commit f725383
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main.cpp b/src/main.cpp
@@ -96,6 +96,14 @@ int main(int argc, char *argv[])
         auto_determine_dynamics_(&p);
         save_and_log_config(p);    
     }
+
+    // Ensure we are running with at least 2 mpi ranks
+    int mpi_world_size;
+    MPI_Comm_size(MPI_COMM_WORLD, &mpi_world_size);
+    if (mpi_world_size < 2){
+        std::cerr << "MPI WORLD SIZE must be greater than or equal to 2" << std::endl;
+        MPI_Abort(MPI_COMM_WORLD, 1);
+    }
 
     initialize_grids_and_directories(p);