diff --git a/pystencils_tests/test_vectorization_specific.py b/pystencils_tests/test_vectorization_specific.py index 7929266d9ca47b7e8987b9431882d7f946c7bd9d..9b9d34694ccd63a4ab3a5aa4ce4f7df23b10f897 100644 --- a/pystencils_tests/test_vectorization_specific.py +++ b/pystencils_tests/test_vectorization_specific.py @@ -61,7 +61,6 @@ def test_vectorized_abs(instruction_set, dtype): @pytest.mark.parametrize('dtype', ('float', 'double')) @pytest.mark.parametrize('instruction_set', supported_instruction_sets) def test_strided(instruction_set, dtype): - npdtype = np.float64 if dtype == 'double' else np.float32 type_string = "float64" if dtype == 'double' else "float32" f, g = ps.fields(f"f, g : {type_string}[2D]") @@ -84,10 +83,16 @@ def test_strided(instruction_set, dtype): ref_config = pystencils.config.CreateKernelConfig(default_number_float=type_string) ref_func = ps.create_kernel(update_rule, config=ref_config).compile() + # For some reason other array creations fail on the emulated ppc pipeline size = (25, 19) - arr = np.reshape(np.linspace(0, 1, size[0] * size[1]), size).astype(npdtype) - dst = np.zeros_like(arr, dtype=npdtype) - ref = np.zeros_like(arr, dtype=npdtype) + arr = np.zeros(size).astype(type_string) + for i in range(size[0]): + for j in range(size[1]): + arr[i, j] = i * j + + + dst = np.zeros_like(arr, dtype=type_string) + ref = np.zeros_like(arr, dtype=type_string) func(g=dst, f=arr) ref_func(g=ref, f=arr)