From 7b1d7319c9b492c21f40e8c0249fe30e37250c7e Mon Sep 17 00:00:00 2001
From: Grzegorz Kowal <grzegorz@amuncode.org>
Date: Tue, 30 Jul 2024 22:24:24 -0300
Subject: [PATCH] AMUN: Print more info about MPI nodes/processes.

Signed-off-by: Grzegorz Kowal <grzegorz@amuncode.org>
---
 sources/amun.F90 | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/sources/amun.F90 b/sources/amun.F90
index ebdbf9c..3abf1e4 100644
--- a/sources/amun.F90
+++ b/sources/amun.F90
@@ -37,7 +37,7 @@ program amun
   use mpitools  , only : initialize_mpitools, finalize_mpitools
   use mpitools  , only : verbose => master, nproc, check_status
 #ifdef MPI
-  use mpitools  , only : reduce_sum, nprocs
+  use mpitools  , only : reduce_sum, nodes, nprocs
 #endif /* MPI */
   use parameters, only : read_parameters, finalize_parameters
   use random    , only : initialize_random, finalize_random
@@ -136,14 +136,19 @@ program amun
   call print_welcome(verbose)
 #ifdef MPI
   call print_section(verbose, "Parallelization")
-  call print_parameter(verbose, "MPI processes" , nprocs)
+  call print_parameter(verbose, "MPI nodes"     , nodes)
+  call print_parameter(verbose, "MPI processes per node" , nprocs / nodes)
 #else /* MPI */
 !$ call print_section(verbose, "Parallelization")
 #endif /* MPI */
 !$omp parallel
 !$omp master
 !$ nthreads = omp_get_num_threads()
+#ifdef MPI
+!$ call print_parameter(verbose, "OpenMP threads per process", nthreads)
+#else /* MPI */
 !$ call print_parameter(verbose, "OpenMP threads", nthreads)
+#endif /* MPI */
 !$omp end master
 !$omp end parallel