compiler: Add Basic._mem_shared_remote

Author: FabioLuporini

devito/types/array.py

@@ -90,15 +90,16 @@ class Array(ArrayBasic):
         to 'local'. Used to override `_mem_local` and `_mem_mapped`.
     scope : str, optional
         The scope in the given memory space. Allowed values: 'heap', 'stack',
-        'static', 'constant', 'shared'. 'static' refers to a static array in a
-        C/C++ sense. 'constant' and 'shared' mean that the Array represents an
-        object allocated in so called constant and shared memory, respectively,
-        which are typical of device architectures. If 'shared' is specified but
-        the underlying architecture doesn't have something akin to shared memory,
-        the behaviour is unspecified. If 'constant' is specified but the underlying
-        architecture doesn't have something akin to constant memory, the Array
-        falls back to a global, const, static array in a C/C++ sense.
-        Note that not all scopes make sense for a given space.
+        'static', 'constant', 'shared', 'shared-remote'. 'static' refers to a
+        static array in a C/C++ sense. 'constant' and 'shared' mean that the
+        Array represents an object allocated in so called constant and shared
+        memory, respectively, which are typical of device architectures. If
+        'shared' is specified but the underlying architecture doesn't have
+        something akin to shared memory, the behaviour is unspecified. If
+        'constant' is specified but the underlying architecture doesn't have
+        something akin to constant memory, the Array falls back to a global,
+        const, static array in a C/C++ sense.  Note that not all scopes make
+        sense for a given space.
     grid : Grid, optional
         Only necessary for distributed-memory parallelism; a Grid contains
         information about the distributed Dimensions, hence it is necessary
@@ -131,7 +132,8 @@ def __init_finalize__(self, *args, **kwargs):
         super().__init_finalize__(*args, **kwargs)
 
         self._scope = kwargs.get('scope', 'heap')
-        assert self._scope in ['heap', 'stack', 'static', 'constant', 'shared']
+        assert self._scope in ['heap', 'stack', 'static', 'constant', 'shared',
+                               'shared-remote']
 
         self._initvalue = kwargs.get('initvalue')
         assert self._initvalue is None or self._scope != 'heap'
@@ -174,6 +176,10 @@ def _mem_heap(self):
     def _mem_shared(self):
         return self._scope == 'shared'
 
+    @property
+    def _mem_shared_remote(self):
+        return self._scope == 'shared-remote'
+
     @property
     def _mem_constant(self):
         return self._scope == 'constant'
@@ -445,9 +451,10 @@ def initvalue(self):
 
     for i in ('_mem_internal_eager', '_mem_internal_lazy', '_mem_local',
               '_mem_mapped', '_mem_host', '_mem_stack', '_mem_constant',
-              '_mem_shared', '__padding_dtype__', '_size_domain', '_size_halo',
-              '_size_owned', '_size_padding', '_size_nopad', '_size_nodomain',
-              '_offset_domain', '_offset_halo', '_offset_owned', '_dist_dimensions',
+              '_mem_shared', '_mem_shared_remote', '__padding_dtype__',
+              '_size_domain', '_size_halo', '_size_owned', '_size_padding',
+              '_size_nopad', '_size_nodomain', '_offset_domain',
+              '_offset_halo', '_offset_owned', '_dist_dimensions',
               '_C_get_field', 'grid', 'symbolic_shape',
               *AbstractFunction.__properties__):
         locals()[i] = property(lambda self, v=i: getattr(self.c0, v))

devito/types/basic.py

@@ -42,7 +42,7 @@ class CodeSymbol:
         * "liveness": `_mem_external`, `_mem_internal_eager`, `_mem_internal_lazy`
         * "space": `_mem_local`, `_mem_mapped`, `_mem_host`
         * "scope": `_mem_stack`, `_mem_heap`, `_mem_global`, `_mem_shared`,
-                   `_mem_constant`
+                   `_mem_shared_remote`, `_mem_constant`
 
     For example, an object that is `<_mem_internal_lazy, _mem_local, _mem_heap>`
     is allocated within the Operator entry point, on either the host or device
@@ -211,6 +211,14 @@ def _mem_shared(self):
         """
         return False
 
+    @property
+    def _mem_shared_remote(self):
+        """
+        True if the associated data is allocated in so called remote shared
+        memory, False otherwise.
+        """
+        return False
+
 
 class Basic(CodeSymbol):