diff --git a/src/pystencils/backends/cbackend.py b/src/pystencils/backends/cbackend.py index cd245e6214065acd8a36dd1f36deb4e2a6636893..8f0bae12f751aacc891fc66fba0ab07195c36287 100644 --- a/src/pystencils/backends/cbackend.py +++ b/src/pystencils/backends/cbackend.py @@ -614,12 +614,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter): return None def _print_Abs(self, expr): - vector_type = False - if isinstance(expr.args[0], VectorMemoryAccess): - vector_type = True - if isinstance(expr.args[0], CastFunc) and isinstance(expr.args[0].args[1], VectorType): - vector_type = True - if vector_type: + if isinstance(get_type_of_expression(expr), (VectorType, VectorMemoryAccess)): return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs) return super()._print_Abs(expr)