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

Fix scalar case

parent ad8cf5b2
No related merge requests found
Pipeline #69181 passed with stages
in 16 minutes and 47 seconds
...@@ -614,7 +614,12 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter): ...@@ -614,7 +614,12 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
return None return None
def _print_Abs(self, expr): def _print_Abs(self, expr):
vector_type = False
if isinstance(expr.args[0], 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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment