compiler: add switch for static_cast vs reinterpret_cast

Author: mloubout

devito/arch/compiler.py

@@ -181,6 +181,8 @@ def __init__(self):
 
     fields = {'cc', 'ld'}
     default_cpp = False
+    _cxxstd = 'c++14'
+    _cstd = 'c99'
 
     def __init__(self, **kwargs):
         _name = kwargs.pop('name', self.__class__.__name__)
@@ -255,7 +257,7 @@ def version(self):
 
     @property
     def std(self):
-        return 'c++14' if self._cpp else 'c99'
+        return self._cxxstd if self._cpp else self._cstd
 
     def get_version(self):
         result, stdout, stderr = call_capture_output((self.cc, "--version"))
@@ -491,7 +493,7 @@ def __init_finalize__(self, **kwargs):
         language = kwargs.pop('language', configuration['language'])
         platform = kwargs.pop('platform', configuration['platform'])
 
-        if platform is NvidiaDevice:
+        if isinstance(platform, NvidiaDevice):
             self.cflags.remove(f'-std={self.std}')
             # Add flags for OpenMP offloading
             if language in ['C', 'openmp']:

devito/operator/operator.py

@@ -1424,7 +1424,8 @@ def parse_kwargs(**kwargs):
         kwargs['compiler'] = configuration['compiler'].__new_with__()
 
     # Make sure compiler and language are compatible
-    if kwargs['compiler']._cpp and kwargs['language'] in ['C', 'openmp']:
+    if compiler is not None and kwargs['compiler']._cpp and \
+            kwargs['language'] in ['C', 'openmp']:
         kwargs['language'] = 'CXX' if kwargs['language'] == 'C' else 'CXXopenmp'
     if 'CXX' in kwargs['language'] and not kwargs['compiler']._cpp:
         kwargs['compiler'] = kwargs['compiler'].__new_with__(cpp=True)

devito/passes/iet/languages/CXX.py

@@ -91,5 +91,6 @@ def _print_Cast(self, expr):
         tstr = self._print(expr._C_ctype)
         if 'void' in tstr:
             return super()._print_Cast(expr)
-        cast = f'static_cast<{tstr}{self._print(expr.stars)}>'
+        caster = 'reinterpret_cast' if expr.reinterpret else 'static_cast'
+        cast = f'{caster}<{tstr}{self._print(expr.stars)}>'
         return self._print_UnaryOp(expr, op=cast, parenthesize=True)

devito/passes/iet/languages/openacc.py

@@ -236,11 +236,11 @@ def place_devptr(self, iet, **kwargs):
 
             dpf = List(body=[
                 self.lang.mapper['map-serial-present'](hp, tdp),
-                Block(body=DummyExpr(tdp, cast_mapper(tdp.dtype)(hp)))
+                Block(body=DummyExpr(tdp, cast_mapper(tdp.dtype)(hp, reinterpret=True)))
             ])
 
             ffp = FieldFromPointer(f._C_field_dmap, f._C_symbol)
-            ctdp = cast_mapper((hp.dtype, '*'))(tdp)
+            ctdp = cast_mapper((hp.dtype, '*'))(tdp, reinterpret=True)
             cast = DummyExpr(ffp, ctdp)
 
             ret = Return(ctdp)

devito/symbolics/extended_sympy.py

@@ -384,9 +384,9 @@ class Cast(UnaryOp):
     """
 
     __rargs__ = ('base', )
-    __rkwargs__ = ('dtype', 'stars')
+    __rkwargs__ = ('dtype', 'stars', 'reinterpret')
 
-    def __new__(cls, base, dtype=None, stars=None, **kwargs):
+    def __new__(cls, base, dtype=None, stars=None, reinterpret=False, **kwargs):
         try:
             if issubclass(dtype, np.generic) and sympify(base).is_Number:
                 base = sympify(dtype(base))
@@ -397,6 +397,7 @@ def __new__(cls, base, dtype=None, stars=None, **kwargs):
         obj = super().__new__(cls, base)
         obj._stars = stars or ''
         obj._dtype = dtype
+        obj._reinterpret = reinterpret
         return obj
 
     def _hashable_content(self):
@@ -412,6 +413,10 @@ def stars(self):
     def dtype(self):
         return self._dtype
 
+    @property
+    def reinterpret(self):
+        return self._reinterpret
+
     @property
     def _C_ctype(self):
         ctype = ctypes_vector_mapper.get(self.dtype, self.dtype)