From e090f70a90d6a9a9107ea0c6bf6a7b283d9e1fe1 Mon Sep 17 00:00:00 2001
From: Daniel Bauer <daniel.j.bauer@fau.de>
Date: Thu, 9 Nov 2023 08:18:26 +0100
Subject: [PATCH] distinguish casts (scalar -> vector) and (vector -> vector)
---
pystencils/backends/cbackend.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pystencils/backends/cbackend.py b/pystencils/backends/cbackend.py
index 33d977e78..93e04ab9f 100644
--- a/pystencils/backends/cbackend.py
+++ b/pystencils/backends/cbackend.py
@@ -678,7 +678,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
else:
if arg.is_Number and not isinstance(arg, (sp.core.numbers.Infinity, sp.core.numbers.NegativeInfinity)):
return self._typed_vectorized_number(arg, data_type)
- elif isinstance(arg, TypedSymbol):
+ elif isinstance(arg, TypedSymbol) and not isinstance(arg.dtype, VectorType):
return self._typed_vectorized_symbol(arg, data_type)
elif isinstance(arg, (InverseTrigonometricFunction, TrigonometricFunction, HyperbolicFunction)) \
and base_type == BasicType('float32'):
--
GitLab