compiler: add local array with fixed size type

Author: mloubout

devito/ir/iet/visitors.py

@@ -26,7 +26,7 @@
                           c_restrict_void_p, sorted_priority)
 from devito.types.basic import AbstractFunction, AbstractSymbol, Basic
 from devito.types import (ArrayObject, CompositeObject, Dimension, Pointer,
-                          IndexedData, DeviceMap)
+                          IndexedData, DeviceMap, LocalObject)
 
 
 __all__ = ['FindApplications', 'FindNodes', 'FindWithin', 'FindSections',
@@ -353,7 +353,10 @@ def _gen_rettype(self, obj):
         elif isinstance(obj, (FieldFromComposite, FieldFromPointer)):
             return self._gen_value(obj.function.base, 0).typename
         else:
-            return None
+            try:
+                return obj._type_.__name__
+            except AttributeError:
+                return None
 
     def _args_decl(self, args):
         """Generate cgen declarations from an iterable of symbols and expressions."""

devito/mpi/routines.py

@@ -1395,7 +1395,7 @@ def _arg_values(self, args=None, **kwargs):
 class AllreduceCall(Call):
 
     def __init__(self, arguments, **kwargs):
-        super().__init__('MPI_Allreduce', arguments)
+        super().__init__('MPI_Allreduce', arguments, **kwargs)
 
 
 class ReductionBuilder(object):
@@ -1422,6 +1422,6 @@ def make(self, dr):
         op = self.mapper[dr.op]
 
         arguments = [inplace, Byref(f), Integer(1), mpitype, op, comm]
-        allreduce = AllreduceCall(arguments)
+        allreduce = AllreduceCall(arguments, writes=f)
 
         return allreduce

devito/symbolics/extended_sympy.py

@@ -172,9 +172,9 @@ def __new__(cls, call, pointer, params=None, **kwargs):
             pointer = Symbol(pointer)
         if isinstance(call, str):
             call = Symbol(call)
-        elif not isinstance(call, Basic):
-            raise ValueError("`call` must be a `devito.Basic` or a type "
-                             "with compatible interface")
+        elif not isinstance(call, (BasicWrapperMixin, Basic)):
+            raise ValueError(f"`call` {call} must be a `devito.Basic` or a type "
+                             f"with compatible interface, not {type(call)}")
         _params = []
         for p in as_tuple(params):
             if isinstance(p, str):

devito/types/dense.py

@@ -148,7 +148,7 @@ def wrapper(self):
                         # Perhaps user only wants to initialise the physical domain
                         self._initializer(self.data)
                 else:
-                    self.data_with_halo.fill(0)
+                    self._data_allocated.fill(0)
 
             return func(self)
         return wrapper
@@ -296,7 +296,8 @@ def shape_global(self):
 
     @property
     def symbolic_shape(self):
-        return tuple(self._C_get_field(FULL, d).size for d in self.dimensions)
+        return DimensionTuple(*[self._C_get_field(FULL, d).size for d in self.dimensions],
+                              getters=self.dimensions)
 
     @property
     def size_global(self):

devito/types/dimension.py

@@ -1482,7 +1482,7 @@ def _arg_defaults(self, **kwargs):
     def _arg_values(self, *args, **kwargs):
         return {}
 
-    def _arg_check(self, *args):
+    def _arg_check(self, *args, **kwargs):
         """A CustomDimension performs no runtime checks."""
         return