From 5f163219750b07d1e8cedd20f1b88b9e6606666b Mon Sep 17 00:00:00 2001 From: Sebastian-Belkner Date: Sun, 2 Jul 2023 17:19:32 +0200 Subject: [PATCH] tweak --- delensalot/core/mpi.py | 3 +-- delensalot/run.py | 7 ++++++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/delensalot/core/mpi.py b/delensalot/core/mpi.py index 1a72daec..527f8b49 100644 --- a/delensalot/core/mpi.py +++ b/delensalot/core/mpi.py @@ -59,7 +59,6 @@ def enable(): global disabled, verbose, has_key, mpisupport, name disabled = False verbose = True - print(os.environ.keys()) has_key = lambda key : key in os.environ.keys() mpisupport = 'srun' in os.environ['_'] or 'mpirun' in os.environ['_'] pmisupport = 'PMI_CRAY_NO_SMP_ORDER' in os.environ.keys() @@ -88,7 +87,7 @@ def disable(): rank = 0 size = 1 finalize = lambda: -1 - log.info('mpi.py : diabled, rank %s in %s' % (rank, size)) + log.info('mpi.py : disabled, rank %s in %s' % (rank, size)) def init(): diff --git a/delensalot/run.py b/delensalot/run.py index 8c2ec553..12cd3648 100644 --- a/delensalot/run.py +++ b/delensalot/run.py @@ -112,17 +112,21 @@ def purge_TEMPconf(self): if __name__ == '__main__': """Entry point for the command line """ + os.environ['USE_PLANCKLENS_MPI'] = "False" lparser = lerepi_parser() if lparser.validate(): parser = lparser.get_parser() config_handler = config_handler(parser) + print(1) if dh.dev_subr in parser.__dict__: dh.dev(parser, config_handler.TEMP) sys.exit() - + print(1.1) if mpi.size > 1: + print(1.2) if mpi.rank == 0: + print(1.3) mpi.disable() config_handler.collect_models() mpi.enable() @@ -130,6 +134,7 @@ def purge_TEMPconf(self): else: mpi.receive(None, source=mpi.ANY_SOURCE) config_handler.collect_models() + print(2) try: config_handler.run()