From e18eb7d0c3848b5f751807f7a865db6c1f025a8c Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Wed, 12 Oct 2022 16:17:44 +0200
Subject: [PATCH] testing
---
pystencils/cpu/vectorization.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/pystencils/cpu/vectorization.py b/pystencils/cpu/vectorization.py
index c1362d34..0fcd32fd 100644
--- a/pystencils/cpu/vectorization.py
+++ b/pystencils/cpu/vectorization.py
@@ -321,7 +321,10 @@ def insert_vector_casts(ast_node, instruction_set, default_float_type='double'):
return sp.Piecewise(*[(r, c) for r, c in zip(casted_results, casted_conditions)])
elif isinstance(expr, TypedSymbol):
- return CastFunc(expr, VectorType(expr.dtype, instruction_set['width']))
+ if isinstance(expr.dtype, BasicType):
+ return CastFunc(expr, VectorType(expr.dtype, instruction_set['width']))
+ else:
+ return expr
elif isinstance(expr, (sp.Number, BooleanAtom)):
return expr
else:
--
GitLab