diff --git a/pystencils_tests/test_vectorization.py b/pystencils_tests/test_vectorization.py index fbf5044e0792417844ad2ba3dd5376590d5fd0c4..19f266b12b55ce66f971e0634e11f7178c6f70bc 100644 --- a/pystencils_tests/test_vectorization.py +++ b/pystencils_tests/test_vectorization.py @@ -185,8 +185,7 @@ def test_piecewise2(instruction_set=instruction_set): np.testing.assert_equal(arr, np.ones_like(arr)) -def test_piecewise3(): - instruction_set = "sve512" +def test_piecewise3(instruction_set=instruction_set): arr = np.zeros((22, 22)) @ps.kernel @@ -196,9 +195,9 @@ def test_piecewise3(): g[0, 0] @= 1.0 / (s.b + s.k) if f[0, 0] > 0.0 else 1.0 ast = ps.create_kernel(test_kernel) - ps.show_code(ast) + # ps.show_code(ast) vectorize(ast, instruction_set=instruction_set) - ps.show_code(ast) + # ps.show_code(ast) ast.compile()