Log on all ranks when PYOP2_SPMD_STRICT is enabled

firedrakeproject · Oct 2, 2024 · 532c0f9 · 532c0f9
1 parent 137e301
commit 532c0f9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/firedrake/logging.py b/firedrake/logging.py
@@ -5,6 +5,7 @@
 import tsfc.logging             # noqa: F401
 import pyop2.logger             # noqa: F401
 
+from pyop2.configuration import configuration
 from pyop2.mpi import COMM_WORLD
 
 
@@ -79,7 +80,7 @@ def set_log_handlers(handlers=None, comm=COMM_WORLD):
             handler = logging.StreamHandler()
             handler.setFormatter(logging.Formatter(fmt="%(name)s:%(levelname)s %(message)s"))
 
-        if comm is not None and comm.rank != 0:
+        if comm is not None and comm.rank != 0 and not configuration["spmd_strict"]:
             handler = logging.NullHandler()
 
         logger.addHandler(handler)