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