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
No related branches found
No related tags found
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 register or to comment