diff --git a/tests/nbackend/test_vectorization.py b/tests/nbackend/test_vectorization.py index 905af57e6de648224ed658e89696a67c775a1d58..b60dc24774566d67eaa271c6ab775374746d89cf 100644 --- a/tests/nbackend/test_vectorization.py +++ b/tests/nbackend/test_vectorization.py @@ -256,15 +256,14 @@ def test_set(vectorization_setup: VectorTestSetup): kernel = create_vector_kernel(update, f, setup) - for trailing_iters in range(1, setup.lanes): - shape = (23, 1) - f_arr = create_numpy_array_with_layout( - shape, layout=(1, 0), dtype=setup.index_dtype.numpy_dtype - ) + shape = (23, 1) + f_arr = create_numpy_array_with_layout( + shape, layout=(1, 0), dtype=setup.index_dtype.numpy_dtype + ) - f_arr[:] = 42 + f_arr[:] = 42 - kernel(f=f_arr) + kernel(f=f_arr) - reference = np.array(range(shape[0])).reshape(shape) - np.testing.assert_equal(f_arr, reference) + reference = np.array(range(shape[0])).reshape(shape) + np.testing.assert_equal(f_arr, reference)