Skip to content
Snippets Groups Projects

[Fix] Vectorisation of Abs

Merged Markus Holzer requested to merge holzer/pystencils:VectorAbs into master
All threads resolved!
@@ -614,7 +614,9 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
return None
def _print_Abs(self, expr):
return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs)
if isinstance(expr.args[0], VectorMemoryAccess):
return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs)
return super()._print_Abs(expr)
def _typed_vectorized_number(self, expr, data_type):
basic_data_type = data_type.base_type