From e454c84e76aa5dae3c8ee1884540c2d025fa52fa Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Thu, 26 Sep 2024 09:58:30 +0200
Subject: [PATCH] Clean implementation
---
src/pystencils/backends/cbackend.py | 7 +------
1 file changed, 1 insertion(+), 6 deletions(-)
diff --git a/src/pystencils/backends/cbackend.py b/src/pystencils/backends/cbackend.py
index cd245e621..8f0bae12f 100644
--- a/src/pystencils/backends/cbackend.py
+++ b/src/pystencils/backends/cbackend.py
@@ -614,12 +614,7 @@ 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:
+ if isinstance(get_type_of_expression(expr), (VectorType, VectorMemoryAccess)):
return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs)
return super()._print_Abs(expr)
--
GitLab