Skip to content
Snippets Groups Projects
Commit 52e67bc6 authored by Jan Hönig's avatar Jan Hönig Committed by Markus Holzer
Browse files

Fix RNG test cases

parent 42a9e630
Branches
Tags
1 merge request!292Rebase of pystencils Type System
...@@ -611,7 +611,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter): ...@@ -611,7 +611,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
basic_data_type = data_type.base_type basic_data_type = data_type.base_type
symbol = self._print(expr) symbol = self._print(expr)
if basic_data_type != expr.dtype: if basic_data_type != expr.dtype:
symbol = f'(({basic_data_type.data_type})({symbol}))' symbol = f'(({basic_data_type})({symbol}))'
instruction = 'makeVecConst' instruction = 'makeVecConst'
if basic_data_type.is_bool(): if basic_data_type.is_bool():
......
...@@ -86,7 +86,7 @@ def test_boolean_before_loop(): ...@@ -86,7 +86,7 @@ def test_boolean_before_loop():
ast = ps.create_kernel(a, cpu_vectorize_info={'instruction_set': supported_instruction_sets[-1]}) ast = ps.create_kernel(a, cpu_vectorize_info={'instruction_set': supported_instruction_sets[-1]})
kernel = ast.compile() kernel = ast.compile()
kernel(f=f_arr, g=g_arr, t2=1.0) kernel(f=f_arr, g=g_arr, t2=1.0)
print(g) # print(g)
np.testing.assert_array_equal(g_arr, 1.0) np.testing.assert_array_equal(g_arr, 1.0)
kernel(f=f_arr, g=g_arr, t2=-1.0) kernel(f=f_arr, g=g_arr, t2=-1.0)
np.testing.assert_array_equal(g_arr, 42.0) np.testing.assert_array_equal(g_arr, 42.0)
...@@ -102,8 +102,7 @@ def test_vec_maskstore(instruction_set, dtype): ...@@ -102,8 +102,7 @@ def test_vec_maskstore(instruction_set, dtype):
c = [Conditional(data.center() < 1.0, Block([SympyAssignment(data.center(), 2.0)]))] c = [Conditional(data.center() < 1.0, Block([SympyAssignment(data.center(), 2.0)]))]
assignmets = NodeCollection(c) assignmets = NodeCollection(c)
ast = ps.create_kernel(assignmets, target=Target.CPU, ast = ps.create_kernel(assignmets, target=Target.CPU, cpu_vectorize_info={'instruction_set': instruction_set})
cpu_vectorize_info={'instruction_set': instruction_set})
kernel = ast.compile() kernel = ast.compile()
kernel(data=data_arr) kernel(data=data_arr)
np.testing.assert_equal(data_arr[:3, :], 2.0) np.testing.assert_equal(data_arr[:3, :], 2.0)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment