diff --git a/devito/data/data.py b/devito/data/data.py index 7a49ad77b5..b3aabd4269 100644 --- a/devito/data/data.py +++ b/devito/data/data.py @@ -50,13 +50,13 @@ class Data(np.ndarray): def __new__(cls, shape, dtype, decomposition=None, modulo=None, allocator=ALLOC_ALIGNED, distributor=None, padding=0): - assert len(shape) == len(modulo) ndarray, memfree_args = allocator.alloc(shape, dtype, padding=padding) obj = ndarray.view(cls) obj._allocator = allocator obj._memfree_args = memfree_args obj._decomposition = decomposition or (None,)*len(shape) obj._modulo = modulo or (False,)*len(shape) + assert len(shape) == len(obj._modulo) obj._distributor = distributor # This cannot be a property, as Data objects constructed from this