Fix #62

Fixes problems around #62 (closed)

pystencils/cpu/vectorization.py

@@ -320,7 +320,12 @@ def insert_vector_casts(ast_node, instruction_set, default_float_type='double'):
                                  for a, t in zip(new_conditions, types_of_conditions)]
 
             return sp.Piecewise(*[(r, c) for r, c in zip(casted_results, casted_conditions)])
-        elif isinstance(expr, (sp.Number, TypedSymbol, BooleanAtom)):
+        elif isinstance(expr, TypedSymbol):
+            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:
             raise NotImplementedError(f'Due to defensive programming we handle only specific expressions.\n'