Skip to content
Snippets Groups Projects

[Fix] Vectorisation of Abs

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