Skip to content
Snippets Groups Projects
Commit 4f1a99ce authored by Michael Kuron's avatar Michael Kuron :mortar_board: Committed by Michael Kuron
Browse files

Correctly check for VectorType in _print_Function

parent 584b4255
No related branches found
No related tags found
1 merge request!248Fix RNG vectorization for LB
......@@ -670,7 +670,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
else:
is_boolean = get_type_of_expression(arg) == create_type("bool")
is_integer = get_type_of_expression(arg) == create_type("int") or \
(isinstance(arg, TypedSymbol) and arg.dtype.is_int())
(isinstance(arg, TypedSymbol) and not isinstance(arg.dtype, VectorType) and arg.dtype.is_int())
instruction = 'makeVecConstBool' if is_boolean else \
'makeVecConstInt' if is_integer else 'makeVecConst'
return self.instruction_set[instruction].format(self._print(arg), **self._kwargs)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment