Skip to content
Snippets Groups Projects
Commit e454c84e authored by Markus Holzer's avatar Markus Holzer
Browse files

Clean implementation

parent cbdf82c2
Branches
Tags
1 merge request!403[Fix] Vectorisation of Abs
Pipeline #69210 passed
...@@ -614,12 +614,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter): ...@@ -614,12 +614,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
return None return None
def _print_Abs(self, expr): def _print_Abs(self, expr):
vector_type = False if isinstance(get_type_of_expression(expr), (VectorType, VectorMemoryAccess)):
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 self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs)
return super()._print_Abs(expr) return super()._print_Abs(expr)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment