Work around different mpi4py versions

firedrakeproject · Feb 17, 2015 · ef6b20d · ef6b20d
1 parent b469c6b
commit ef6b20d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/firedrake/halo.py b/firedrake/halo.py
@@ -18,7 +18,11 @@ def _get_mtype(dat):
         return _MPI_types[key]
     except KeyError:
         try:
-            btype = MPI.__TypeDict__[dat.dtype.char]
+            tdict = MPI.__TypeDict__
+        except AttributeError:
+            tdict = MPI._typedict
+        try:
+            btype = tdict[dat.dtype.char]
         except KeyError:
             raise RuntimeError("Unknown base type %r", dat.dtype)
         if dat.cdim == 1: