From 3379489d0efe17b57bce70c767734fdb8de3fdc4 Mon Sep 17 00:00:00 2001 From: Max Date: Tue, 21 Oct 2025 19:47:54 +0200 Subject: [PATCH] Added verbose argument to MPIwrapper. default: False --- psydac/ddm/mpi.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/psydac/ddm/mpi.py b/psydac/ddm/mpi.py index 40f29ef06..840df835b 100644 --- a/psydac/ddm/mpi.py +++ b/psydac/ddm/mpi.py @@ -38,16 +38,22 @@ def Barrier(self): class MPIwrapper: - def __init__(self, use_mpi: bool = False): + def __init__( + self, + use_mpi: bool = False, + verbose: bool = False, + ): self.use_mpi = use_mpi if use_mpi: from mpi4py import MPI self._MPI = MPI - print("MPI is enabled") + if verbose: + print("MPI is enabled") else: self._MPI = MockMPI() - print("MPI is NOT enabled") + if verbose: + print("MPI is NOT enabled") @property def MPI(self): @@ -89,7 +95,10 @@ def COMM_WORLD(self): mpi_enabled = False # TODO: add environment variable for mpi use -mpi_wrapper = MPIwrapper(use_mpi=mpi_enabled) +mpi_wrapper = MPIwrapper( + use_mpi=mpi_enabled, + verbose=False, +) # TYPE_CHECKING is True when type checking (e.g., mypy), but False at runtime. if TYPE_CHECKING: @@ -97,4 +106,4 @@ def COMM_WORLD(self): mpi = MPI else: - mpi = mpi_wrapper.MPI \ No newline at end of file + mpi = mpi_wrapper.MPI