diff --git a/src/pystencils/backends/cbackend.py b/src/pystencils/backends/cbackend.py index f2a35e4390079f99668f1f8046fd3a1aae47346a..cd245e6214065acd8a36dd1f36deb4e2a6636893 100644 --- a/src/pystencils/backends/cbackend.py +++ b/src/pystencils/backends/cbackend.py @@ -614,7 +614,12 @@ 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: return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs) return super()._print_Abs(expr)