From 85a11efc344f8b6781478e5d45a1843fc4d0f8eb Mon Sep 17 00:00:00 2001 From: Mathew Madhavacheril Date: Mon, 2 Apr 2018 17:29:50 -0400 Subject: [PATCH 1/2] mpi_disable_option --- mpi.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/mpi.py b/mpi.py index f0e4996..fd46505 100644 --- a/mpi.py +++ b/mpi.py @@ -1,6 +1,14 @@ """Utilities for making mpi use safer and easier.""" import sys -from mpi4py.MPI import * + +try: + disable_mpi_env = os.environ['DISABLE_MPI'] + disable_mpi = True if disable_mpi_env.lower().strip() == "true" else False +except: + disable_mpi = False + +if not(disable_mpi): + from mpi4py.MPI import * # Uncaught exceptions don't cause mpi to abort. This can lead to thousands of # wasted CPU hours def cleanup(type, value, traceback): From 77926b8fa9a48bb9f2a520f1fb91daf0b0717f58 Mon Sep 17 00:00:00 2001 From: Mathew Madhavacheril Date: Mon, 2 Apr 2018 17:34:56 -0400 Subject: [PATCH 2/2] mpi_disable_option --- mpi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mpi.py b/mpi.py index fd46505..d702475 100644 --- a/mpi.py +++ b/mpi.py @@ -1,5 +1,5 @@ """Utilities for making mpi use safer and easier.""" -import sys +import sys,os try: disable_mpi_env = os.environ['DISABLE_MPI']