From cbdf82c2d26a825d57911794a081bc04215da2ed Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Wed, 25 Sep 2024 11:42:15 +0200
Subject: [PATCH] Fix scalar case
---
src/pystencils/backends/cbackend.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/src/pystencils/backends/cbackend.py b/src/pystencils/backends/cbackend.py
index f2a35e43..cd245e62 100644
--- a/src/pystencils/backends/cbackend.py
+++ b/src/pystencils/backends/cbackend.py
@@ -614,7 +614,12 @@ 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:
return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs)
return super()._print_Abs(expr)
--
GitLab