Skip to content
Snippets Groups Projects

Fix #62

Closed Markus Holzer requested to merge holzer/pystencils:FixCasting into master
Files
2
@@ -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'
Loading