shzeros return shared memory array only if nprocs>1

evalf · Oct 24, 2016 · e6f7e4b · e6f7e4b
1 parent 15c2145
commit e6f7e4b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/nutils/parallel.py b/nutils/parallel.py
@@ -165,8 +165,11 @@ def shzeros( shape, dtype=float ):
     dtype = numpy.int8
   else:
     raise Exception( 'invalid dtype: %r' % dtype )
-  buf = multiprocessing.RawArray( typecode, int(size) )
-  return numpy.frombuffer( buf, dtype ).reshape( shape )
+  if core.getprop( 'nprocs', 1 ) == 1:
+    return numpy.zeros( shape=shape, dtype=dtype )
+  else:
+    buf = multiprocessing.RawArray( typecode, int(size) )
+    return numpy.frombuffer( buf, dtype ).reshape( shape )
 
 def pariter( iterable ):
   'iterate parallel'