diff --git a/pystencils_tests/test_vectorization_specific.py b/pystencils_tests/test_vectorization_specific.py index 2dc4e2de998afe02cd207308ebe11f6264fee43a..0148f2f411d17ff8be0828f5e7f21a7de76e049e 100644 --- a/pystencils_tests/test_vectorization_specific.py +++ b/pystencils_tests/test_vectorization_specific.py @@ -77,6 +77,7 @@ def test_strided(instruction_set, dtype): default_number_float=npdtype) ast = ps.create_kernel(update_rule, config=config) assert len(warn) == 0 + ps.show_code(ast) func = ast.compile() ref_func = ps.create_kernel(update_rule).compile() @@ -86,6 +87,13 @@ def test_strided(instruction_set, dtype): func(g=dst, f=arr) ref_func(g=ref, f=arr) + + # print(dst) + # print(ref) + + print(np.sum(dst)) + print(np.sum(ref)) + np.testing.assert_almost_equal(dst, ref, 13 if dtype == 'double' else 5)