Skip to content
Snippets Groups Projects
Commit 1acd3925 authored by Markus Holzer's avatar Markus Holzer
Browse files

Fix imports

parent 0b7de74d
No related branches found
No related tags found
No related merge requests found
...@@ -917,4 +917,4 @@ ...@@ -917,4 +917,4 @@
}, },
"nbformat": 4, "nbformat": 4,
"nbformat_minor": 2 "nbformat_minor": 2
} }
\ No newline at end of file
...@@ -270,7 +270,7 @@ def insert_vector_casts(ast_node, instruction_set, default_float_type='double'): ...@@ -270,7 +270,7 @@ def insert_vector_casts(ast_node, instruction_set, default_float_type='double'):
new_arg = visit_expr(expr.args[0], default_type) new_arg = visit_expr(expr.args[0], default_type)
base_type = get_type_of_expression(expr.args[0]).base_type if type(expr.args[0]) is VectorMemoryAccess \ base_type = get_type_of_expression(expr.args[0]).base_type if type(expr.args[0]) is VectorMemoryAccess \
else get_type_of_expression(expr.args[0]) else get_type_of_expression(expr.args[0])
pw = sp.Piecewise((-new_arg, new_arg < cast_func(0, base_type.numpy_dtype)), pw = sp.Piecewise((-new_arg, new_arg < CastFunc(0, base_type.numpy_dtype)),
(new_arg, True)) (new_arg, True))
return visit_expr(pw, default_type) return visit_expr(pw, default_type)
elif expr.func in handled_functions or isinstance(expr, sp.Rel) or isinstance(expr, BooleanFunction): elif expr.func in handled_functions or isinstance(expr, sp.Rel) or isinstance(expr, BooleanFunction):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment